jdZddlmZddlmZddlmZmZmZm Z ddl m Z ddl m Z ddlmZddlmZmZmZmZdd lmZmZGd d Zy ) a Upcoming events service for Aimantis dashboard. Generates chronological list of upcoming check-ins and check-outs with optimized queries and proper prefetching. ORM Optimization Strategy: - NO select_related() for property/property_type because we only access booking fields (platform, length_of_stay, total_price, id, check_in_date, check_out_date) - prefetch_related() is used for guests collection to avoid N+1 queries - .only() explicitly selects minimal fields needed for event display )date)chain)DictAnyListOptional)Prefetch)Booking)Guest)DEFAULT_UPCOMING_EVENTS_LIMITMAX_UPCOMING_EVENTS_LIMITEVENT_TYPE_CHECK_INEVENT_TYPE_CHECK_OUT) get_todaynormalize_channel_namec eZdZdZ ddeedeefdZdeee ffdZ de de e fd Zd e defd Zd e d ed e deee ffdZde eee ffdZy)UpcomingEventsServicez Service for generating upcoming dashboard events. Properly prefetches main guests to avoid N+1 queries and normalizes channel names for display. N structure_idlimitcj||_t|_t|xstt |_y)z Initialize the upcoming events service. Args: structure_id: Optional structure ID for multi-tenant filtering limit: Maximum number of events to return (default: 5) N)rrtodayminr r r)selfrrs 6/backend/dashboard/services/upcoming_events_service.py__init__zUpcomingEventsService.__init__(s*)[ ?"?AZ[ returnc:|jrd|jiSiS)z&Get structure filter dict for queries.r)r)rs r_get_structure_filterz+UpcomingEventsService._get_structure_filter8s 6:6G6G 1 12OROr start_datec 0tdtjjdd}t t jjdd|i|j j|jddd d d d d jdS)a9 Fetch bookings with prefetched main guests. Uses Prefetch to only load the main guest for each booking, avoiding N+1 queries in the event generation loop. For check-in events: - Only include future check-ins (check_in_date >= today) - Exclude already checked-in bookings from today ORM Optimization: - NO select_related(): We only access booking.platform, booking.length_of_stay, booking.total_price, booking.id, booking.check_in_date, booking.check_out_date. We do NOT access booking.property or booking.property_type objects. - prefetch_related('guests'): Required because we iterate booking.main_guest_list to get guest names. Without prefetch, this would cause N+1 queries. - .only(): Selects only fields needed for event display to minimize data transfer. Args: start_date: Fetch bookings from this date onwards Returns: List of Booking objects with prefetched guests guestsT) is_main_guestmain_guest_list)querysetto_attrcheck_in_date__gteid check_in_datecheck_out_datelength_of_stay total_priceplatform is_checked_in) r r objectsfilterlistr rprefetch_relatedonlyorder_by)rr main_guest_prefetchs r_get_bookings_with_guestsz/UpcomingEventsService._get_bookings_with_guests<s2' ]])))=%  OO " " #- ,,.   1 2 T  Xo &!  rbookingc@t|dg}|r|djSy)u Extract main guest name from booking. Uses the prefetched main_guest_list attribute. Args: booking: Booking object Returns: Main guest full name or "—" if not found r$ru—)getattr full_name)rr8 main_guestss r_get_main_guest_namez*UpcomingEventsService._get_main_guest_nameps)g'8"= q>++ +r event_type event_datec|j||j|t|j|jxsd|j r|j dnd|j dS)a Build a single event dictionary. Args: booking: Booking object event_type: "check_in" or "check_out" event_date: Date of the event Returns: Event dictionary rz.2fz0.00) booking_idr> guest_namechannelnightsamountr)r(r=rr-r+r, isoformat)rr8r>r?s r _build_eventz"UpcomingEventsService._build_eventsf$"**$33G<-g.>.>?,,16=6I6I,,S1v((*  rc |j|j}|Dcgc])}|j|jk(r |js|+}}|}t t |Dcgc]}t ||jfc}|Dcgc]}t||jfc}}|jdg}t}|D]\}} } |tk(r| j|vr|j| || } |j| |t k(r|j| jt||j k\s|S|Scc}wcc}wcc}w)a Get chronological list of upcoming events. Fetches check-ins and check-outs, combines them, sorts by date, and removes duplicates. Rules: - Only include FUTURE events - Exclude already checked-in bookings for today - Show check-in OR check-out (not both for same booking) Returns: List of event dictionaries, sorted chronologically c |dS)Nr/)xs rz2UpcomingEventsService.get_events..s1Q4r)key)r7rr)r.r2rrrr*sortsetr(rGappendaddlenr) r all_bookingsbbookings_checkinbookings_checkoutcombined_eventseventsadded_booking_idsr>r8r?events r get_eventsz UpcomingEventsService.get_eventssv"55djjA $ #!OOtzz1aoo |  ) .-)!Q__=- /.*1a.>.>?.    0E/> +J11gjjDU6U%%gz:FE MM% 00!%%gjj16{djj( !0?  [ s.E  E E)NN)__name__ __module__ __qualname____doc__rintrrstrrrrrr r7r=rGr[r/rrrr s'+#\sm\}\ PtCH~P2 D2 T']2 hG"     c3h  6BDc3h0BrrN)r_datetimer itertoolsrtypingrrrrdjango.db.modelsr bookings.modelsr guests.modelsr dashboard.constantsr r rrdashboard.services.utilsrrrr/rrrjs= ,,%# G~~r