U hL@s$dZddlmZddlmZddlmZddlmZm Z ddl m Z m Z m Z ddlmZmZmZmZmZmZmZmZmZmZmZmZmZddlZddlZGd d d e e ZGd d d e e ZGd dde e Z Gddde e e Z!Gddde e Z"ddl#m$Z$ddl%m&Z&e&e$e"dddZ'dS)a Stream Models for Stream Processor Application This module contains Django models for managing streaming operations, including channel configuration, stream sessions, HLS segments, and monitoring data. These models provide the data structure for the entire stream processing workflow. )models)User)timezone)MinValueValidatorMaxValueValidator)TimestampedModel UUIDModel StatusModel) validate_hls_urlvalidate_directory_pathvalidate_name_formatvalidate_slug_formatvalidate_segment_durationvalidate_max_segmentsvalidate_retry_attemptsvalidate_video_dimensionsvalidate_video_bitratevalidate_audio_bitratevalidate_sample_ratevalidate_audio_channelsvalidate_framerateNc@seZdZdZejddegddZejdde gddZ ej de gdd Z ejdd d Zejddd d Zejdegdd ZejdegddZejdegddZejdegddZejdededgddZejeej dddZ!GdddZ"dd Z#d!d"Z$d#d$Z%d%d&Z&d'd(Z'd)S)*Channela6 Model representing a streaming channel configuration. A channel defines the source stream URL, encoding parameters, and output configuration for a specific streaming operation. Multiple stream sessions can be associated with a single channel. Attributes: name (CharField): Human-readable name for the channel slug (SlugField): URL-friendly identifier for the channel hls_url (URLField): Source HLS stream URL to capture description (TextField): Optional description of the channel is_active (BooleanField): Whether the channel is currently active output_directory (CharField): Directory path for stream output segment_duration (PositiveIntegerField): HLS segment duration in seconds max_segments (PositiveIntegerField): Maximum number of segments to keep retry_attempts (PositiveIntegerField): Number of retry attempts on failure retry_interval (PositiveIntegerField): Seconds between retry attempts created_by (ForeignKey): User who created this channel dTz#Human-readable name for the channel) max_lengthunique validators help_textz'URL-friendly identifier for the channelz Source HLS stream URL to capture)rrrz+Optional description of the channel contentblankrz(Whether this channel is currently activedefaultdb_indexrz&Directory path for stream output filesz'Duration of each HLS segment in secondsr!rr z.Maximum number of segments to keep in playlistz+Number of retry attempts on capture failurei,z&Seconds to wait between retry attemptschannelszUser who created this channel on_delete related_namerc@sBeZdZdZdZdgZejdgdejdgdejdgdgZdS) z Channel.MetarChannelsnameslugfields is_active created_atN __name__ __module__ __qualname__ verbose_nameverbose_name_pluralorderingrIndexindexesr=r=8/var/www/html/StreamProcessor/src/apps/streams/models.pyMetazs   r?cCs|jS)z%String representation of the channel.)r.selfr=r=r>__str__szChannel.__str__cCstj|j|jS)z*Get the full output path for this channel.)ospathjoinoutput_directoryr/r@r=r=r>get_output_pathszChannel.get_output_pathcCstj|dS)z)Get the HLS output path for this channel.Zhls)rCrDrErGr@r=r=r> get_hls_pathszChannel.get_hls_pathcCstj|dS)z)Get the path to the master playlist file.z playlist.m3u8)rCrDrErHr@r=r=r>get_playlist_pathszChannel.get_playlist_pathcCs|jjddS)z9Get the currently active stream session for this channel.active)status)sessionsfilterfirstr@r=r=r>get_active_sessionszChannel.get_active_sessionN)(r5r6r7__doc__r CharFieldr r. SlugFieldr r/URLFieldr Zhls_url TextField description BooleanFieldr2r rFPositiveIntegerFieldrZsegment_durationrZ max_segmentsrZretry_attemptsrrZretry_interval ForeignKeyrCASCADEZ created_byr?rBrGrHrIrOr=r=r=r>rs rc @seZdZdZejddddZejdddd Zejd d d d Zej d e de dgddZ ejdddd Z ejdddd ZejdddddgddZejdddddd d!d"d#d$d%g d&dZejdd'd(d)d*gd+dZejd d,d-d ZGd.d/d/Zd0d1Zd2S)3VideoConfigurationaM Model for video encoding configuration parameters. This model stores video encoding settings that can be applied to stream processing operations, allowing for different quality profiles and encoding options. Attributes: name (CharField): Name for this video configuration resolution (CharField): Video resolution (e.g., "1920x1080") aspect_ratio (CharField): Video aspect ratio (e.g., "16:9") frame_rate (PositiveIntegerField): Video frame rate in FPS min_bitrate (CharField): Minimum video bitrate max_bitrate (CharField): Maximum video bitrate codec (CharField): Video codec to use preset (CharField): Encoding preset for speed/quality trade-off profile (CharField): Video profile setting level (CharField): Video level setting 2Tz!Name for this video configurationrrrZ1280x720z!Video resolution (width x height)rr!rr&z16:9zVideo aspect ratior(xz%Video frame rate in frames per secondr%Z2000kz%Minimum video bitrate (e.g., '2000k')Z4000kz%Maximum video bitrate (e.g., '4000k')h264)razH.264)Zh265zH.265)Zvp9ZVP9zVideo codec to use for encodingrr!choicesrmedium)Z ultrafastz Ultra Fast)Z superfastz Super Fast)Zveryfastz Very Fast)ZfasterZFaster)fastZFast)rdZMedium)ZslowZSlow)ZslowerZSlower)Zveryslowz Very Slowz+Encoding preset for speed/quality trade-offmain)ZbaselineZBaseline)rfZMain)highZHighzVideo profile settingz3.1z(Video level setting (e.g., '3.1', '4.0')c@seZdZdZdZdgZdS)zVideoConfiguration.MetazVideo ConfigurationzVideo Configurationsr.Nr5r6r7r8r9r:r=r=r=r>r?sr?cCs|jd|jdS)z1String representation of the video configuration. ())r. resolutionr@r=r=r>rBszVideoConfiguration.__str__N)r5r6r7rPrrQr.rkZ aspect_ratiorWrrZ frame_rateZ min_bitrateZ max_bitratecodecpresetZprofilelevelr?rBr=r=r=r>rZs   rZc@seZdZdZejddddZejdddd d gd d Zejdd ddZej dddddddgddZ ej ddddgddZ ej dddZ Gd d!d!Zd"d#Zd$S)%AudioConfigurationao Model for audio encoding configuration parameters. This model stores audio encoding settings that can be applied to stream processing operations, allowing for different audio quality profiles and encoding options. Attributes: name (CharField): Name for this audio configuration codec (CharField): Audio codec to use bitrate (CharField): Audio bitrate setting sample_rate (PositiveIntegerField): Audio sample rate in Hz channels (PositiveIntegerField): Number of audio channels normalize (BooleanField): Whether to apply audio normalization r[Tz!Name for this audio configurationr\r]aac)rpZAAC)Zmp3ZMP3)ZopusZOpuszAudio codec to use for encodingrbZ128kz$Audio bitrate (e.g., '128k', '256k')r^逻)i@z8 kHz)i>z16 kHz)i"Vz 22.05 kHz)iDz44.1 kHz)rqz48 kHz)iwz96 kHzzAudio sample rate in Hz)r!rcr)r(ZMono)rrZStereo)r$z 5.1 SurroundzNumber of audio channelsz+Apply audio normalization (loudnorm filter)r!rc@seZdZdZdZdgZdS)zAudioConfiguration.MetazAudio ConfigurationzAudio Configurationsr.Nrhr=r=r=r>r?Usr?cCs|jd|jdS)z1String representation of the audio configuration.rirj)r.bitrater@r=r=r>rBZszAudioConfiguration.__str__N)r5r6r7rPrrQr.rlrtrWZ sample_rater)rV normalizer?rBr=r=r=r>ro sZ   roc@seZdZdZejeejdddZeje ej ddddZ eje ej ddddZ ejddd d Zejddd d Zejd d dZejd ddZejdddZejddddZejedddZGdddZddZddZddZdd Zd!S)" StreamSessiona Model representing an active or historical stream session. A stream session tracks a specific instance of stream capture and processing, including start/end times, configuration used, and processing statistics. Attributes: channel (ForeignKey): Channel this session belongs to video_config (ForeignKey): Video configuration used audio_config (ForeignKey): Audio configuration used started_at (DateTimeField): When the session started ended_at (DateTimeField): When the session ended (if applicable) segments_processed (PositiveIntegerField): Number of segments processed errors_count (PositiveIntegerField): Number of errors encountered last_error (TextField): Last error message encountered process_id (CharField): System process ID for monitoring statistics (JSONField): Additional statistics data rLzChannel this session belongs tor*Tz)Video configuration used for this session)r+nullrrz)Audio configuration used for this sessionzWhen the stream capture startedrwrrzWhen the stream capture endedrz)Number of segments successfully processedrsz.Number of errors encountered during processingzLast error message encounteredrr[z System process ID for monitoring)rrrz"Additional statistics and metadata)r!rrc@sDeZdZdZdZdgZejddgdejdgdejdgdgZdS) zStreamSession.MetazStream SessionzStream Sessionsz -started_atchannelrKr0 started_atNr4r=r=r=r>r?s  r?cCs|jjd|jd|jdS)z,String representation of the stream session.z - rirj)ryr.rKrzr@r=r=r>rBszStreamSession.__str__cCs"|jr|jpt}||jSdS)z&Calculate the duration of the session.N)rzended_atrnow)rAend_timer=r=r>durations zStreamSession.durationcCs|jdko|j S)z*Check if the session is currently running.)rJ processing)rKr{r@r=r=r> is_runningszStreamSession.is_runningcCs(|jd7_||_|jddgddS)z6Add an error to the session and increment error count.r( errors_count last_error update_fieldsN)rrsave)rA error_messager=r=r> add_errorszStreamSession.add_errorN) r5r6r7rPrrXrrYryrZSET_NULLZ video_configroZ audio_config DateTimeFieldrzr{rWZsegments_processedrrTrrQ process_id JSONFielddictZ statisticsr?rBr~rrr=r=r=r>rv_st  rvc@seZdZdZejeejdddZej dddZ ej dd dZ ej d d Z ejed ed gddZejddddZejdddZejddddZGdddZddZddZddZdd Zd!S)" HLSSegmenta Model representing an individual HLS segment file. This model tracks individual .ts segment files that make up the HLS stream, including their metadata, file paths, and processing status. Attributes: session (ForeignKey): Stream session this segment belongs to filename (CharField): Name of the segment file file_path (CharField): Full path to the segment file sequence_number (PositiveIntegerField): Sequence number in playlist duration (FloatField): Duration of the segment in seconds file_size (PositiveBigIntegerField): Size of the segment file in bytes processed_at (DateTimeField): When the segment was processed is_available (BooleanField): Whether the file is currently available segmentsz&Stream session this segment belongs tor*r#zName of the segment file)rrrzFull path to the segment filez#Sequence number in the HLS playlist)rg?gN@z"Duration of the segment in seconds)rrTz!Size of the segment file in bytesrxzWhen the segment was processed) auto_now_addrz/Whether the segment file is currently availabler c@sNeZdZdZdZddgZddgZejddgdejdgdejdgdgZ dS) zHLSSegment.Metaz HLS Segmentz HLS Segmentssessionsequence_numberr0 processed_at is_availableN) r5r6r7r8r9r:unique_togetherrr;r<r=r=r=r>r?s  r?cCs|jjjd|jS)z)String representation of the HLS segment.z - Segment )rryr.rr@r=r=r>rB$szHLSSegment.__str__cCstj|jS)z2Check if the segment file actually exists on disk.)rCrDexists file_pathr@r=r=r> file_exists(szHLSSegment.file_existscCs,ztj|jWStk r&YdSXdS)z#Get the actual file size from disk.N)rCrDgetsizerOSErrorr@r=r=r> get_file_size,szHLSSegment.get_file_sizecCsRz8tj|jr6t|jd|_|jdgdWdSWntk rLYnXdS)z"Delete the segment file from disk.FrrT)rCrDrrremoverrrr@r=r=r> delete_file3s  zHLSSegment.delete_fileN)r5r6r7rPrrXrvrYrrQfilenamerrWr FloatFieldrrr~PositiveBigIntegerField file_sizerrrVrr?rBrrrr=r=r=r>rsR r) post_save)receiver)sendercKs*|r&|jr&ddlm}|t|jdS)a. Automatically trigger jingle detection when a new HLS segment is created. Args: sender: The model class (HLSSegment) instance: The actual instance being saved created: Boolean indicating if this is a new instance **kwargs: Additional keyword arguments r)process_new_segmentN)rZapps.streams.tasksrdelaystrid)rinstancecreatedkwargsrr=r=r>trigger_jingle_detectionDs  r)(rP django.dbrdjango.contrib.auth.modelsr django.utilsrZdjango.core.validatorsrrZapps.core.modelsrrr Zapps.core.validatorsr r r r rrrrrrrrrrCjsonrrZrorvrZdjango.db.models.signalsrdjango.dispatchrrr=r=r=r>s"   <tRxj