a %i`)@sdZddlmZmZddlmZmZmZddlZddlm Z ddl m Z eddd d edd d d edd dd ddd dZ e d dde eeeefdddZe eeeeefdddZe eeefdddZeeeeeffdddZdS)ze Time Skip System Handles time skip purchases and simulation of game events during skipped periods. )datetime timedelta)DictAnyListN)get_database_connection) rate_limithoursdurationdiamondsdays)weeksd2)Z1hourZ1dayZ1week next_event<) max_calls time_window) player_id skip_typereturnc CsTd}d}z&z|tvrDdd|dWW|r6||rB|St|}|d}t}|jdd}|d|f|}|sdd dWW|r||r|S|d rdd dWW|r||r|S|d|krdd |d |ddWW|r ||r|S|d}t|tr:t |}|dkrT|t dd}n ||d}t ||||} |d||| d| d| d| d||f|j dkr| dddWW|r||r|S|d|} |d|||||t| df|td|d|d|d||d } dd!| || | ddd"| d| d| d| dd#d$d%WW|r||r|Sty} z\|r| tjd&|d'| dd(dd)dWYd} ~ W|r||r|Sd} ~ 00W|r"||rP|n|r@||rN|0dS)*a1 Handle time skip purchase and simulation. Rate limited to 5 purchases per minute to prevent abuse. Args: player_id: The player's ID skip_type: Type of skip ('1hour', '1day', '1week', 'next_event') Returns: dict with 'success', 'message', and 'summary' keys NFzInvalid skip type: )successmessagerT) dictionaryzSELECT diamonds, date, hourOfDay, minuteOfHour, energy, health, happiness, age, dead FROM players WHERE id = %szPlayer not founddeadzCannot skip time after deathzNot enough diamonds. Need z, have daterrrraDUPDATE players SET diamonds = diamonds - %s, date = %s, energy = GREATEST(0, energy + %s), health = GREATEST(0, health + %s), happiness = GREATEST(0, happiness + %s), money = money + %s WHERE id = %s AND diamonds >= %s energy_change health_changehappiness_change money_earnedrz>Insufficient diamonds (concurrent purchase or balance changed)zINSERT INTO time_skip_purchases (player_id, skip_type, diamond_cost, time_before, time_after, events_simulated) VALUES (%s, %s, %s, %s, %s, %s)eventszPlayer z skipped z for z diamondsizTime skipped successfully )moneyenergyhealth happiness)rZnew_timeduration_hoursr& stat_changes)rrsummaryzTime skip error for player z: )exc_infozServer error. Please try again.) SKIP_TIERSclosercursorexecutefetchone isinstancestrr fromisoformatrsimulate_time_periodrowcountrollbacklencommitlogginginfo total_seconds isoformat Exceptionerror) rrconnr2tier diamond_costplayerZ time_beforeZ time_afterZevents_summary new_diamondsr,erI8/var/www/lichun.app/lichun/ws/monetization/time_skips.pypurchase_time_skipsw  a ^Z     2    rK)r start_timeend_timercCsg}d}d}d}d}|d|f|} | s>gdddddS|} tdd} d} d} d}| |kr(| dkr(| j}| dr|dkr|d kr| d kr| d r| d d }||7}| d7} | d r|dkr|dkr| d kr|d7}|dks|dkr|d7}|d8}|d8}|d8}| | 7} | d7} qX| dkrP|dd| dt|d|dkrx|dd|d|ddtdtd|}td td!|}td"td#|}|d$d%| dt|t|t|d&d'|t|t|t|t|dS)(a0 Simulate game events during time skip period. Args: player_id: The player's ID start_time: Start of simulation period end_time: End of simulation period cursor: Database cursor (for efficiency) Returns: Summary dict with events and stat changes rzSELECT occupation, occupationSalary, educationLevel, age, relationships FROM players WHERE id = %s)r&r%r"r#r$rr occupation r ZoccupationSalaryeducationLevelg?g?g?workzWorked z hours)type descriptionr%schoolz Attended z hours of classes)rZr[Z smarts_gainedirir'ir.zTime advanced )r)r*r+)rZr[r-) r3r4rhourweekdayappendintmaxmin)rrLrMr2r&r%r"r#r$Z player_state current_timeZsimulation_stepZ hours_elapsed work_hoursZ school_hoursZ hour_of_dayZ hourly_wagerIrIrJr8s            r8)r message_datacCsv|d}|s$||dddddSt||}|dr\||dd |d d ||d d in||dd|dddS)z WebSocket message handler for 'purchaseTimeSkip' messages. Args: player_id: The player's ID message_data: Message data containing 'skipType' send_to_client: Function to send message back to client ZskipTyperBINVALID_REQUESTzMissing skipType)rZ error_coderNrZtimeSkipCompleteTr.)rZrr.rZZrequestFullUpdateZTIME_SKIP_FAILEDr)getrK)rrfsend_to_clientrresultrIrIrJhandle_purchase_time_skips.  rl)rcCsddtDS)z[ Get all available time skip tiers. Returns: Dictionary of skip tiers cSs4i|],\}}||dr"|dnd|ddqS)rNrr )r?).0keyvaluerIrIrJ Hs z"get_skip_tiers..)r0itemsrIrIrIrJget_skip_tiersAsrr)__doc__rrtypingrrrr=databaser validationr r0rar6rKr8rlrrrIrIrIrJs$    r(