`j(dZddlmZddlZddlmZddlmZmZddl m Z ddl m Z ddl mZdd lmZdd lmZmZmZdd lmZdd lmZdd lmZmZddlmZej<eZ ddZ!ddZ" ddZ# ddZ$y)u Build Ross1000 daily movement payloads for a date range. Business rules implemented here: - One movement block per calendar day (including empty days) - Arrivals = guests whose check_in_date == target_date - Departures = guests whose check_out_date == target_date - Overnight presence (check_in < day < check_out) is NOT a separate block in Ross1000 (unlike C59 presenze) — only arrivals and departures are reported - struttura block is always present, even on empty days - idswh is read from GuestStay — never generated here Performance: - Structure metadata fetched once - All bookings for the full range fetched in a single query - GuestStay records fetched in a single query and indexed by (booking_id, guest_id) - Per-day filtering done in Python — zero N+1 queries ) annotationsN) defaultdict)date timedelta)Any)Booking) GuestStay)Ross1000ValidationError)Ross1000GuestPayloadRoss1000MovementDayPayloadRoss1000StrutturaPayload)build_ross1000_guest_payload)build_struttura_payload)validate_date_rangevalidate_structure) Structurecttjj|d||j dddj dj ddS) aHFetch all bookings that have any activity in [start_date, end_date]. Includes: - Bookings with check_in_date in range (arrivals) - Bookings with check_out_date in range (departures) - Bookings that span the entire range (overnight presence) Uses select_related + prefetch_related to avoid N+1 queries. T) structure_id is_checked_incheck_in_date__ltecheck_out_date__gteproperty property_type structureguests check_in_dateid)listrobjectsfilterselect_relatedprefetch_relatedorder_by)r start_dateend_dates 4/backend/istat/ross1000/services/movement_builder.py_fetch_bookings_for_ranger'+s[ %' *   O[ A  ( # /4 (  c|siStjj|djddd}|Dcic]}|d|df|dc}Scc}w)uFetch all GuestStay records for the given bookings. Returns a dict keyed by (booking_id, guest_id) → idswh. Only records with a non-null idswh are included. F)booking_id__in idswh__isnull booking_idguest_ididswh)r rr values) booking_idsstaysstays r&_fetch_idswh_indexr3Fs     $ $" %  f\:w/  D l T*-.W =  sAc ,g}|D]}|r|j|k7r|s|j|k7r't|jj }|D]}}t |dd}||j |j|f} | s"tjd||jT t||j| } |j| t|S#t$r,} tjd||j| Yd} ~ d} ~ wwxYw)z?Build guest payloads for arrivals or departures on a given day.rNu>Ross1000: missing idswh for guest %s / booking %s — skipping)guestr,r.uFRoss1000: guest validation failed — skipping guest %s booking %s: %s)rcheck_out_daterrallgetattrgetrloggerwarningrappendr tuple) bookings target_date idswh_index is_arrivalpayloadsbookingrr5r-r.payloadexcs r&_build_day_guestsrFZs,.H '//;> g44 C gnn((*+EudD1HOOWZZ$:;ETJJ  6&zz (/T ?+ \JJ  s&)C D'"DDct|t||t|j||}|Dcgc]}|j}}t |}t |}g}||kr|Dcgc]"}|j k(s|jk(r|$} }t| |d} t| |d} tfd|D} t|j|j| |j|j} |jt!| | | t#d z |krt$j'd t)||j|||Scc}wcc}w) uhBuild one Ross1000MovementDayPayload per calendar day in [start_date, end_date]. Empty days (no arrivals, no departures) are included with empty tuples and the struttura block still present — as required by the spec. Args: structure: Structure model instance (already validated) start_date: First day of the export range (inclusive) end_date: Last day of the export range (inclusive) Returns: List of movement day payloads, one per calendar day, sorted by date. Raises: Ross1000ValidationError: On invalid inputs or missing structure data. )rT)rAFc3~K|]4}|jcxkr|jkrnn|jd6yw)N)rr6 property_id).0b current_days r& z&build_movement_days..s9 #!+@0@0@@ ) |s:=)codiceaperturacamere_occupatecamere_disponibililetti_disponibili)r strutturaarrivipartenzerI)daysu=Ross1000: built %d movement days for structure %s (%s → %s))rrr'rr3rrr6rFsumr rOrPrRrSr<r rr:infolen)rr$r% all_bookingsrLr0r@struttura_baserW day_bookingsrUrVoccupiedrTrMs @r&build_movement_daysr_s,y! H--Y\\:xPL"..A144K.$[1K-yAN-/DK  !$ #!+-1A1A[1P |  # +{t % +{u   #   -!((#,,$-@@,>>   & #!    ya(( M  !P KKG D    Kq/ s E!-'E&)rintr$rr%rreturn list[Booking])r0z list[int]radict[tuple[int, int], str]) r>rbr?rr@rcrAboolraz tuple[Ross1000GuestPayload, ...])rrr$rr%rraz list[Ross1000MovementDayPayload])%__doc__ __future__rlogging collectionsrdatetimerrtypingrbookings.modelsr guests.modelsr istat.ross1000.exceptionsr &istat.ross1000.models.movement_payloadr r r &istat.ross1000.services.guest_resolverr*istat.ross1000.services.structure_resolverr+istat.ross1000.validators.payload_validatorrrstructures.modelsr getLogger__name__r:r'r3rFr_r(r&rvs$##$##= PN_'   8 $ 6(444,4  4 & 4nUUU U & Ur(