U ZhI@sdZddlmZddlmZddlmZddlmZm Z ddl m Z m Z m Z ddlmZmZddlmZmZmZmZmZdd lZdd lZGd d d e e ZGd d d e e ZGddde e e ZGddde e Zd S)a# Jingle Detection Models for Stream Processor Application This module contains Django models for managing jingle detection, ad break analysis, and related functionality. These models handle the storage and management of jingle templates, detection results, and ad break timing information. )models)User)timezone)MinValueValidatorMaxValueValidator)TimestampedModel UUIDModel StatusModel) StreamSession HLSSegment)validate_file_pathvalidate_slug_formatvalidate_name_formatvalidate_confidence_scorevalidate_template_contentNc @seZdZdZejddegddZejdde gddZ ej dddZ ejd e gd d Zejd egd dZejddddZejddddddddgdddZejdejdddddZejeejd d!d"ZGd#d$d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-S).JingleTemplatea6 Model representing a jingle template for detection. Jingle templates are reference images or video frames that are used to identify specific jingles in stream content. Each template represents a unique jingle that can be detected during stream processing. Attributes: name (CharField): Human-readable name for the jingle slug (SlugField): URL-friendly identifier description (TextField): Description of the jingle content image_path (CharField): Path to the reference image file similarity_threshold (FloatField): Threshold for detection matching is_active (BooleanField): Whether this template is active category (CharField): Category classification for the jingle created_by (ForeignKey): User who uploaded this template dTz"Human-readable name for the jingle) max_lengthunique validators help_textz&URL-friendly identifier for the jinglez-Description of the jingle content and contextblankrz0Path to the reference image file for this jingle)rrrg?z5Similarity threshold for positive detection (0.0-1.0))defaultrrz4Whether this jingle template is active for detectionrdb_indexr2)Zad_startzAdvertisement Start)Zad_endzAdvertisement End)Z program_startz Program Start)Z program_endz Program End)Z transitionZ Transition)ZbrandingzChannel Branding)otherOtherrz'Category classification for this jinglerchoicesrrzstreams.ChannelZjingle_templatesz'Channel this jingle template belongs to on_delete related_namenullrrZcreated_jingle_templatesz&User who uploaded this jingle templater#r$rc@sDeZdZdZdZddgZejdgdejdgdejdgdgZdS) zJingleTemplate.MetazJingle TemplatezJingle Templatescategorynameslugfields is_activeN __name__ __module__ __qualname__ verbose_nameverbose_name_pluralorderingrIndexindexesr6r67/var/www/html/JingleDetector/src/apps/jingles/models.pyMetaus   r8cCs|jd|jdS)z-String representation of the jingle template. ())r(r'selfr6r6r7__str__szJingleTemplate.__str__cCstj|jS)z1Check if the reference image file exists on disk.)ospathexists image_pathr;r6r6r7 image_existsszJingleTemplate.image_existscCs,ztj|jWStk r&YdSXdS)z)Get the size of the reference image file.N)r>r?getsizerAOSErrorr;r6r6r7get_image_sizeszJingleTemplate.get_image_sizecCs>z$tj|jr"t|jWdSWntk r8YnXdS)z*Delete the reference image file from disk.TF)r>r?r@rAremoverDr;r6r6r7 delete_images  zJingleTemplate.delete_imageN)r.r/r0__doc__r CharFieldrr( SlugFieldr r) TextField descriptionr rA FloatFieldrZsimilarity_threshold BooleanFieldr,r' ForeignKeyCASCADEchannelr created_byr8r=rBrErGr6r6r6r7rs   rc@seZdZdZejeejdddZeje ejdddZ eje ejdddZ ej eded gd d Zejd d dZejdddZej edgdd ZejddddZejedddZGdddZddZddZddZd d!Zd"S)#JingleDetectionaO Model representing a detected jingle instance. This model records when a jingle has been detected in a stream, including the confidence score, frame information, and associated segment data. Attributes: session (ForeignKey): Stream session where detection occurred segment (ForeignKey): HLS segment containing the detection template (ForeignKey): Jingle template that was matched confidence_score (FloatField): Confidence level of the detection frame_path (CharField): Path to the extracted frame image detection_time (DateTimeField): When the detection occurred frame_timestamp (FloatField): Timestamp within the segment is_confirmed (BooleanField): Whether detection has been confirmed metadata (JSONField): Additional detection metadata Zjingle_detectionsz,Stream session where this detection occurredr&z%HLS segment containing this detection detectionsz Jingle template that was matched?z+Confidence level of the detection (0.0-1.0))rrrz!Path to the extracted frame imagerrTz When the detection was processed) auto_now_addrz5Timestamp within the segment where detection occurredFz2Whether this detection has been manually confirmedrz2Additional detection metadata and analysis resultsrrrc@sReZdZdZdZdgZejddgdejddgdejdgdejd gdgZd S) zJingleDetection.MetazJingle DetectionzJingle Detections-detection_timesessiondetection_timer*template is_confirmedconfidence_scoreNr-r6r6r6r7r8s  r8cCs |jjd|jdd|jdS)z.String representation of the jingle detection. - z.3fr9r:)r]r(r_r\r;r6r6r7r=szJingleDetection.__str__cCstj|jS)z1Check if the extracted frame file exists on disk.)r>r?r@ frame_pathr;r6r6r7 frame_existsszJingleDetection.frame_existscCsdtj|jS)z*Get URL for accessing the extracted frame.z/media/frames/)r>r?basenamerar;r6r6r7 get_frame_urlszJingleDetection.get_frame_urlcCs>z$tj|jr"t|jWdSWntk r8YnXdS)z*Delete the extracted frame file from disk.TF)r>r?r@rarFrDr;r6r6r7 delete_frames  zJingleDetection.delete_frameN) r.r/r0rHrrOr rPr[r segmentrr]rMrrr_rIra DateTimeFieldr\Zframe_timestamprNr^ JSONFielddictmetadatar8r=rbrdrer6r6r6r7rSsd rScseZdZdZejeejdddZej dddZ ej dd d d gd d dZ eje ejdddddZ eje ejdddddZejddZejddddZejddededgddZejddddZejedd d!Zejdd"d#ZGd$d%d%Zd&d'Zd(d)Zd*d+Z fd,d-Z!Z"S).AdBreaka Model representing a detected advertisement break. Ad breaks are identified by pairs of jingle detections (start and end) and represent commercial interruptions in the stream content. Attributes: session (ForeignKey): Stream session containing the ad break channel_name (CharField): Channel identifier for external APIs region (CharField): Geographic region for the ad break start_detection (ForeignKey): Jingle detection marking the start end_detection (ForeignKey): Jingle detection marking the end start_time (DateTimeField): When the ad break started end_time (DateTimeField): When the ad break ended duration_seconds (PositiveIntegerField): Duration in seconds is_sent_to_api (BooleanField): Whether data was sent to external API api_response (JSONField): Response from external API notes (TextField): Additional notes about the ad break Z ad_breaksz'Stream session containing this ad breakr&rz/Channel identifier for external API integrationrWr) Netherlandsrl)Francerm)Globalrnrnz#Geographic region for this ad breakr Zad_breaks_startedTz7Jingle detection that marked the start of this ad breakr"Zad_breaks_endedz5Jingle detection that marked the end of this ad breakz$When the advertisement break started)rz"When the advertisement break endedr%rriz#Duration of the ad break in secondsr%rrrFz3Whether this ad break data was sent to external APIrz%Response data from external API callsrYz4Additional notes or observations about this ad breakrc@sReZdZdZdZdgZejddgdejddgdejd gdejd gdgZd S) z AdBreak.MetazAdvertisement BreakzAdvertisement Breaksz -start_timer[ start_timer* channel_nameregionis_sent_to_apistatusNr-r6r6r6r7r8qs  r8cCs4|jrd|jdnd}|jd|jd|S)z&String representation of the ad break.r9zs)r`z%H:%M:%S)duration_secondsrsrrstrftime)r<Z duration_strr6r6r7r=|szAdBreak.__str__cCs(|jr$|jr$|j|j}t|SdS)z'Calculate the duration of the ad break.N)rrend_timeint total_seconds)r<deltar6r6r7calculate_durations   zAdBreak.calculate_durationcCs|jdk o|jdk S)z3Check if the ad break has both start and end times.N)rrrzr;r6r6r7 is_completeszAdBreak.is_completecs*|r|js||_tj||dS)z2Override save to automatically calculate duration.N)rrxr~supersave)r<argskwargs __class__r6r7rs z AdBreak.save)#r.r/r0rHrrOr rPr[rIrsrtrSZstart_detectionZ end_detectionrgrrrzPositiveIntegerFieldrrrxrNrurhriZ api_responserKZnotesr8r=r~rr __classcell__r6r6rr7rk s    rkc@seZdZdZejeejdddZeje ejdddZ ej ddd Z ej dd d Z ej dd d Zejd d ed edgddZejd d ed edgddZejd d ed edgddZejd d ed gddZejd d ddZGdddZddZddZddZddZd S)!DetectionStatisticsa Model for tracking jingle detection statistics and metrics. This model aggregates detection data for reporting and analysis, providing insights into detection accuracy, frequency, and performance. Attributes: session (ForeignKey): Stream session for these statistics template (ForeignKey): Jingle template for these statistics total_detections (PositiveIntegerField): Total number of detections confirmed_detections (PositiveIntegerField): Number of confirmed detections false_positives (PositiveIntegerField): Number of false positive detections avg_confidence (FloatField): Average confidence score min_confidence (FloatField): Minimum confidence score max_confidence (FloatField): Maximum confidence score detection_rate (FloatField): Detections per hour last_detection (DateTimeField): Time of last detection Zdetection_statisticsz#Stream session for these statisticsr& statisticsz$Jingle template for these statisticsrz,Total number of detections for this template)rrz'Number of manually confirmed detectionsz.Number of identified false positive detectionsTrUrVz+Average confidence score for all detectionsrqz!Minimum confidence score recordedz!Maximum confidence score recordedzNumber of detections per hourz&Timestamp of the most recent detectionroc@s@eZdZdZdZddgZdgZejddgdejdgdgZ dS)zDetectionStatistics.MetazDetection Statisticsr[r]z-last_detectionr*last_detectionN) r.r/r0r1r2unique_togetherr3rr4r5r6r6r6r7r8s r8cCs|jjd|jdS)z2String representation of the detection statistics.r`z detections)r]r(total_detectionsr;r6r6r7r=szDetectionStatistics.__str__cCs|jdkr|j|jdSdS)zBCalculate the accuracy rate as percentage of confirmed detections.rr)rconfirmed_detectionsr;r6r6r7 accuracy_rates z!DetectionStatistics.accuracy_ratecCs|jdkr|j|jdSdS)z0Calculate the false positive rate as percentage.rr)rfalse_positivesr;r6r6r7false_positive_rates z'DetectionStatistics.false_positive_ratecCstjj|j|jd}||_|jdd|_|jdkr|jddd}t |t ||_ t ||_ t||_|d}|r|j|_|dS) z7Recalculate statistics based on current detection data.)r[r]T)r^rr_)flatrZN)rSobjectsfilterr[r]countrr values_listsumlenavg_confidenceminmin_confidencemaxmax_confidenceorder_byfirstr\rr)r<rTZconfidence_scoresZlatest_detectionr6r6r7update_statisticss    z%DetectionStatistics.update_statisticsN)r.r/r0rHrrOr rPr[rr]rrrrrMrrrrrZdetection_ratergrr8r=rrrr6r6r6r7rsv r)rH django.dbrdjango.contrib.auth.modelsr django.utilsrdjango.core.validatorsrrZapps.core.modelsrrr Zapps.streams.modelsr r apps.core.validatorsr r rrrr>jsonrrSrkrr6r6r6r7s   r