U h_@sdZddlZddlmZmZddlmZmZmZmZddl m Z m Z m Z ddl mZddlmZddlmZdd lmZdd lmZdd lmZdd lmZd dlmZmZmZd dlm Z m!Z!m"Z"m#Z#d dl$m%Z%eZ&e'e(Z)e dddd?ee*e+e+ee*ee*ee+ee+ee+e,ee-ee+eee+efe+d ddZ.e dddeee+efee+dddZ/e dddd@e+ee*ee*e,dddZ0e dddee1e*dddZ2e ddddAe*ee+efd!d"d#Z3e ddde+ee+efd$d%d&Z4e ddddBe*e*ee+efd(d)d*Z5e ddddCe*eee+efd,d-d.Z6e dddeee+efe,d/d0d1Z7e ddddDe*ee+efe+ee+efd3d4d5Z8e ddde+ee+efe,d6d7d8Z9e d9d:Z:e d;d<Z;e d=d>Z0/var/www/html/Focus/src/apps/activities/tasks.pylog_activity_async2s, r@)activities_datar*cCs zg}|D]v}t|d|d|d|d|d|d|d|d|d d |d |d |d id }||q tj|}dd|D}tdt|d|WStk r}zDt dt ||j j |j kr|jdd|j j d|W5d}~XYnXdS)a Log multiple activities in bulk asynchronously. This task creates multiple activity records efficiently using bulk operations. Args: activities_data: List of activity data dictionaries Returns: list: List of activity IDs rrr r!r"r#r$r%r&Tr'r(r)r+cSsg|]}t|jqSr>)r5r4).0r<r>r>r? sz'log_activities_bulk..z Bulk logged z activitieszFailed to bulk log activities: r,r-r.N)rgetappendr0 bulk_creater2r3lenr6r7r5r8r9rr:)r;rA activitiesdatar<Zcreated_activitiesZ activity_idsr=r>r>r?log_activities_bulkos6    rJ)daterr!r*c Csz*t|d}ttjj|||d\}}tjj |d}|rT|j |d}|rd|j |d}| |_ |j dd |_ |j dd |_ |j dd jd dd }|rt|t||_nd |_|d  |_tdd|_|td|d|d|dW5QRWdSQRXWnftk r} zFtdt| |jj |j!kr||j"dd|jj dWYdSd} ~ XYnXdS)aV Update activity summary for a specific date. This task calculates and updates activity statistics for a given date, user, and/or category. Args: date: Date string (YYYY-MM-DD) user_id: User ID (optional) category_id: Category ID (optional) Returns: bool: Success status %Y-%m-%d)rKrr!created_at__daterr!Tr&F)duration_ms__isnullr')flatrr$Z celery_task)Z last_updatedZcalculation_methodzActivity summary updated for z (user: z , category: )z#Failed to update activity summary: r,r-r.N)#rstrptimerKratomicrr0 get_or_createrfiltercounttotal_activitiessuccessful_activitiesfailed_activities values_listsumrGavg_duration_msvaluesdistinct unique_ipsr now isoformatr)saver2r3r6r7r5r8r9rr:) r;rKrr!Z summary_datesummarycreatedrH durationsr=r>r>r?update_activity_summarysN      ri) summary_keysr*c CszNd}|D],\}}}tj|||gd}|r |d7}q td|d|WStk r}zDtdt||jj |j kr|j dd|jj d WYdSd }~XYnXd S) z Update multiple activity summaries in bulk. Args: summary_keys: List of tuples (date, user_id, category_id) Returns: int: Number of summaries updated r)argsrz Bulk updated z activity summariesz*Failed to bulk update activity summaries: r,r-r.N) riapplyrDr2r3r6r7r5r8r9rr:)r;rj updated_countZdate_strrr!successr=r>r>r?update_activity_summaries_bulks   roZ) days_to_keepr*c Csz^t}t|}t}||}tj|j|d||||d}t d|d|WSt k r}zBt dt ||j j|jkr|jdd|j jd|W5d }~XYnXd S) a' Clean up old activities based on retention policy. This task removes activities older than the specified number of days to maintain database performance. Args: days_to_keep: Number of days to keep activities Returns: dict: Cleanup results )sender deleted_count)rsrqduration_secondsZ completed_atzActivity cleanup completed: z activities deletedz"Failed to cleanup old activities: r,r-r.N)r rcr total_secondsrsend __class__rdr2r3r6r7r5r8r9rr:)r;rq start_timersend_timedurationresultr=r>r>r?cleanup_old_activities_tasks* r|)rKr*c Csz\t|d}tjj|d}ttt |tj tt |tj d}t |jdddddjtd d d d d }t |ddjtd dd}t |jddiddjtd dd}|jdd}|t |djtd ddd dd} |||||| td} d|} t| | dtd|| WStk r} zDtdt| |jj|jkr|j dd|jjd | W5d } ~ XYnXd S)!z Generate daily activity report. This task creates a comprehensive report of activities for a specific date. Args: date: Date string (YYYY-MM-DD) Returns: dict: Activity report data rLrM) start_dateend_dateF) user__isnull user__emailuser__first_nameuser__last_namer4)activity_countz-activity_countN category__namecategory__code)rYz-counthourzEXTRACT(hour FROM created_at))selectrQr()Z total_errors top_errors)rK overall_stats top_userscategory_breakdownhourly_distribution error_summary generated_atZdaily_activity_report_iQz$Daily activity report generated for z*Failed to generate daily activity report: r,r-r.)!rrUrKrr0rXrr make_awarecombinemintimemaxlistr`annotateCountorder_byextrarYrcrdrsetr2r3r6r7r5r8r9rr:) r;rK report_daterHrrrrZerror_activitiesrreport cache_keyr=r>r>r?generate_daily_activity_reportEs     r)rdaysr*c Cs zddlm}tjj|d}|||}|j|j||j|j d|d<t |d<d|d|}t ||d td ||WStjk rtd |d iYStk r}zDtd t||jj|jkr|jdd|jjd|W5d}~XYnXdS)a- Generate user activity report. This task creates a comprehensive report of activities for a specific user over a given period. Args: user_id: User ID days: Number of days to include in the report Returns: dict: User activity report data r)get_user_activity_summaryr4)r4email full_name is_active date_joineduserrZuser_activity_report__z(User activity report generated for user User z not found for activity reportz)Failed to generate user activity report: r,r-r.N)utilsrUserr0rDr4r get_full_namerrrdr rcrrr2r3 DoesNotExistr7r6r5r8r9rr:)r;rrrrrfrr=r>r>r?generate_user_activity_reports.   r)hoursr*c CszZt|}dd|D}|r&t|d|d}t||dtdt|d|WStk r}zBt dt ||j j |j kr|jd d |j j d |W5d }~XYnXd S) a Monitor and process security alerts. This task checks for security-related activities and generates alerts for suspicious behavior. Args: hours: Number of hours to check for alerts Returns: list: List of security alerts cSsg|]}|ddkr|qS)severityhighr>)rBalertr>r>r?rCs z+monitor_security_alerts..Zsecurity_alerts_hrzSecurity monitoring completed: z alerts foundz#Failed to monitor security alerts: r,r-r.N)rsend_security_alert_emaildelayrrr2r3rGr6r7r5r8r9rr:)r;ralertsZhigh_severity_alertsrr=r>r>r?monitor_security_alertss  r)rr*c Cszttdg}|s"tdWdSdt|d}|t|td}td|}td|}t|||tj |dd t d t|d Wd St k r}zDt d t ||jj|jkr|jdd|jjdWYdSd}~XYnXdS)z Send security alert email notifications. Args: alerts: List of security alerts Returns: bool: Success status ZSECURITY_ALERT_EMAILSz.No admin emails configured for security alertsFzSecurity Alert: z suspicious activities detected)rZ alert_count timestampz%activities/emails/security_alert.htmlz$activities/emails/security_alert.txtsubjectmessage html_message from_emailrecipient_list fail_silentlyzSecurity alert email sent to z recipientsTz%Failed to send security alert email: r,r-r.N)getattrr r2warningrGr rcrr DEFAULT_FROM_EMAILr3r6r7r5r8r9rr:)r;rZ admin_emailsrcontextr plain_messager=r>r>r?rs6      rcsv)rfiltersformatr*c Csz~ddlm}ddlm}tjj|d}tj}|drN|j |dd}|drh|j |dd}|d r|j |d d }|d r|j |d d }|d dk r|j |d d}|dkrt |}dt dd} d} n.ddlm} | |}dt dd} d} d| } || ||d} | | t||||t d}t|j|td|d| |WStjk rtd|diYStk r}zDtd t||jj|j kr|j!d!d"|jjd#|W5d}~XYnXdS)$ae Export activities to file asynchronously. This task exports activities based on filters and sends the result to the requesting user. Args: user_id: User ID requesting the export filters: Activity filters format: Export format ('csv' or 'json') Returns: dict: Export result information r) ContentFile)default_storagerr})created_at__gter~)created_at__lterrOr!rPr&NrQrZactivities_export_z %Y%m%d_%H%M%Sz.csvztext/csvr)export_activities_to_jsonz.jsonzapplication/jsonzexports/activities/zutf-8)filename file_path file_sizeZ record_countrr created_atz#Activity export completed for user z: rz not found for activity exportzFailed to export activities: r,r-r.)"django.core.files.baserdjango.core.files.storagerrr0rDrallrXrr rcstrftimerrreencoderGrYrdsend_export_notification_emailrrr2r3rr7r6r5r8r9rr:)r;rrrrrrrHcontentr content_typerrZ saved_pathr{r=r>r>r?export_activities_task8sX           r)r export_infor*c Csz`d}|tjd|dd}td|}td|}t|||tj|gddtd |Wd Stk r}zDtd t ||j j |j kr|j d d |j j dWYdSd}~XYnXdS)z Send export completion notification email. Args: email: User email address export_info: Export information Returns: bool: Success status zActivity Export Completedz/activities/exports/r)rZ download_urlz*activities/emails/export_notification.htmlz)activities/emails/export_notification.txtFrz"Export notification email sent to Tz*Failed to send export notification email: r,r-r.N)r ZSITE_URLrr rr2r3r6r7r5r8r9rr:)r;rrrrrrr=r>r>r?rs,    rcCsxttdd}t|dtjj dd}|D]}tj|d|j dq8t |dt d|dS) z Daily task to update activity summaries. This task runs daily to update activity summaries for the previous day. r)rrLT)rrPz*Daily activity summary task completed for N)r rcrrKrirrrr0rXr4rr2r3) yesterday categoriescategoryr>r>r?daily_activity_summary_tasksrcCs$ttdd}t|tddS)z Weekly task to clean up old activities. This task runs weekly to remove old activities based on the retention policy. ZACTIVITIES_RETENTION_DAYSrpzWeekly cleanup task initiatedN)rr r|rr2r3)retention_daysr>r>r?weekly_cleanup_tasks  rcCstjddtddS)z Hourly task to monitor security alerts. This task runs hourly to check for security-related activities and generate alerts. r)rz)Hourly security monitoring task completedN)rrr2r3r>r>r>r?hourly_security_monitoring_tasks r) NNNNNTNNN)NN)rp)r)r)r)=__doc__loggingrrtypingrrrrceleryrr r django.confr django.utilsr django.core.mailr django.template.loaderrdjango.contrib.authr django.dbrdjango.core.cachermodelsrrrrrrrrsignalsrr getLogger__name__r2intr5boolfloatr@rJritupleror|rrrrrrrrrr>r>r>r?s           < "2  N " 0 Y 7  ) 6  W /