`jDdZddlmZddlmZddlmZddlmZddl m Z m Z m Z m Z mZddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddZddZ ddZ!d dZ"d!dZ#d"dZ$d#dZ% d$dZ& d%dZ' d&dZ(d'dZ)d(dZ*d(dZ+y))z@Daily C59 aggregation service for Regione Liguria movement rows.) annotations) defaultdict)date)Iterable)ExpressionWrapperF IntegerFieldQSum)Coalesce)Booking)normalize_guest_for_istat)XmlPayloadValidationError)IstatC59RowPayload)IstatLookupService)Property) StructurecJ|j|cxkxr|jkScSN) check_in_datecheck_out_date)booking target_dates =/backend/istat/xml_export/services/c59_aggregation_service.pybooking_overlaps_dayrs$  K H'2H2H HH HHcPt||t|zt|zS)N)check_in_date__ltecheck_out_date__gt)r)r)r )rs rc59_relevant_booking_filterr s+ [[I + & ' ; ' (rctjjt||dj dddj dj ddS) NT) structure_id is_checked_inproperty property_type structureguestsrid)r objectsfilterr select_relatedprefetch_relatedorder_byr"rs rfetch_c59_bookingsr/!sS ' 4%   O[ A  ( # /4 ( rcF|xsdjjS)N)stripupper)values r_normalize_coder5.s KR   & & ((rct|xsdj}|jr|jd}|jrt |dk7rt d|d|d|S)Nr1zInvalid z code 'z'' in C59 aggregation (must be 3 digits))strr2isdigitzfilllenr)r4contextcodes r_three_digit_coder>2si u{  ! ! #D ||~zz!} <<>SY!^'wiwug-T U   Krct|}|s td|dk(r`t|}|s tdtj|}|td|ddt |j dd|d  fStj |}|td |dd t |j d d|d  fS)Nz'Country is required for C59 aggregationITz=Province is required for Italian residents in C59 aggregationz$Missing ISTAT province mapping for 'z' in C59 aggregationi province_codez province '')r<z#Missing ISTAT country mapping for 'e istat_codez country ')r5rr get_provincer>get get_country)countryprovince country_codeprovince_siglaprovince_mappingcountry_mappings rresolve_c59_residence_coderO=s "7+L '(QRRt(2+O .::>J  #+6~6FFZ[ %   1  02   )44\BO'1,?S T   !L)L>+ rct|}t|dd}t|tr|j dnd}t t|dd|S)N extra_datarJrI)rgetattr isinstancedictrGrO)guestnormalized_guestrQrJs r_guest_residence_keyrW]sR07)<>J-7 D-Iz~~j)tH % )T2 rc td}|D]}|j|k(}|j|k(}t||}|s|s|s4|jj D]E} t |} |r|| dxxdz cc<|r|| dxxdz cc<|s6|| dxxdz cc<G|jD cgc]1\\} } } | ds | ds| drt| | | d| d| dd 3}} } } |jd |S#t$r.} td|jd|jd| | d} ~ wwxYwcc} } } w) NcddddS)Nr)arrivipartenzepresenzer]rrz&build_daily_c59_rows..ms q&QrzGuest z (booking z): rZr[r\r)nazione residenzarZr[r\diurnic2|j|jfSr)r`ra)rows rr^z&build_daily_c59_rows..ss{{CMM:r)key) rrrrr'allrWrr(itemsrsort)r&rbookings aggregationr is_arrival is_departure is_presencerUreexcr`racountsrowss rbuild_daily_c59_rowsrqgs QRK**k9 --< *7K@ lk ^^'')E *51 C *a/*C ,1,C ,1,*@-8,=,=,? -@ ( Wi& ( vj1VJ5G (#J'J'  -@  II:I; K3- /UXXJj CuE  s! D ;6E E)D<<Ecrtjj|}t||t ||S)N)r()r&rri)rr)rGrqr/)r"rr&s r"build_daily_c59_rows_for_structurerss: !!%%%6I #L+> rctjj|d||djdj j S)NTF)r"r#rrproperty_id__isnull property_id)r r)r*valuesdistinctcountr.s rcalculate_c59_occupied_roomsrzsI %** %       rctjj|tjjj dS)Nr& availabilityr%)rr)r* Availability AVAILABLEr+)r&s r _active_properties_for_structurers>    " "**44 # n_%&rct|ddxs t|dd}|r|dkDr t|St|jS)N total_rooms total_unitsr)rRintrry)r&rs rcalculate_c59_available_roomsrsM)]D9W=$>K{Q; +I 6 < < >>rc ntjj|tjjj t ttddttddztjttdd}t|d S) uCalculate total available beds for a structure. Each PropertyType defines beds *per room/unit* (num_beds + num_sofa_beds). The total is the sum of (beds_per_room × number_of_active_rooms) across all property types that have at least one active room in this structure. Uses a single aggregated DB query to avoid N+1 issues: SUM(num_beds + num_sofa_beds) grouped per active Property row. The ``structure.total_beds`` shortcut is intentionally NOT used here because that field is not present on the Structure model and the per-room calculation is the authoritative source of truth for C59 reporting. r|property_type__num_bedsrproperty_type__num_sofa_beds) output_field) beds_per_roomr)totalr) rr)r*r~rannotaterr rr aggregater r)r&results rcalculate_c59_available_bedsrs !..88   +45q91;rOrWrqrsrzrrrr]rrrsF"#GG/#;ABG&'I )@,,, ,  ,^     &&? r