o Ah @sddlmZddlmZddlmZddlmZmZddlm Z ddl m Z m Z ddl Z e eZd Zhd Zd ed ejd efddZdede ededed ejd e ef ddZded ejd e efddZdeded ejd e efddZded ejd e efddZdS))datetime) Collection)database)SupportActivityCreateSupportActivityUpdate)str_to_objectid)OptionalListNsupport_activities>ticketsupport_requestsubscription_requestactivitydbreturnc s|jtvr tdz+|t}|}t|d<||}t|j |d<|d|d<t d|d|WSt yH}zt d|d}~ww)z Insert a new customer support activity into the support_activities collection. Only allowed activity types are accepted. z9This activity type is not permitted for customer support. created_date_id activity_idzCreated customer activity %sz$Error creating customer activity: %sN)typeALLOWED_CUSTOMER_TYPES ValueErrorCOLLECTION_NAMEdictrutcnow insert_onestr inserted_idloggerinfo Exceptionerror)rractivities_collection activity_dataresulter&@/var/www/html/moveengine/app/v1/services/saas/supportcustomer.py create_customer_activity_services"     r( account_id type_filterskiplimitc szL|t}|dttid}|r|tvrtd||d<||||dd}g}|2z3dHW} t| d| d<| d| d <| | q06|WSt y`} zt d | d} ~ ww) z Retrieve a paginated list of customer support activities for the given account. Only returns records with allowed types. z$in)r)rz1Invalid activity type filter for customer supportrrNrrz%Error listing customer activities: %s) rlistrrfindr+r,sortrappendr rr!) r)r*r+r,rr"querycursor activitiesdocr%r&r&r' list_customer_activities_service%s*   r6rc sz+|t}|dt|i}|r*|dtvrtdt|d|d<|d|d<|WSty@}zt d||d}~ww)z@ Retrieve a single customer support activity by its ID. rr"Access denied to internal activityrz)Error retrieving customer activity %s: %sN) rfind_onergetrrrr rr!rrr"rr%r&r&r'get_customer_activity_service<s r;activity_updatec szE|t}|jdd}|dt|id|i}|jdkr WdS|dt|i}|rD|dtvr6tdt |d|d<|d|d <|WSt yZ}zt d ||d}~ww) z7 Update an existing customer support activity. T) exclude_unsetrz$setrNrr7rz'Error updating customer activity %s: %s) rr update_oner matched_countr8r9rrrr rr!)rr<rr" update_datar$rr%r&r&r' update_customer_activity_serviceMs&   rAc sz7|t}|dt|i}|sWdS|dtvrtd|dt|it|d|d<|d|d<|WStyL}zt d||d}~ww)zK Delete a customer support activity and return the deleted record. rNrr7rz'Error deleting customer activity %s: %s) rr8rr9rr delete_onerr rr!r:r&r&r' delete_customer_activity_servicebs" rC)rpymongo.collectionrapp.dbrmodels.saas.supportmodelrrlibraries.objectrtypingr r logging getLogger__name__rrrMongoDBrr(rintr6r;rArCr&r&r&r's    ,