a 'Ni 9@sdZddlZddlZddlmZmZmZddlmZddZ ddZ d d Z d d Z d dZ ddZddZddZddZddZddZddZddZdS)a Stats Manager Module This module contains all stats and state update functions extracted from functions.py. It manages player statistics, character updates, event checking, and various game state modifications. Functions included: - connect(player): Handle player reconnection and offline stats - checkDilemmas(player): Check and trigger active dilemmas - checkEvents(player, type): Check and trigger regular events - checkDayEvents(player, type): Check and trigger day-specific events - checkTutorialEvents(player, type): Check and trigger tutorial events - scheduleComplete(person, scheduleID): Check if a schedule is complete - updateAge(player): Update character ages and handle birthdays/deaths - setLikesDislikes(person): Set random likes and dislikes for a character - handleUpdates(updateObject, player, websocket): Handle incremental state updates - parseOneTimeEvents(player): Parse and trigger one-time scheduled events - handleMoods(player, person): Update character moods based on stats - handleFinances(person): Handle character finances and savings - getPeakEnergy(person): Calculate peak energy based on activities and habits Note: - FocusClass, getFocus(), getFocuses() are imported from education.education_manager - updateDeathChance() is imported from health.health_manager N) FocusClass getFocusesgetFocus)updateDeathChancecCsd|_|jj}|dkr||dkr(d}|}n"|dkr>d}|d}n d}|d}|dkrZ|d 7}|jd |d |d d|j_d S)zv Handle player reconnection and display offline time message. Args: player: playerClass instance connectedr<minuteihourdayszWelcome back, you missed  z of your life.N) connection offlineStatsminutesOffline messageQueueappend)playerminutesunittimer4/var/www/lichun.app/lichun/ws/stats/stats_manager.pyconnect's rcCs`|jD]T}tddkrddl}t|D].}||jkr*t||}||d|dSq*qdS)z Check active dilemmas and trigger them randomly. Args: player: playerClass instance Returns: Event result if triggered, None otherwise drNdilemma)typer)activeDilemmasrandomeventsdirfunctiongetattr)rrr iitemrrr checkDilemmasDs    r&cCsfddl}ddl}hd}t|D]@}t||}t|r |j|vr ||s |||}|r |Sq dS)z Check and trigger regular events from events.py. Args: player: playerClass instance type (str): Event type to check for Returns: Event result if triggered, None otherwise rN> apply_tutorial_modifiers answerOption dilemmaClass timeEventcheck_tutorial_triggers questionEventZget_event_type oneTimeEventZparse_event_dataConversationContextManager messageEvent)r inspectr!r#callable__name__iscoroutinefunction)rrr r0excluded_namesr$r%resultrrr checkEventsXs     r6cCsfddl}ddl}hd}t|D]@}t||}t|r |j|vr ||s |||}|r |Sq dS)z Check and trigger day-specific events from dayEvents.py. Args: player: playerClass instance type (str): Event type to check for Returns: Event result if triggered, None otherwise rN>r)r*r, get_dailyPlanr-r.r/) dayEventsr0r!r#r1r2r3)rrr8r0r4r$r%r5rrrcheckDayEventsvs     r9c CszHddl}||sWdS||}|D]}|||}|r(|WSq(Wn4ty|}ztd|WYd}~n d}~00dS)z Check and trigger tutorial events if player is in tutorial mode. Args: player: playerClass instance type (str): Event type to check for Returns: Event result if triggered, None otherwise rNz Error checking tutorial events: )tutorial_eventsis_tutorial_moder+ Exceptionloggingerror)rrr:ZtriggersZ trigger_funcr5errrcheckTutorialEventss    &r@cCsHtdt|jD]2}||j|jvr|j|j|j|jkrdSqdS)z Check if a schedule has been completed. Args: person: personClass instance scheduleID (str): ID of the schedule to check Returns: bool: True if schedule is complete, False otherwise rTF)rangelen schedulesid executionsduration)personZ scheduleIDr$rrrscheduleCompletes (rHcCsddlm}|jjd|j_|jjddkr|jjd|j_t|jD]\}}|jdkrH|jd|j|_|jjdkr|jddkr|j|j d8_ |jddkr@|j|jd7_|j|j d8_ t ||j|_ d |j vr@t |d d r@d |jd |jd |jdt|jd}|d||dS|j |jdtdksj|jdkrd|j|_d |jd |jd |jdt|jd|_|j|j dkrHd|j|_ qH|S)z Update character ages and handle birthdays, deaths, and relationship decay. Args: player: playerClass instance Returns: playerClass or messageEvent: Updated player or birthday message event r)messageFunctionr aliveim classmatetitleFzYour r z is now z years old.birthdayTrxZdeadz has died at the age of i)r rIcageHoursageDays enumeraterstatusageYearsaffinityr deathChance relationshipsr#rO firstnamelastnamestrhealthr capitalizemessage)rrIindexr%rarrr updateAges,  0* :rccsJgd}t|tdd_fdd|D}t|tdd_S)z Set random likes and dislikes for a character. Args: person: personClass instance Returns: personClass: Updated person with likes and dislikes )0ZsportsZmusicZartreadingwritingZsciencemathhistoryZpoliticsZ philosophyZreligionZnatureZanimalstravelfoodZcookingZcarsZfashionZfitnessr_Z technologyz video gamesZmoviesZtvZtheaterZdanceZcomedyZdramaZhorrorromanceactionZ adventureZfantasyzsci-fiZmysteryZthrillerZcrimeZ documentaryZanimeZmangaZcartoonsz board gamesz card gamesBroccoliCelery MushroomszBrussels sproutsOnionsr csg|]}|jvr|qSr)likes).0ZinterestrGrr z$setLikesDislikes..)rsamplerandintrqdislikes)rGZpossible_interestsrrsrsetLikesDislikess rycCs\gd}|D]4}t|j|}|||kr4|||<q ||dq ddg}|D]>}t|j|}t|t||gkr|||<qN||dqN|j|dkrt|j|j|d<n |dd|j|dkr||j |j|j dn\|jdkr.|j |d kr.|jd kr.||j |j|j dndD]}||dq2|rXd |d <|Sd S)a Handle incremental state updates to send to client. Only includes changed values to minimize data transfer. Args: updateObject (dict): Object to populate with updates player: playerClass instance websocket: WebSocket connection (not currently used) Returns: dict or False: Update object with type 'u' if changes exist, False otherwise ) energy calcEnergy happinessstressmoneyprestige occupationintraDayMessagelocationNrC dailyPlan gameSpeed hourOfDay)dater weekDayTextr rrurF) r#rRgetpoprBrprintrupdaterr)Z updateObjectr websocket attributes attributeZ player_valueZlist_attributeskeyrrr handleUpdatessD        * rcCstdt|jjD]t}|jj|}|j|jkr|j|jkr|j|j |jj ||j dkrt |drt| |d|_ d|_d|_qdS)zf Parse and trigger one-time scheduled events. Args: player: playerClass instance rNrun_funcF)rArBrR oneTimeEventsrrr rrrarcompletionFunchasattrrcompletionArgscompletionKwargs)rr$eventrrrparseOneTimeEvents;s    rcCs|jdkr|jdkrd|_|jdkr*d|_|jdkr:d|_|jdkrT|jdkrTd|_|jdkrn|jdkrnd|_|jdkr|jdkr|jdkrd |_d S) z Update character mood based on energy and happiness levels. Args: player: playerClass instance (not currently used) person: personClass instance to update mood for ( Stressed ExhaustedCalmr Fulfilled DepressedHappyN)rzmoodr|)rrGrrr handleMoodsOs   rcCsd}|jdkrd}n|jdkr"d}d}|jdkr|jD]}|jdkr6||jj7}q6t|dd }|rt|d d d krz|d 9}|j||7_d S) zq Handle character finances including income and savings. Args: person: personClass instance g?frugalg? extravagantg?rFjobNhourTypepartTimeg333333?)spendingHabitsractivityRecordsrlevelsalaryr#r~)rGZ savingRatioincomerecordrrrrhandleFinancesgs      rcCsd}|jD]}|jdkr |d7}q |jD]`}t|ddrB||j7}|jD]>}|j|jkrH|jdkrn||jj 7}t |j }|rH||j7}qHq(||_ |j |j |_dS)z Calculate peak energy based on activities, habits, and focus settings. Updates person.peakEnergy and person.calcEnergy. Args: person: personClass instance rquittingrpenergyModifierFrN)habitsrW activitiesr#rrrDrrenergy_modifierrfocus peakEnergyrzr{)rGrhabitactivityrZ foundFocusrrr getPeakEnergys            r)__doc__rr=education.education_managerrrrhealth.health_managerrrr&r6r9r@rHrcryrrrrrrrrrs" $E