a iZB @sHdZddlmZmZmZddlmZmZddlZddl Z ddl Z ddl m Z dddd d d dd d d ddddd ddddd ddddd ddddd ddddd gZ ddZeeeefddd Zeeeeefd!d"d#Zeeeefdd$d%Zeeeefdd&d'Zed(d)d*Zeeeefdd+d,Zed-d.d/ZdS)0z Daily Login Reward System Tracks 7-day login streaks and awards diamonds, energy, items, or prestige. Streak resets if player misses a day. )DictAnyOptional)date timedeltaN)get_database_connectiondiamondsz 5 Diamonds)daytypeamountname z 10 Diamondsenergy2z 50 Energyz 15 Diamondsz20 Diamonds + Itemz 25 Diamondsz50 Diamonds Bonusc sddlm}zLtD],}|d|d|d|d|dfIdHqtd ttd Wn8ty}z tjd |d d WYd}~n d}~00dS)z Insert reward definitions into database. Called during server startup. Uses async database pool for proper connection management. r) execute_querya@INSERT INTO daily_login_rewards (day_number, reward_type, reward_amount, display_name) VALUES (%s, %s, %s, %s) AS new_reward ON DUPLICATE KEY UPDATE reward_amount=new_reward.reward_amount, display_name=new_reward.display_namer r r rNz Initialized z daily login rewardsz"Error initializing daily rewards: Texc_info)database_asyncr DAILY_REWARDSlogginginfolen Exceptionerror)rrewarder&8/var/www/lichun.app/lichun/ws/retention/daily_rewards.pyinitialize_daily_rewardss  r() player_idreturnc Cs\d}d}t}z&zt}|jdd}|d|f|}|s|d||f|td}t d|ddd|d d WW|r| |r| S|d }||krd |d dd d WW|r| |r| S||j }d }|dkr|d d} |d dd} n$d} d} d}t d|d|d|d| || |f|t|d }t d|d| d|d d| ||d WW|r| |r| St y} zRtj d|d| ddd ddd d WYd} ~ W|r| |r| Sd} ~ 00W|r*| |rX| n|rH| |rV| 0dS)a Check and update daily login streak. Called when player connects to server. Args: player_id: The player's ID Returns: dict with keys: - reward_available: bool, whether reward can be claimed - streak: int, current streak count - reward: dict or None, today's reward details - streak_broken: bool, whether streak was broken today NT dictionary6SELECT * FROM player_login_streak WHERE player_id = %szINSERT INTO player_login_streak (player_id, current_streak, last_login_date, total_logins, next_reward_day) VALUES (%s, 1, %s, 1, 1)rPlayer z first login - starting streakF)Zreward_availableZstreakr$ streak_brokenlast_login_datecurrent_streaknext_reward_dayrz streak broken after z dayszUPDATE player_login_streak SET current_streak = %s, last_login_date = %s, total_logins = total_logins + 1, next_reward_day = %s WHERE player_id = %sz daily login - streak: z, day: z&Error checking daily login for player : rr)rtodayrcursorexecutefetchonecommitget_daily_rewardrr closedaysr"r#) r)connr5r4 streak_datar$ last_loginZdays_since_loginr/Z new_streakZnext_dayr%r&r&r'check_daily_login6s A6       r?) day_numberr*c CsJd}d}zzt}|jdd}|d|f|}|sjtd|WW|rZ||rf|dS|d|d|d|d |d d WW|r||r|Sty}zBtj d |d |ddWYd}~W|r||r|dSd}~00W|r||rF|n|r6||rD|0dS)z Get reward for specific day. Args: day_number: Day number (1-7) Returns: Dict with reward details or None if not found NTr+z7SELECT * FROM daily_login_rewards WHERE day_number = %szDaily reward not found for day r@ reward_type reward_amountZreward_item_id display_name)r r r Zitem_idrz#Error getting daily reward for day r3r) rr5r6r7rwarningr:getr"r#)r@r<r5r$r%r&r&r'r9sZ     r9c Csd}d}z~zt}|jdd}|d|f|}|sdddddWW|rV||rb|S|dtkrt|d d kr|d nd }t|}|sddd dWW|r||r|S|d dkrd dlm }||d||dn|d dkr|d|d|f| n|d dkrH|d|d|f| nh|d dkr|d| }|r|d }|d||df| t d|d|ddt d|d|d|ddWW|r||r|Styh} z^|r|t jd |d!| dd"ddd#dWYd} ~ W|rP||r^|Sd} ~ 00W|rz||r|n|r||r|0dS)$z Claim today's reward. Awards the reward to player and marks as claimed. Args: player_id: The player's ID Returns: dict with keys: - success: bool - reward: dict or None - message: str NTr+r-FzNo login streak found)Zsuccessr$messager0r2rrzReward not foundr r )award_diamondsZdaily_reward_dayr rzHUPDATE players SET energy = LEAST(energy + %s, max_energy) WHERE id = %sprestigez9UPDATE players SET prestige = prestige + %s WHERE id = %sitemzKSELECT * FROM store_items WHERE category = 'luxury' ORDER BY RAND() LIMIT 1zINSERT INTO player_inventory (player_id, item_id, quantity) VALUES (%s, %s, 1) ON DUPLICATE KEY UPDATE quantity = quantity + 1idr.z received luxury item: runknownz claimed daily reward: Day zReward claimed successfullyz'Error claiming daily reward for player r3rz Server error)rr5r6r7r:rr4r9monetization.diamond_economyrGr8fetchallrr rEr"rollbackr#) r)r<r5r=Z reward_dayr$rGZ luxury_itemsrIr%r&r&r'claim_daily_rewards U ?        rOc Cstd}d}t}z>zzt}|jdd}|d|f|}|sRd}d}d}n|d}|d}|d}|rv||knd }| } |d |} g} | D]~} | d } | d }| d }| |kp| |ko|}| |dkr|nd|dkr|nd|dkr|nd| dkrdnd|d}| |q|tdd}ddl m }| ||j  }|rX| n| }|||| || dWW|r||r|Sty0}ztjd|d|ddddl m }|tdd}d| | ||j  dd gdWYd}~W|r||r&|Sd}~00W|rB||rp|n|r`||rn|0dS)z Get complete daily reward state for iOS DailyRewardState model. Args: player_id: The player's ID Returns: Dictionary matching iOS DailyRewardState structure NTr+zSELECT current_streak, last_login_date, next_reward_day FROM player_login_streak WHERE player_id = %srr1r0r2FzSELECT day_number, reward_type, reward_amount, display_name FROM daily_login_rewards ORDER BY day_numberr@rArBr rrmoneyr z Luxury Item)rJr rrPZ bonusItemclaimed)r;)datetime)Z currentStreakZ lastLoginDateZ nextResetDateZcanClaimZ todaysClaimedrewardsz,Error getting daily reward state for player r3r)rr4rr5r6r7rMappendrrRcombinemintime isoformatr:r"rr#)r)r<r5r4Z streak_infor1r>r2Ztodays_claimedZ can_claimZreward_definitionsrSZ reward_defZday_numrAr rQZ day_rewardZtomorrowrRZ next_resetZlast_login_strr%r&r&r'get_daily_reward_stateOs         rY)r)c Cszz6t|t|}||ddi|td|Wn>tyt}z&tjd|d|ddWYd}~n d}~00dS) z WebSocket handler for daily login check. Called when player connects to server. Args: player_id: The player's ID send_to_client: Function to send messages to client (player_id, message_dict) r ZdailyRewardStatusz"Sent daily reward state to player z,Error handling daily login check for player r3TrN)r?rYrr r"r#)r)Zsend_to_clientstater%r&r&r'handle_daily_login_checks r[c Cs\d}d}z.zt}|jdd}|d|f|}|sddddddWW|rV||rb|S|d|d |d |d r|d nddWW|r||r|Sty}zRtjd |d |dddddddWYd}~W|r||r|Sd}~00W|r*||rX|n|rH||rV|0dS)z Get player's login streak information. Args: player_id: The player's ID Returns: dict with streak info NTr+r-rr)r1 total_loginsr2r0r1r\r2r0z%Error getting streak info for player r3r) rr5r6r7r:rXr"rr#)r)r<r5r=r%r&r&r'get_login_streak_infosf   r])r*c Csd}d}zz@t}|jdd}|d|WW|r>||rJ|Sty}z>tjd|ddgWYd}~W|r||r|Sd}~00W|r||r|n|r||r|0dS)z] Get all daily reward definitions. Returns: List of reward dictionaries NTr+z5SELECT * FROM daily_login_rewards ORDER BY day_numberzError getting all rewards: r)rr5r6rMr:r"rr#)r<r5r%r&r&r'get_all_rewards s<   r^)__doc__typingrrrrRrrrrandomasynciodatabase.db_operationsrrr(intstrr?r9rOrYr[r]listr^r&r&r&r's,         s.xi6