a ~ it@sdZddlmZmZmZmZmZmZddlZe e Z GdddZ e Z eeddddZeeed d d Zed d dZdS)zi Event registry with condition-based indexing. Replaces O(n) event scanning with O(1) filtered lookups. )CallableOptionalTupleListDictAnyNc @sxeZdZdZddZdeeeee e fe e ee ddddZ e e ed d d Ze d d dZe ed ddZdS) EventRegistrya Registry for game events with efficient filtering. Events can be registered with conditions: - age_range: (min_age, max_age) - requires_relationship: bool - requires_job: bool - day_of_year: int or range Usage: registry = EventRegistry() registry.register("birthday", birthday_event, age_range=(1, 120)) applicable = registry.get_applicable_events(player) for event_info in applicable: result = event_info['func'](player, 'check') cCsi|_i|_g|_dS)N)_events_by_age_no_conditionsselfr//var/www/lichun.app/lichun/ws/event_registry.py__init__ szEventRegistry.__init__NF)event_idfunc age_rangerequires_relationship requires_job day_of_yearreturncCs||jvr td|ddS||||||d|j|<|r`||jvrPg|j|<|j||t||||gs||j|td|dS)a Register an event with optional conditions. Args: event_id: Unique event identifier func: Event function(player, mode) -> event object age_range: (min_age, max_age) tuple requires_relationship: Event needs relationships requires_job: Event needs player to have a job day_of_year: Specific day of year (1-365) zEvent z4 already registered, skipping duplicate registrationN)idrrrrrzRegistered event: )r loggerdebugr appendanyr )r rrrrrrrrrregister's"     zEventRegistry.registerplayerrc Csg}|jj}t|dd}tt|dgdk}tt|jdd}|jD]\}}|\} } | |krl| krHnqH|D]T} | |jvrt|j| } | dr|sqt| dr|sqt| dr| d|krqt| | qtqH|j D]} | |jvr| |j| q|S) z Get events applicable to player's current state. Args: player: playerClass instance Returns: List of event info dicts: [{'id', 'func', ...}, ...] dayOfYearNrr occupationrrr) cageYearsgetattrlenboolr itemseventsr rr ) r rZ applicableZ player_ageZ player_dayZhas_relationshipsZhas_jobrZ event_idsZmin_agemax_agerZ event_inforrrget_applicable_eventsUs,        z#EventRegistry.get_applicable_eventsrcCs t|jS)zGet number of registered events)r&r r rrrcount~szEventRegistry.countcCst|jS)zList all registered event IDs)listr keysr rrr list_eventsszEventRegistry.list_events)NFFN)__name__ __module__ __qualname____doc__rstrrrrintr'rrrdictr+r-r0rrrrr s$  .)r)rrrcKstj||fi|dS)z(Register an event in the global registryN) _registryr)rr conditionsrrrregister_eventsr:rcCs t|S)z*Get applicable events from global registry)r8r+)rrrrr+sr+r,cCstS)z Get total registered event count)r8r-rrrr event_countsr;)r4typingrrrrrrlogging getLoggerr1rrr8r5r:r7r+r6r;rrrrs  {