U Œæ‹hž½ã@s¦ddlZddlZddlZddlmZddlmmZddl m Z m Z ddl m Z ddl mZddlmZe e¡Zdd„Zd d „Zd d „Zd d„ZGdd„dƒZdS)éN)Úminidom)ÚdatetimeÚ timedelta)Úsettings)Útimezone)Ú ContentFilec Cs®|sdSzbddddg}|D]:}zt | ¡|¡ ¡WWStk rRYqYqXqt d|›¡WdStk r¨}z t d|›d|›¡WY¢dSd}~XYnXdS) a3 Helper function to convert time strings to time objects. Useful for migrating existing data from CharField to TimeField. Args: time_string (str): Time in string format (e.g., "14:30:00", "2:30 PM") Returns: time: Python time object or None if conversion fails Nz%H:%M:%Sz%H:%Mz %I:%M:%S %pz%I:%M %pzCould not parse time string: zError converting time string 'ú': ) rÚstrptimeÚstripÚtimeÚ ValueErrorÚloggerÚwarningÚ ExceptionÚerror)Z time_stringÚformatsÚfmtÚe©rú//var/www/html/Focus/src/apps/playlists/utils.pyÚconvert_time_string_to_times$ ü rc Csæ|sdSzšd|krt| d¡}t|ƒdkrJtt|ƒ\}}}tj|||dWSt|ƒdkrttt|ƒ\}}tj||dWS| ¡rŽtjt|ƒdWSt d|›¡WdSt k rà}z t  d |›d |›¡WY¢dSd}~XYnXdS) aU Helper function to convert duration strings to timedelta objects. Useful for migrating existing data from CharField to DurationField. Args: duration_string (str): Duration in string format (e.g., "00:30:00", "30 minutes") Returns: timedelta: Python timedelta object or None if conversion fails Nú:é)ÚhoursÚminutesÚsecondsé)rr)rz!Could not parse duration string: z"Error converting duration string 'r) ÚsplitÚlenÚmapÚintrrÚisdigitr rrr)Úduration_stringÚpartsrrrrrrrÚ#convert_duration_string_to_duration5s$    r$c Cs²|sdSzfddddddg}|D]:}zt | ¡|¡ ¡WWStk rVYqYqXqt d|›¡WdStk r¬}z t d |›d |›¡WY¢dSd}~XYnXdS) a8 Helper function to convert date strings to date objects. Useful for migrating existing data from CharField to DateField. Args: date_string (str): Date in string format (e.g., "2023-12-25", "25/12/2023") Returns: date: Python date object or None if conversion fails Nú%Y-%m-%dz%d/%m/%Yz%m/%d/%Yz%Y/%m/%dz%d-%m-%Yz%m-%d-%YzCould not parse date string: zError converting date string 'r) rr r Údater r rrr)Z date_stringrrrrrrÚconvert_date_string_to_dateZs( ú  r'cCs†|jr|j d¡nd}|jr$|jjnd}|jr6|jjnd}d dd„|Dƒ¡ ¡}d dd„|Dƒ¡ ¡}d |›d |›d |›d |›S) z Generate upload path for playlist XML files. Args: instance: Playlist model instance filename: Original filename Returns: str: Upload path z%Y_%mZno_dateZ no_channelZ no_regionÚcss"|]}| ¡s|dkr|VqdS©)ú-Ú_N©Úisalnum©Ú.0ÚcrrrÚ ‘sz+playlist_xml_upload_path..css"|]}| ¡s|dkr|VqdSr)r,r.rrrr1’sz playlists/ú/)Úbroadcast_dateÚstrftimeÚchannelÚ channel_nameÚ zone_channelÚregionÚjoinr )ÚinstanceÚfilenameZ date_pathr6Zchannel_regionrrrÚplaylist_xml_upload_paths r<c@sˆeZdZdZddd„Zddd„Zdd„Zd d „Zd d „Zd d„Z dd„Z dd„Z dd„Z dd„Z edd„ƒZdd„Zdd„Zdd„ZdS) ÚPlaylistXMLGeneratora; Comprehensive XML generation and management system for broadcast playlist schedules. This class provides enterprise-grade XML generation capabilities for converting playlist database records into industry-standard SCTE-118-3 compliant XML schedules. It handles the complete transformation from hierarchical playlist data structures to broadcast-ready XML documents suitable for automation systems. The generator supports the full playlist hierarchy including: - Playlist metadata and scheduling information - Time windows with precise timing specifications - Advertisement availability slots (avails) within windows - Individual advertisement records with traffic management Key features include: - SCTE-118-3 schema compliance for broadcast industry standards - Comprehensive error handling and validation - File system integration with automatic storage management - Detailed logging for audit trails and debugging - Traffic ID management for advertisement tracking - Flexible date formatting and timezone handling Business Applications: - Broadcast automation system integration - Compliance reporting and regulatory requirements - Advertisement inventory management and billing - Schedule distribution to multiple broadcast facilities - Quality assurance and validation workflows Technical Standards: - SCTE-118-3 XML schema compliance - ISO 8601 date/time formatting - UTF-8 encoding for international character support - Structured logging for operational monitoring Example Usage: generator = PlaylistXMLGenerator() result = generator.generate_schedule(playlist_instance) if result['status']: xml_file_path = result['file'] print(f"XML generated: {xml_file_path}") NcCs ||_t |jj¡|_d|_dS)a Initialize the XML generator with optional playlist context and logging infrastructure. Sets up the generator instance with logging capabilities, traffic management, and optional playlist binding for streamlined processing workflows. Args: playlist (Playlists, optional): Playlist model instance to bind to this generator Enables streamlined processing for repeated operations Can be overridden in individual method calls Attributes: playlist (Playlists): Bound playlist instance for default operations logger (Logger): Configured logger for operation tracking and debugging traffic_id (int): Global traffic counter for advertisement identification rN)ÚplaylistÚloggingÚ getLoggerÚ __class__Ú__name__r Ú traffic_id)Úselfr>rrrÚ__init__ÄszPlaylistXMLGenerator.__init__c CsÞ|p|j}|s| d¡Szb| |¡}| |¡}|jj|t| d¡ƒdd|j  d|j ›¡dd|jrp|jj nd|dœWSt k rØ}z@|j  d t|d d ƒ›d t|ƒ›¡| d t|ƒ›¡WY¢Sd}~XYnXdS)a1 Generate comprehensive XML schedule content for broadcast automation systems. This method orchestrates the complete XML generation workflow including structure creation, content population, file system storage, and validation. It produces SCTE-118-3 compliant XML documents suitable for broadcast automation systems and advertisement management platforms. The generation process handles the complete playlist hierarchy transformation: 1. Playlist metadata and scheduling parameters 2. Time window definitions with precise timing 3. Advertisement availability slots within windows 4. Individual advertisement records with traffic management The method provides comprehensive error handling and detailed logging to support production deployment and troubleshooting workflows. Args: playlist (Playlists, optional): Playlist model instance to process Uses bound instance if not provided Must contain complete hierarchy relationships Returns: dict: Comprehensive generation result containing: - status (bool): Success/failure indicator - message (str): Descriptive result message - file (str): Full file system path to generated XML - filename (str): Generated filename for the XML document - xml_content (str): Complete XML content for immediate use Raises: AttributeError: If playlist lacks required model relationships FileSystemError: If XML file cannot be saved to storage ValidationError: If generated XML fails schema validation DatabaseError: If playlist data cannot be retrieved Example Usage: result = generator.generate_schedule(my_playlist) if result['status']: deploy_to_broadcast_system(result['file']) else: handle_generation_error(result['message']) zNo playlist providedzutf-8T)Úsavez1Successfully generated XML schedule for playlist z#XML schedule generated successfullyN©ÚstatusÚmessageÚfiler;z-Failed to generate XML schedule for playlist ÚidÚunknownú: z Exception: )r>Ú_error_responseÚ_create_playlist_xmlÚ_generate_xml_filenameÚxml_filerFrÚencoder ÚinforKÚpathrrÚgetattrÚstr)rDr>Ztarget_playlistÚ xml_contentr;rrrrÚgenerate_scheduleís(2     ý ü$z&PlaylistXMLGenerator.generate_schedulec Csˆ|jr|jjnd}d dd„|Dƒ¡ ¡}|jj}|jj}|jrL|j d¡nd}t |j ƒ}|j rddnd}|›d|›d|›d|›|›d S) a Generate descriptive filename for XML schedule files based on playlist metadata. Creates standardized filenames that include essential playlist identification information for file management, organization, and automated processing systems. The filename format supports sorting, filtering, and identification of specific playlist versions and configurations. Args: playlist (Playlists): Playlist model instance containing metadata Must have channel, zone, and date relationships Returns: str: Generated filename in format: YYYYMMDD-zone-network-version[_draft].sch Examples: "20250730-EST-NBC-1.sch", "20250731-PST-CBS-2_draft.sch" Filename Components: - Date: YYYYMMDD format for chronological sorting - Zone: Geographic or programming zone identifier - Network: Broadcast network identifier - Version: Playlist version number for change tracking - Draft suffix: Added for draft playlists to prevent confusion - Extension: .sch for schedule file identification rLr(css"|]}| ¡s|dkr|VqdSr)r,r.rrrr1ƒsz>PlaylistXMLGenerator._generate_xml_filename..ú%Y%m%dZnodateZ_draftr*z.sch) r5r6r9r Úid_zone_channelÚzonenameÚ networknamer3r4rVÚversionÚis_draft)rDr>r6Ú zone_nameÚ network_nameZdate_strZ version_strZ draft_strrrrrP`s z+PlaylistXMLGenerator._generate_xml_filenamecCsÞd|_| |j¡}t d¡}| dt|ƒ¡| dt|ƒ›d¡| dt|ƒ›d¡| dt|jjƒ¡| d t|jj ƒ¡| d t|j ƒ¡| d d ¡| d d¡| dd¡|  ||¡t j |dd}| |¡}|S)a± Create comprehensive XML document structure conforming to SCTE-118-3 broadcast standards. This method orchestrates the complete transformation of playlist database records into industry-standard XML format suitable for broadcast automation systems. It handles the hierarchical structure creation, attribute population, and formatting to ensure compliance with broadcast industry standards. The XML structure includes: - Root Schedule element with broadcast metadata - Window elements representing time slots - Avail elements for advertisement opportunities - Ad elements with traffic management information Args: playlist (Playlists): Complete playlist model instance with relationships Must include windows, avails, and ads hierarchy Returns: str: Formatted XML document string ready for broadcast system consumption Includes proper indentation and encoding for human readability XML Schema Compliance: - SCTE-118-3 standard for broadcast scheduling - ISO 8601 date/time formatting - Proper namespace declarations - Required attribute specifications rZScheduleÚ broadcastDateZ begDateTimezT00:01:00+00:00Z endDateTimezT23:59:59+00:00Ú networkNameÚzoneNameÚrevisionÚlevelÚ0Zxmlnsz&http://www.scte.org/schemas/118-3/201XZ schemaVersionz http://www.w3.org/2001/XMLSchemaÚunicode)Úencoding)rCÚ_format_broadcast_dater3ÚxmlZElementÚsetrVr7r\r[r]Ú _add_windowsÚETÚtostringÚ _format_xml)rDr>r3ÚrootÚ xml_stringZ formatted_xmlrrrrO¨s #       z)PlaylistXMLGenerator._create_playlist_xmlcCsd|j ¡ d¡}|D]J}t |d¡}| dt|jƒ dd¡›d¡| d|j ¡|  ||¡qdS) a Add window elements and their nested content to the XML structure. This method processes all time windows associated with the playlist and creates corresponding XML elements with proper timing attributes and nested content. Windows represent discrete time periods within the broadcast schedule where advertisements can be scheduled. Args: windows_element (xml.Element): Parent XML element to contain window elements Typically the root Schedule element playlist (Playlists): Playlist instance containing window relationships Must have accessible windows queryset Processing Flow: 1. Query all windows ordered by start time 2. Create XML element for each window 3. Set timing attributes (start time, duration) 4. Add nested avails for each window Ú window_startÚWindowZ windowStartú ÚTú+00:00ZwindowDurationN) ÚwindowsÚallÚorder_byrmÚ SubElementrkrVrrÚreplaceÚwindow_durationÚ _add_avails)rDZwindows_elementr>rwÚwindowZwindow_elementrrrrlûs   z!PlaylistXMLGenerator._add_windowscCst|j ¡ d¡}|D]Z}t |d¡}| dd¡| dt|jƒ¡| dt|jƒ  dd¡›d ¡|  ||¡qd S) a) Add advertisement availability elements and their nested advertisements to XML. This method processes all avails (advertisement slots) within a window and creates corresponding XML elements with proper positioning and timing attributes. Avails represent specific opportunities for advertisement placement within broadcast windows. Args: avails_element (xml.Element): Parent window element to contain avail elements Receives avail children for hierarchical structure window (Windows): Window instance containing avail relationships Must have accessible avails queryset Processing Flow: 1. Query all avails ordered by position within window 2. Create XML element for each avail 3. Set positioning and timing attributes 4. Add nested advertisements for each avail Ú availinwindowÚAvailZavailNumrfÚ availInWindowZ availStartrtrurvN) ÚavailsrxryrmrzrkrVrÚ avail_startr{Ú_add_ads)rDZavails_elementr~r‚ÚavailÚ avail_elementrrrr}4s   z PlaylistXMLGenerator._add_availsc CsÖ|j ¡ d¡}|D]¼}|jd7_t |d¡}| d|jj›d¡| dd¡| dt   d t   t |jjƒ¡¡¡| d t |jƒ¡| d d ¡| d |jjjj¡| d|j|jkrÄt |jƒnt |jƒ¡qdS)NÚpositioninavailéZSpotZadIdÚsecZ eventTypeZLOIÚlengthz%H%M%S00ÚpositionInAvailZ schedSourceÚLocalÚspotIdÚ trafficId)Úadspots_in_availrxryrCrmrzrkÚadspotÚdurationr r4Úgmtimer rVr‡Ú original_filerJÚnameÚ trafficid)rDr†r…Z ad_placementsZ adplacementZ ad_elementrrrr„ps  " zPlaylistXMLGenerator._add_adscCs@t|tƒr tt |d¡ d¡ƒSt|dƒr4| d¡St|ƒSdS)a„ Format broadcast date to standardized YYYYMMDD string for XML attributes. This method handles multiple input date formats and converts them to the standardized 8-digit date format required by broadcast systems and XML schemas. It provides robust date handling for various input types including strings, datetime objects, and other date representations. Args: broadcast_date (str|datetime|object): Broadcast date in various formats Supports string dates, datetime objects, and other date-like objects Returns: str: Formatted date string in YYYYMMDD format Examples: "20250730", "20251215", "20240101" Supported Input Formats: - String dates in YYYY-MM-DD format - Python datetime objects - Other objects with strftime method - Fallback string conversion for unknown types r%rYr4N)Ú isinstancerVrr r4Úhasattr)rDr3rrrri€s    z+PlaylistXMLGenerator._format_broadcast_datec Cs~z8t |¡}|jddd}dd„| d¡Dƒ}d |¡WStk rx}z"|j dt|ƒ›¡|WY¢Sd}~XYnXdS)a0 Format XML string with proper indentation and cleanup for human readability. This method transforms raw XML strings into properly formatted documents with consistent indentation, line breaks, and whitespace management. It enhances readability for debugging, validation, and manual review while maintaining XML validity and compliance. Args: xml_string (str): Raw XML string without formatting Generated from ElementTree or similar XML creation tools Returns: str: Formatted XML string with proper indentation and structure Includes consistent spacing and line breaks for readability Formatting Features: - Two-space indentation for nested elements - Removal of empty lines for clean output - Proper line breaks between elements - UTF-8 encoding compatibility z N)ÚindentrhcSsg|]}| ¡r|‘qSr)r )r/ÚlinerrrÚ Üsz4PlaylistXMLGenerator._format_xml..Ú zFailed to format XML: ) rZ parseStringZ toprettyxmlrr9rr rrV)rDrqÚdomÚ formattedÚlinesrrrrro³s  z PlaylistXMLGenerator._format_xmlcCsddt|ƒ›dddœS)a Create standardized error response structure for consistent error handling. This method generates uniform error response dictionaries that match the success response format, enabling consistent error handling across all XML generation operations. It provides structured error information suitable for logging, user feedback, and automated error processing. Args: message (str): Descriptive error message explaining the failure Should provide sufficient context for debugging Returns: dict: Standardized error response containing: - status (bool): False to indicate failure - message (str): Formatted error message with context - file (None): No file path for failed operations - filename (None): No filename for failed operations Error Response Structure: Matches success response format for consistent handling Enables uniform error processing across different failure scenarios Fz!Failed to generate XML schedule: NrG)rV)rDrIrrrrNís  üz$PlaylistXMLGenerator._error_responsec Cs´z,ddlm}|j dd¡j|d}t|ƒWS|jk rXdd|›dd d d œYStk r®}z:t  d |›d t |ƒ›¡dd t |ƒ›d d d œWY¢Sd }~XYnXd S)a¶ Regenerate XML for an existing playlist using static method interface. This static method provides a convenient interface for regenerating XML schedules for existing playlists without requiring generator instantiation. It handles playlist retrieval, generator creation, and XML generation with comprehensive error handling for production deployment scenarios. Args: playlist_id (int): Database ID of the playlist to regenerate Must correspond to existing Playlists record Returns: dict: Generation result from generateSchedule method Contains status, message, file path, and XML content Use Cases: - Batch XML regeneration for schedule updates - API endpoints for on-demand XML generation - Maintenance scripts for system updates - Error recovery for failed initial generation Note: Method references 'generateSchedule' function that should be 'generate_schedule' r©Ú Playlistsr5r7©rKFúPlaylist with ID ú not foundNrGú&Failed to regenerate XML for playlist rMúFailed to regenerate XML: © Zapps.playlists.modelsr ÚobjectsÚselect_relatedÚgetZgenerateScheduleÚ DoesNotExistrr rrV©Ú playlist_idr r>rrrrÚregenerate_playlist_xmls"   ü  üz,PlaylistXMLGenerator.regenerate_playlist_xmlc Cs´z,ddlm}|j dd¡j|d}t|ƒWS|jk rXdd|›dd d d œYStk r®}z:t  d |›d t |ƒ›¡dd t |ƒ›d d d œWY¢Sd }~XYnXd S)zã Regenerate XML for an existing playlist. Args: playlist_id: ID of the playlist to regenerate Returns: dict: Result dictionary from generateSchedule rrŸr5r7r¡Fr¢r£NrGr¤rMr¥r¦r«rrrr­Ws"    ü  üc CsXzt |¡dddœWStjk rR}zddt|ƒ›dœWY¢Sd}~XYnXdS)zº Validate the generated XML content. Args: xml_content: XML content string Returns: dict: Validation result Tz XML is valid)ÚvalidrIFz Invalid XML: N)rmÚ fromstringZ ParseErrorrV)rWrrrrÚvalidate_playlist_xmlws   z*PlaylistXMLGenerator.validate_playlist_xmlc Cs¨z\|j ¡}tdd„|j ¡Dƒƒ}tdd„|j ¡Dƒƒ}|j|j|j||||j|jdœWSt k r¢}z(t   d|j›dt |ƒ›¡iWY¢Sd}~XYnXdS)zÇ Get a summary of the playlist XML structure. Args: playlist: Playlist model instance Returns: dict: Summary information css|]}|j ¡VqdS©N)r‚Úcount)r/r~rrrr1“sz@PlaylistXMLGenerator.get_playlist_xml_summary..css(|] }|j ¡D]}|j ¡VqqdSr±)r‚rxÚadspotsr²)r/r~r…rrrr1”s þ)r¬r]r^Ú windows_countÚ avails_countÚ ads_countr3Úduration_minutesz#Failed to get playlist summary for rMN) rwr²ÚsumrxrKr]r^r3r·rr rrV)r>r´rµr¶rrrrÚget_playlist_xml_summary‡s$  þø z-PlaylistXMLGenerator.get_playlist_xml_summary)N)N)rBÚ __module__Ú __qualname__Ú__doc__rErXrPrOrlr}r„rirorNÚ staticmethodr­r°r¹rrrrr=˜s + ) sHS9<3:$ E r=)Úosr r?Zxml.domrZxml.etree.ElementTreeZetreeZ ElementTreermrrÚ django.confrÚ django.utilsrÚdjango.core.files.baserr@rBr rr$r'r<r=rrrrÚs     %%'