jrdZddlmZddlmZmZmZddlmZm Z ddl m Z ddl m Z ddlmZmZddlmZdd lmZGd d Zy ) z Overview service for Aimantis dashboard. Calculates today's operational metrics: - Check-ins and check-outs - Guests currently in structure - Room occupancy status )date)DictAnyOptional)CountQ)Booking)Guest)Property PropertyType) Structure) get_todayceZdZdZddeefdZdeee ffdZ dZ defdZ defd Z defd Zdefd Zdefd Zdefd ZdefdZdeee ffdZy)OverviewServicez Service for calculating dashboard overview metrics. All calculations are optimized to minimize database queries and avoid N+1 problems. N structure_idc0||_t|_y)z Initialize the overview service. Args: structure_id: Optional structure ID for multi-tenant filtering N)rrtoday)selfrs //backend/dashboard/services/overview_service.py__init__zOverviewService.__init__s)[ returnc:|jrd|jiSiS)z&Get structure filter dict for queries.rrrs r_get_structure_filterz%OverviewService._get_structure_filter)s 6:6G6G 1 12OROrctjj}|jr|j |j}|S)zGet filtered property queryset.r)r objectsallrfilter)rqss r_get_property_querysetz&OverviewService._get_property_queryset-s:    ! ! #   (9(9:B rctjjdd|ji|j j S)z Count bookings checking in today. Uses check_in_date == today. Counts all arrivals scheduled for today regardless of check-in status. check_in_dater rr rrcountrs rget_checkins_todayz"OverviewService.get_checkins_today4sA%% ** ((*  %' rctjjdd|ji|j j S)zc Count bookings checking out today. Uses check_out_date == today. check_out_dater%r&rs rget_checkouts_todayz#OverviewService.get_checkouts_today@sA %% :: ((*  %' rctjjd|j|jd|j j d}t jj|jS)a% Count guests currently staying in the structure. Uses ACTIVE booking logic (date-based, not check-in status): - check_in_date <= today - check_out_date > today (exclusive) Counts all guests from active reservations for today. check_in_date__ltecheck_out_date__gtid) booking__inr%)r rr rronlyr r')rcurrent_bookingss rget_guests_in_structurez'OverviewService.get_guests_in_structureKst#??11 #zz#zz ((*  $t*  }}##($ %' rc>|jjS)z#Count total rooms in the structure.)r"r'rs rget_total_roomszOverviewService.get_total_rooms_s**,2244rc(tjjd|j|jd|j j ddj }tjj|j}|S)aV Count physically occupied rooms. Uses ACTIVE booking logic (date-based, not check-in status): - check_in_date <= today - check_out_date > today (exclusive) Counts all rooms with active reservations for today. Returns DISTINCT property count to avoid double-counting. r- property_idT)flat)id__inr%) r rr rr values_listdistinctr r')roccupied_property_idsoccupied_counts rget_occupied_roomsz"OverviewService.get_occupied_roomscs!( 6 6! #zz#zz! ((*!  +m$+ / "))00(1 %' rc`|j}|j}td||z S)zV Count available rooms. Available = Total - Occupied r)r6r?max)rtotaloccupieds rget_available_roomsz#OverviewService.get_available_rooms}s2 $$&**,1eh&''rctjjd}|jr|j |j}d}|D]/}|t d|j jDz }1|S)z Count total beds across all property types in the structure. Optimized to use prefetch_related to avoid N+1 queries. bedsrrc34K|]}|jywN)quantity).0beds r z1OverviewService.get_total_beds..sDmscllms)r rprefetch_relatedrr sumrFr)rproperty_types total_bedspts rget_total_bedszOverviewService.get_total_bedssr &-->>vF   +22@Q@Q2RN  B #DbggkkmDD DJ!rc|j}|j}td||z }|j|j |j ||||j dS)z Get complete overview metrics in a single optimized call. Returns: Dict with all overview metrics r)checkins_todaycheckouts_todayguests_in_structureavailable_roomsoccupied_rooms total_roomsrP)r?r6rAr(r+r4rR)rrXrYrWs r get_overviewzOverviewService.get_overviewsv002**, a~!=>#557#779#'#?#?#A.,&--/  rrH)__name__ __module__ __qualname____doc__rintrrstrrrr"r(r+r4r6r?rDrRrZr%rrrrs!Xc]!PtCH~P C  S (55C4(S($ d38n rrN)r^datetimertypingrrrdjango.db.modelsrrbookings.modelsr guests.modelsr properties.modelsr r structures.modelsr dashboard.services.utilsrrr%rrris3&&%#4'.W W r