o V h @s>ddlmZmZddlmZddlmZddlmZmZddl m Z ddl m Z m Z ddlZeeZd Zd ed ejd efd dZdede ededed ejd e ef ddZded ejd e efddZdeded ejd e efddZded ejd e efddZded ejd efddZdeded ejd efd d!ZdS)")datetime timedelta) Collection)database)SupportActivityCreateSupportActivityUpdate)str_to_objectid)OptionalListNsupport_activitiesactivitydbreturnc sz+|t}|}t|d<||}t|j|d<|d|d<td|d|WSt y?}zt d|d}~ww)zO Insert a new support activity into the support_activities collection. created_date_id activity_idzCreated activity %szError creating activity: %sN) COLLECTION_NAMEdictrutcnow insert_onestr inserted_idloggerinfo Exceptionerror)r ractivities_collection activity_dataresulter!8/var/www/html/moveengine/app/v1/services/saas/support.pycreate_activity_service s    r# account_id type_filterskiplimitc sz?|t}d|i}|r||d<||||dd}g}|2z3dHW} t| d| d<| d| d<|| q#6|WStyS} zt d| d} ~ ww) z| Retrieve a paginated list of support activities for the specified account. Optionally filter by activity type. r$typerNrrzError listing activities: %s) rfindr&r'sortrappendrrr) r$r%r&r'rrquerycursor activitiesdocr r!r!r"list_activities_services&   r1rc slz |t}|dt|i}|rt|d|d<|d|d<|WSty5}ztd||d}~ww)z7 Retrieve a single support activity by its ID. rrz Error retrieving activity %s: %sN)rfind_oner rrrrrrrr r r!r!r"get_activity_service3s r4activity_updatec sz:|t}|jdd}|dt|id|i}|jdkr WdS|dt|i}|r9t|d|d<|d|d<|WStyO}zt d||d}~ww) z. Update an existing support activity. T) exclude_unsetr$setrNrzError updating activity %s: %s) rr update_oner matched_countr2rrrr)rr5rr update_datarr r r!r!r"update_activity_serviceBs"   r;c sz,|t}|dt|i}|sWdS|dt|it|d|d<|d|d<|WStyA}ztd||d}~ww)zB Delete a support activity and return the deleted record. rNrzError deleting activity %s: %s)rr2r delete_onerrrrr3r!r!r"delete_activity_serviceUs r=c sz;|t}||dddgid}t}||dd|id}d}|2z 3d HW}|d 7}q%6||d |d }|WStyO} ztd | d } ~ ww) a3 Aggregate support metrics for an account: - Total count of support activities (of type 'comment' or 'note') - Count of upcoming meetings (activity type 'meeting' with scheduled date in additional_data) - For pipeline, the latest update might be shown elsewhere (here set as None) z$incommentnote)r$r(meetingz$gte)r$r(zadditional_data.scheduled_daterN)comments_countupcoming_meetings_counttrial_days_remaining last_updatedz'Error aggregating dashboard metrics: %s) rcount_documentsrrr* isoformatrrr) r$rrrBnowmeetings_cursormeetings_count_ dashboardr r!r!r"get_support_dashboard_servicefs8   rM new_statusc szW|d}|dt|i}|std|dd}|dt|id|tdid|d d |d |||d td }|t}||}t |j |d<|d|d<|WSt yl} zt d|| d } ~ ww)z Update the sales pipeline status for an account. This function does two things: 1. Updates the account record in the 'accounts' collection. 2. Inserts a support activity record of type 'pipeline_update' to log the change. accountsrzAccount not foundsales_pipeline_statusunknownr7)rP updated_datepipeline_updateNzPipeline status changed from z to ) old_statusrN)r(r$user_idcontentadditional_datarrz1Error updating pipeline status for account %s: %s)r2r ValueErrorgetr8rrrrrrrrr) r$rNraccounts_collectionaccountrTrrrr r!r!r"update_pipeline_status_services8    r\)rrpymongo.collectionrapp.dbrmodels.saas.supportmodelrrlibraries.objectr typingr r logging getLogger__name__rrMongoDBrr#rintr1r4r;r=rMr\r!r!r!r"s     ,  "