h. dZddlZddlZddlmZmZmZddlmZddl m Z dede defd Z d ed ee dee dee fd Z y) zh Common utility functions for handling object permission updates across organizations, teams, and keys. N)DictOptionalUnion)verbose_proxy_logger) PrismaClient data_dict prisma_clientreturnc K| td|jd}|rH|jjj d|id{}|r |j }||d<|S7#t $r|j}Y$wxYww)a Helper method to attach object_permission to a dictionary if object_permission_id is set. This function: 1. Checks if the dictionary has an object_permission_id 2. If found, queries the database for the corresponding object permission 3. Converts the object permission to a dictionary format 4. Attaches it to the input dictionary under the 'object_permission' key Args: data_dict: The dictionary to attach object_permission to prisma_client: The database client Returns: Dict: The input dictionary with object_permission attached if found Raises: ValueError: If prisma_client is None NPrisma client not foundobject_permission_idwhereobject_permission) ValueErrorgetdblitellm_objectpermissiontable find_unique model_dump Exceptiondict)rr r rs t/var/www/Befach/backend/env/lib/python3.12/site-packages/litellm/proxy/management_helpers/object_permission_utils.py attach_object_permission_to_dictrs.233$==)?@"/"2"2"P"P"\"\)+?@#]#    =$5$@$@$B!.?I) *   =$5$:$:$<! =s6A BA-BA/&B/B B B  B data_jsonexisting_object_permission_idcbK| td|jdd}|y|xsttj}i}|j j jd|id{}||jdd}t|trtj|}t|tr|j||j j jd|i||d d{}tj d ||j"S77*w) a Common logic for handling object permission updates across organizations, teams, and keys. This function: 1. Extracts `object_permission` from data_json 2. Looks up existing object permission if it exists 3. Merges new permissions with existing ones 4. Upserts to the LiteLLM_ObjectPermissionTable 5. Returns the object_permission_id Args: data_json: The data dictionary containing the object_permission to update existing_object_permission_id: The current object_permission_id from the entity (can be None) prisma_client: The database client Returns: Optional[str]: The object_permission_id after the update/creation, or None if no object_permission to process Raises: ValueError: If prisma_client is None Nr rr rT) exclude_unset exclude_none)createupdate)rdatazcreated_object_permission_row: )rpopstruuiduuid4rrrr isinstancejsonloadsrr!upsertrdebugr )rrr new_object_permissionobject_permission_id_to_use existing_object_permissions_dictexisting_object_permissioncreated_object_permission_rows r&handle_update_object_permission_commonr17se4233 /8mmr8sy  ((1,&&& &RO>O>#+C=O>L)O>c] O>r7