`j0dZddlmZddlZddlmZmZddlmZddl m Z ddl m Z ddl mZdd lmZdd lmZmZmZmZdd lmZdd lmZdd lmZddlmZddlmZm Z m!Z!m"Z"m#Z#ddl$m%Z%ddl&m'Z'm(Z(ddl)m*Z*ejVe,Z-ddZ. ddZ/ddZ0ddZ1 ddZ2 d dZ3 d!dZ4 d"dZ5y)#z=Orchestration service for isolated ISTAT XML file generation.) annotationsN)date timedelta)Booking),build_c59_aggregation_payloads_for_structure)build_guest_payload)XmlPayloadValidationError)build_xml_export_result)build_c59_daily_xml_filenamebuild_c59_xml_filenamebuild_c59_zip_filenamebuild_guest_xml_filename)build_c59_zip_result)IstatXmlExportResult)IstatZipExportResult)build_c59_xml_document)build_daily_c59_rowscalculate_c59_available_bedscalculate_c59_available_roomscalculate_c59_occupied_roomsfetch_c59_bookings)serialize_payloads)generate_guest_nightswith_guest_night_prefetch) Structurecb|t|dt|tst|d|S)Nz is requiredz must be a date instance)r isinstancer)value field_names 8/backend/istat/xml_export/services/xml_export_service.py_validate_dater!+s; }':,l(CDD eT "':,6N(OPP LcZt|d}t|d}||kr td||fS)N start_dateend_datez4end_date must be greater than or equal to start_date)r!r )r$r% valid_start valid_ends r _validate_date_ranger(3sA!\:Kx4I;' B    !!r"c|tjj|j}| t d|S)N)idzStructure not found)robjectsfilterfirstr ) structure_id structures r _get_structure_or_raiser0@s<!!((L(9??AI'(=>> r"ct|jxsdj}|std|jd|S)Nz Structure z is missing ISTAT structure code)str istat_codestripr r*)r/r4s r "_get_structure_istat_code_or_raiser6GsIY))/R0668J '&F G   r"czttjj|d||j ddS)aReturn checked-in bookings that overlap the given date range. Only bookings with ``is_checked_in=True`` are included so that the XML export dataset exactly matches the ISTAT page dataset, which also filters to checked-in reservations only. T)r. is_checked_incheck_in_date__ltecheck_out_date__gt check_in_dater*)rrr+r,order_byr.r$r%s r _fetch_overlapping_bookingsr>PsB %%')   h%&r"ct|}t||\}}|tdz}tt |||}t |||}|D cgc]} | j s| } } |D cic]} | j| c} |D cic]:} | jjD]} | j | j| <c} } tfd| Dd} | td| jd| D cgc]*} t| j| j| ,}} t|}t!|||}t#|| }t$j'd |t)|j*xsd j-xsdd |j/|j/t1||dddd |Scc} wcc} wcc} } wcc} w)zDGenerate a guest movement XML export for a structure and date range.daysr=) period_startperiod_end_exclusiveNc3rK|].}|j|jvs|jvr|0yw)N)guest_id booking_id).0 guest_nightbookings_by_id guests_by_ids r z,generate_guest_xml_export..sA 3 ##+##<7%%^; 3s47z$Guest export failed because booking z& occupancy exceeds saved guest records)bookingguestrIfilenamecontentzGenerated guest XML exportr2 guest_xmlFsuccess) r.structure_istat_code export_typer$r% payload_countrPxsd_validation_enabledxsd_pathstatusextra)r0r(rlistr>ris_arrival_nightr*guestsallnextr rGrrFrrr loggerinfor3r4r5 isoformatlen)r.r$r%r/r&r'rDbookings guest_nightsrIarrival_guest_nightsrMrNmissing_guest_nightpayloads xml_contentrPresultrJrKs @@r generate_guest_xml_exportrlfs<( 5I1*hGK$ya'88#%" H) 1L (4'3 {7S7S |:BBggjj')BN G^^'')E 88  %) L  3   &' 2"--..T V  0  0K ";#9#9:{334# 0  %X.K'!H %h LF KK$(I((.B/557?4&%//1!++- ] &+  " MwC,sG'G1G ?G/Gc ft|}t|}t|d}t|||}t |||t |t |t||}t||}t||}tjd||d|jt||d d d d |S)z3Generate a C59 XML export for a single report date. report_dater=r. target_datestructure_codernrowsavailable_roomsavailable_bedsoccupied_rooms)r. export_daterOzGenerated C59 XML exportc59_xmlFNrS) r.rTrUrn row_countrPrWrXrYrZ)r0r6r!rrrrrr r rarbrcrd) r.rnr/rTvalid_report_datersrjrPrks r generate_c59_xml_exportr{s( 5I=iH&{MB 7!$" D )+% 5i@3I>3%) K&!%H%h LF KK"($8$,668T &+    Mr"c @t||\}}t|}t|}t|}t |}t t jj|d||jdddjdjdd} g} |} | |kr| D cgc]"} | j| kr| j| k\r| $} } t|| | }t|| }t!|| |||| }t#| }| j%||f| t'd z } | |krt)||}t+|| }t,j/d||d|j1|j1|j2||j4dd |Scc} w)u~Generate a ZIP archive containing one C59 XML file per calendar day. The Liguria / Ross1000 specification requires one XML file per day. Days with no guest activity still produce a valid (empty) XML document. Performance notes: - Structure metadata (istat_code, room/bed counts) is fetched once. - All bookings that overlap the full date range are fetched in a single query and held in memory; per-day aggregation is done in Python to avoid N+1 database round-trips. - Occupied-room counts still require one lightweight DB query per day (distinct property_id count) — this is unavoidable without a more complex pre-aggregation step, but each query is O(1) in rows returned. Args: structure_id: ID of the Structure to report on. start_date: First day of the reporting period (inclusive). end_date: Last day of the reporting period (inclusive). Returns: An :class:`IstatZipExportResult` with the raw ZIP bytes ready for an HTTP download response. Raises: XmlPayloadValidationError: On invalid inputs or missing structure data. T)r.r8r9check_out_date__gteproperty property_typer/r^r;r*)r/rprerorq)rnr@rA)r$r%) zip_filename daily_fileszGenerated C59 ZIP exportc59_ziprS) r.rTrUr$r% file_countr zip_byte_sizerYrZ)r(r0r6rrr\rr+r,select_relatedprefetch_relatedr<r;check_out_daterrrr appendrr rrarbrcr byte_size)r.r$r%r&r'r/rTrtru all_bookingsr current_dayb day_bookingsrsrvrjdaily_filenamerrks r generate_c59_zip_exportrs@2*hGK( 5I=iH4I>O1)rlr{rr"r rs=C"$#JAG DFJR?(   8 $ " " " "&&& &,SSS S  Sl... .bttt t  tr"