a i]@s0dZddlmZmZmZmZddlmZmZddlZddl Z ddl Z ddddd d d d d dddd ddd ddddd ddd ddddd ddd ddddddd d d!d"dd#ddd$d d%d&dddd'd(d d)d*d+ddd,d-d d.d/d0d'd1d0d2d d3d4d5d6d1d7d8d d9d:d#d;d1dd?Z e eeeefd@dAdBZdhe ee eeeefdCdDdEZe e e edFdGdHZe eeeefd@dIdJZe eeefd@dKdLZeeefeeefdMdNdOZe eeeefdPdQdRZe dSdTdUZdVdWZdXdYZdZd[Zd\d]Zd^d_Zd`daZdbdcZdddeZdfdgZ dS)iz Daily Quest System Generates and tracks daily quests for player engagement. Players receive 3 quests daily (easy/medium/hard) that award diamonds on completion. )DictAnyListOptional)datedatetimeNtalk_to_characterszTalk to 3 different characters easy message)typedescrequiredreward difficultyenergyiconbuy_itemzBuy an item from the storeZcart attend_classzAttend 2 classesbook socializezSocialize with friendszperson.2 work_hourszWork for 6 hoursmedium briefcase go_on_datez Go on a dateheartcomplete_activitieszComplete 5 activitieszcheckmark.circlestudyzStudy for 4 hours z book.fill spend_energyzSpend 50 energy on activities2hardZbolt earn_moneyz Earn $500i(Z dollarsignincrease_affinityz$Increase affinity by 20 points total#-z heart.fillc sddlm}z^tD]>}|d|d|d|d|d|d|d |d fId Hqtd ttd Wn8ty}z tjd|ddWYd }~n d }~00d S)z Insert quest template definitions into database. Called during server startup. Uses async database pool for proper connection management. r execute_queryaINSERT INTO daily_quest_templates (quest_type, description, progress_required, diamond_reward, difficulty, energy_cost, icon_name) VALUES (%s, %s, %s, %s, %s, %s, %s) AS new_quest ON DUPLICATE KEY UPDATE description=new_quest.description, progress_required=new_quest.progress_required, diamond_reward=new_quest.diamond_rewardrrrrrrrNz Initialized z daily quest templatesz$Error initializing quest templates: Texc_info)database_asyncr5QUEST_TEMPLATESlogginginfolen Exceptionerror)r5questerA7/var/www/lichun.app/lichun/ws/retention/daily_quests.pyinitialize_quest_templates/s  rC) player_idreturncsRddlm}m}m}t}z|d||fIdH}|rf|ddkrftd|dt|IdHWS|dIdH}|st dgWSd d |D}d d |D}d d |D} |r|r| st d t |dt |dt | gWSt |t |t | g} g} |4IdH} | 4IdHx} | D]Z}| d||d|fIdH| | j|d|dd|d|d|d|ddd q2WdIdHq1IdHs0YWdIdHq1IdHs0Ytdt | d|| WStyL}z*tj d|d|ddgWYd}~Sd}~00dS) z Assign 3 random quests to player each day (1 easy, 1 medium, 1 hard). Called at midnight or on first login of the day. Args: player_id: The player's ID Returns: List of assigned quest dictionaries r) fetch_onefetch_dict_allget_connectionz]SELECT COUNT(*) as count FROM player_daily_quests WHERE player_id = %s AND assigned_date = %sNPlayer z already has quests for todayz#SELECT * FROM daily_quest_templatesz$No quest templates found in databasecSsg|]}|ddkr|qS)rr rA.0qrArArB lz)generate_daily_quests..cSsg|]}|ddkr|qS)rr rArJrArArBrMmrNcSsg|]}|ddkr|qS)rr-rArJrArArBrMnrNz Missing quest templates - easy: z , medium: z, hard: zINSERT INTO player_daily_quests (player_id, quest_template_id, assigned_date, progress, completed) VALUES (%s, %s, %s, 0, FALSE)id quest_type descriptionprogress_requireddiamond_rewardr icon_nameF rOrPrQprogressrRrSrrT completedz Assigned z daily quests to player z)Error generating daily quests for player : Tr6)r8rFrGrHrtodayr:r;get_active_questswarningr>r<randomchoicecursorexecuteappend lastrowidr=)rDrFrGrHrYresultZ all_questsZ easy_questsZ medium_questsZ hard_questsselectedZassigned_questsconnr^r?r@rArArBgenerate_daily_questsKs`    ( h re)rDrPamountrEc stddlm}m}t}z |d|||fIdH}|s) rDrPrfrgr5rYr?Z new_progressr@rArArBupdate_quest_progresssR   &   rk)rDquest_idrrEc sddlm}z\|d||fIdHddlm}||d||IdHtd|d|d |d Wd Sty}z.tjd |d |d|d dWYd}~dSd}~00dS)z Complete quest and award diamonds. Args: player_id: The player's ID quest_id: The quest record ID reward: Diamond reward amount Returns: True if successful, False otherwise rr4zUPDATE player_daily_quests SET completed = TRUE, completed_date = NOW() WHERE id = %s AND player_id = %sNaward_diamondsZ daily_quest_rIz completed quest z , awarded diamondsTzError completing quest z for player rXr6F)r8r5monetization.diamond_economyrnr:r;r=r>)rDrlrr5rnr@rArArBrjs    rjcsddlm}t}z|d||fIdH}g}|D]`}||d|d|d|d|d |d |d |d t|d |dr|dndd q2|WSty}z*tj d|d|ddgWYd}~Sd}~00dS)z Get player's current daily quests. Args: player_id: The player's ID Returns: List of quest dictionaries r)rGaSELECT pq.id, pq.progress, pq.completed, pq.completed_date, qt.quest_type, qt.description, qt.progress_required, qt.diamond_reward, qt.difficulty, qt.icon_name FROM player_daily_quests pq JOIN daily_quest_templates qt ON pq.quest_template_id = qt.id WHERE pq.player_id = %s AND pq.assigned_date = %s ORDER BY qt.difficulty DESCNrOrPrQrVrRrSrrTrWcompleted_date) rOrPrQrVrRrSrrTrWrqz'Error getting active quests for player rXTr6) r8rGrrYr`bool isoformatr=r:r>)rDrGrYquestsrbr?r@rArArBrZ s2    rZc Cs^d}d}z0zt}|jdd}|d|f|}|d|f|}|rX|dnd|rr|drrt|dnd|r|d ndd d |Dd WW|r||r|Sty}zRtj d |d|dddddid WYd}~W|r||r|Sd}~00W|r,||rZ|n|rJ||rX|0dS)z Get player's quest completion statistics. Args: player_id: The player's ID Returns: Dict with quest statistics NT dictionaryawSELECT COUNT(*) as total_completed, SUM(qt.diamond_reward) as total_diamonds_earned, COUNT(DISTINCT pq.assigned_date) as days_completed FROM player_daily_quests pq JOIN daily_quest_templates qt ON pq.quest_template_id = qt.id WHERE pq.player_id = %s AND pq.completed = TRUEa$SELECT qt.difficulty, COUNT(*) as completed_count FROM player_daily_quests pq JOIN daily_quest_templates qt ON pq.quest_template_id = qt.id WHERE pq.player_id = %s AND pq.completed = TRUE GROUP BY qt.difficultytotal_completedrtotal_diamonds_earnedZdays_completedcSsi|]}|d|dqS)rZcompleted_countrA)rKitemrArArB hrNz(get_quest_statistics..)rwrxZdays_with_quests by_difficultyz*Error getting quest statistics for player rXr6) get_database_connectionr^r_fetchonefetchallintcloser=r:r>)rDrdr^statsr{r@rArArBget_quest_statistics:sZ      r)r?rEc Csddlm}dddd}dddddddddddd }|d d }||||d d d}|ddot|ddu}t|dd|dvr|dddn|ddd|d||dddd|d|d|d|d S)z Format quest data to match iOS DailyQuest model expectations. Args: quest: Raw quest dictionary from database Returns: Formatted quest dictionary matching iOS model r)r activitiescareerZwealth)r r r-social education) rrr#r1rr.rrr(r&r+rPrr rWFrqNrOz - rQr/rSdiamondsrmoneyrVrR) rOnamerQcategoryrrVtargetrWclaimed)rgetstrsplit capitalize)r?rZ category_mapZtype_to_categoryrPrrrArArBformat_quest_for_client{sB   ,r)rDrlrEc Cs(d}d}zzXt}|jdd}|dt||f|}|sjddddWW|r\||rh|S|dsdd ddWW|r||r|S|d rdd ddWW|r||r|S|d t||fd dlm}|||dd|t d|d|d|dddd|dd|dddddWW|rX||rf|St y}zd|r| t j d|d|d|ddddddWYd}~W|r||r|Sd}~00W|r||r$|n|r||r"|0dS)z Claim reward for a completed quest. Args: player_id: The player's ID quest_id: The quest ID to claim Returns: Dictionary with success status and reward info NTruzSELECT pq.*, qt.diamond_reward FROM player_daily_quests pq JOIN daily_quest_templates qt ON pq.quest_template_id = qt.id WHERE pq.id = %s AND pq.player_id = %sFzQuest not found)successr rrWzQuest not completed yetrqzReward already claimedztUPDATE player_daily_quests SET completed_date = NOW() WHERE id = %s AND player_id = %srrmrSzDaily Quest RewardrIz claimed quest z reward: rozClaimed z diamonds!rz'Error claiming quest reward for player z, quest rXr6zFailed to claim reward)r|r^r_rr}rrprncommitr:r;r=rollbackr>)rDrlrdr^r?rnr@rArArBclaim_quest_rewards   :3+  "   r)rDc Cszddlm}m}m}t|}|s*t|}|rt y} z&t jd |d | d dWYd} ~ n d} ~ 00dS)z WebSocket handler for daily quest check. Called when player connects to server or at midnight. Args: player_id: The player's ID send_to_client: Function to send messages to client (player_id, message_dict) r)rr timedeltacSsg|] }t|qSrA)rrJrArArBrM&rNz,handle_daily_quest_check..r)daysZdailyQuestsStatus)rrtZ lastResetDate nextResetDatez"Sent daily quests state to player rXz questsz,Error handling daily quest check for player Tr6N)rrrrZrerYcombineritimersr:r;r<r=r>) rDsend_to_clientrrrZ active_questsZformatted_questsrYtomorrowZ last_reset next_resetr@rArArBhandle_daily_quest_checks&  rcCsdS)aI Example: Update quest when player talks to a character Integration point: ws/conversationEvents.py or wherever conversations are handled Add this after a successful conversation: from retention.daily_quests import update_quest_progress update_quest_progress(player_id, 'talk_to_characters', 1) NrArArArArB example_integration_conversationCs rcCsdS)a Example: Update quest when player works Integration point: ws/intradayActivity.py or work event handlers Add this after work hours are completed: from retention.daily_quests import update_quest_progress # Track work hours update_quest_progress(player_id, 'work_hours', hours_worked) # Track activity completion update_quest_progress(player_id, 'complete_activities', 1) NrArArArArBexample_integration_workPsrcCsdS)a_ Example: Update quest when player studies or attends class Integration point: Education/school event handlers Add this after attending class: from retention.daily_quests import update_quest_progress update_quest_progress(player_id, 'attend_class', 1) update_quest_progress(player_id, 'study', hours_studied) NrArArArArBexample_integration_studybs rcCsdS)a Example: Update quest when player makes a purchase Integration point: Store/purchase handlers Add this after successful purchase: from retention.daily_quests import update_quest_progress update_quest_progress(player_id, 'buy_item', 1) NrArArArArBexample_integration_shoppingps rcCsdS)a Example: Update quest when player goes on a date Integration point: Dating/relationship event handlers Add this when date activity starts: from retention.daily_quests import update_quest_progress update_quest_progress(player_id, 'go_on_date', 1) NrArArArArBexample_integration_dating}s rcCsdS)a! Example: Update quest when player spends energy Integration point: Activity/energy deduction handlers Add this when energy is spent: from retention.daily_quests import update_quest_progress update_quest_progress(player_id, 'spend_energy', energy_amount) NrArArArArBexample_integration_energys rcCsdS)a Example: Update quest when player earns money Integration point: Work/income event handlers Add this when money is earned: from retention.daily_quests import update_quest_progress update_quest_progress(player_id, 'earn_money', amount_earned) NrArArArArBexample_integration_moneys rcCsdS)a. Example: Update quest when affinity increases Integration point: Relationship/affinity update handlers Add this when affinity increases: from retention.daily_quests import update_quest_progress update_quest_progress(player_id, 'increase_affinity', affinity_increase) NrArArArArBexample_integration_affinitys rcCsdS)a Example: Update quest when player socializes Integration point: Social activity handlers Add this when social activity completes: from retention.daily_quests import update_quest_progress update_quest_progress(player_id, 'socialize', 1) NrArArArArBexample_integration_socializes r)r)!__doc__typingrrrrrrr:r\asyncior9rCrrrerkrrrjrZrrrrrrrrrrrrrrArArArBsrQ"J$0A :]1