U h7@sdZddlmZmZmZddlmZddlmZddl m Z ddl m Z ddl mZddlmZmZmZdd lmZdd lmZmZmZmZdd lmZmZmZmZmZdd l m!Z!Gd ddej"Z#Gdddej"Z$Gdddej"Z%Gdddej&Z'dS)z REST API Views for Jingles Application This module contains Django REST Framework views for managing jingle detection, ad break analysis, and related operations through a RESTful API interface. )viewsetsstatus permissions)action)Response)PageNumberPagination)DjangoFilterBackend)timezone)CountAvgQ) timedelta)JingleTemplateJingleDetectionAdBreakDetectionStatistics)JingleTemplateSerializerJingleDetectionSerializerAdBreakSerializerDetectionStatisticsSerializerDetectionSummarySerializer)extract_iframes_from_segmentc@seZdZdZejdZe Z e j gZ egZdddgZdddgZddd gZdgZd d Zed d gddddZed d gddddZedd gdddZdS)JingleTemplateViewSetz ViewSet for managing jingle templates via REST API. Provides CRUD operations for jingle templates along with custom actions for template management and statistics. created_by is_activecategorynameslug description created_atsimilarity_thresholdcCs|j|jjddS)z-Set the current user as the template creator.)rN)saverequestuser)self serializerr&;/var/www/html/StreamProcessor/src/apps/jingles/api/views.pyperform_create+sz$JingleTemplateViewSet.perform_createTgetdetailmethodsNcCsj|}|jddd}|jd}|jd}|rD|j|d}|rT|j|d}t|dd }t|j S) a  Get all detections for a specific template. Args: request: HTTP request with optional filters pk: Template primary key Returns: Response: List of detections for the template session__channelsegment-detection_time date_fromdate_toZdetection_time__gte)Zdetection_time__lteTmany) get_object detectionsselect_relatedorder_by query_paramsr)filterrrdata)r$r"pktemplater6r0r1r%r&r&r'r6/s      z JingleTemplateViewSet.detectionsc Cs|}|j}|}|jdd}|dkrr||d}|jtddd}|jtt dd d } n d}d}d} ||t |d |rt |d nd| d } t | S)z Get detailed statistics for a specific template. Args: request: HTTP request pk: Template primary key Returns: Response: Template statistics data T is_confirmedrdconfidence_score)avgrBdaysr2)total_detectionsconfirmed_detections accuracy_rateavg_confidencerecent_detections) r5r6allcountr: aggregater r nowr roundr) r$r"r<r=r6rHrIrJrKrLstatsr&r&r' statisticsLs(    z JingleTemplateViewSet.statisticsFcCs&|jjdd}|j|dd}t|jS)z Get all active jingle templates.Trr3querysetr:get_serializerrr;)r$r"active_templatesr%r&r&r'activevszJingleTemplateViewSet.active)N)N)__name__ __module__ __qualname____doc__robjectsrMr7rVrserializer_classrIsAuthenticatedpermission_classesrfilter_backendsfilterset_fields search_fieldsordering_fieldsorderingr(rr6rSrYr&r&r&r'rs        ) rc@seZdZdZejdddZe Z e j gZ egZdddgZddgZd gZed d gd dddZed d gd dddZeddgd ddZeddgd ddZd S)JingleDetectionViewSetz ViewSet for managing jingle detections via REST API. Provides access to detection data with filtering and confirmation capabilities. r=r-r.sessionr?detection_timerAr/Tpostr*NcCs*|}d|_|||}t|jS)z Confirm a jingle detection as accurate. Args: request: HTTP request pk: Detection primary key Returns: Response: Updated detection data T)r5r?r!rWrr;)r$r"r< detectionr%r&r&r'confirms  zJingleDetectionViewSet.confirmcCsp|}d|_|z.tjj|j|jd}|jd7_|Wntj k rZYnX| |}t |j S)z Mark a detection as false positive. Args: request: HTTP request pk: Detection primary key Returns: Response: Updated detection data F)rhr=) r5r?r!rr^r)rhr=false_positives DoesNotExistrWrr;)r$r"r<rkrRr%r&r&r'rejects   zJingleDetectionViewSet.rejectFr)cCs,|jddd}|j|dd}t|jS)z+Get recent detections across all templates.r/N2Tr3)rVr8rWrr;)r$r"rLr%r&r&r'recentszJingleDetectionViewSet.recentc Cst}|}|tdd}|j}|jj|d}|jjdd}d}|dkrd||d}ttj j|dj t d d  d d d  dd} t|jj|ddd dd d dddd} |||t|d| | d} t| } t| jS)z Get detection summary statistics. Args: request: HTTP request Returns: Response: Summary statistics data rCrD)detection_time__dateTr>rr@)Zdetections__detection_time__gter6)detection_countz-detection_countNrrtr2r=r-r/ template__namesession__channel__namerirArF)rHdetections_todayrIrJ top_templatesrecent_activity)r rPdater rVrNr:listrr^annotater r8valuesr7rQrrr;) r$r"rPtodayZ last_7_daysrHryrIrJrzr{Z summary_datar%r&r&r'summarysb      zJingleDetectionViewSet.summary)N)N)rZr[r\r]rr^rMr7rVrr_rr`rarrbrcrerfrrlrprrrr&r&r&r'rg~s(         rgc@seZdZdZejdddZe Z e j gZ egZdddgZdd gZd gZed d gd dddZeddgd ddZeddgd ddZdS)AdBreakViewSetz ViewSet for managing ad breaks via REST API. Provides access to ad break data with external API integration capabilities. r-start_detection__templateend_detection__templaterregionis_sent_to_api start_timeduration_secondsz -start_timeTrjr*Nc Cs|}|s"tdditjdS|jr:tdditjdSz6ddlm}|t |j }tdd|j d tj dWSt k r}ztdt |itj dWYSd }~XYnXd S) z Send ad break data to external API. Args: request: HTTP request pk: Ad break primary key Returns: Response: API submission result errorzAd break is not completerzAd break already sent to APIrF)send_ad_break_notificationTz"Ad break queued for API submission)successmessagetask_idN)r5 is_completerrHTTP_400_BAD_REQUESTrtasksrdelaystridHTTP_202_ACCEPTED ExceptionHTTP_500_INTERNAL_SERVER_ERROR)r$r"r<ad_breakrtasker&r&r' send_to_apis4   zAdBreakViewSet.send_to_apiFr)cCs&|jjdd}|j|dd}t|jS)zGet all active ad breaks.rYrTr3rU)r$r"Z active_breaksr%r&r&r'rYEszAdBreakViewSet.activecCs2t}|jj|d}|j|dd}t|jS)zGet ad breaks from today.)start_time__dateTr3)r rPr|rVr:rWrr;)r$r"rZ today_breaksr%r&r&r'rLs zAdBreakViewSet.today)N)rZr[r\r]rr^rMr7rVrr_rr`rarrbrcrerfrrrYrr&r&r&r'rs$    ,   rc@sbeZdZdZejddZe Z e j gZ egZddgZdddgZdgZed d gd d d ZdS)DetectionStatisticsViewSetz ViewSet for detection statistics (read-only). Provides access to performance metrics and analysis data for jingle detection accuracy. r=r-rhlast_detectionrHrJz-last_detectionFr)r*c Cs|j}|s$tdddgdStdd|D}tdd|D}|dkr\||dnd}g}|jddd d d D] }||jj | |j d q~t |d |t jjdd|d}t|S)z Get overall performance statistics. Args: request: HTTP request Returns: Response: Performance summary data r)overall_accuracyrHrXbest_performerscss|] }|jVqdSN)rH.0sr&r&r' }sz9DetectionStatisticsViewSet.performance..css|] }|jVqdSr)rIrr&r&r'r~sr@rv)Ztotal_detections__gtez-accuracy_rateNru) template_namerJrHrFTrT)rVrMexistsrsumr:r8appendr=rrJrHrQrr^rN) r$r"rRrHZtotal_confirmedrrstatZperformance_datar&r&r' performancegs0  z&DetectionStatisticsViewSet.performanceN)rZr[r\r]rr^rMr7rVrr_rr`rarrbrcrerfrrr&r&r&r'rUs   rN)(r]rest_frameworkrrrrest_framework.decoratorsrrest_framework.responserZrest_framework.paginationrZdjango_filters.rest_frameworkr django.utilsr django.db.modelsr r r datetimer apps.jingles.modelsrrrrZapps.jingles.api.serializersrrrrrapps.streams.tasksr ModelViewSetrrgrReadOnlyModelViewSetrr&r&r&r's        d O