a %ßi"ã@s¸dZddlmZmZmZddlZddlmZee ee dœdd„Z ee eee efdœdd „Z eeed œd d „Z deeedœdd„Zdddd ddddddddddœ Ze edœdd„ZdS)zx Diamond Economy System Handles diamond earning, spending, and tracking. Diamonds are the premium currency in BaoLife. é)ÚDictÚAnyÚOptionalN)Úget_database_connection)Ú player_idÚreasonÚamountÚreturnc Cs†|dkr$t d|›d|›d¡dSd}d}z4z¦tƒ}|jdd}| d ||f¡| d |||f¡| ¡| d |f¡| ¡}|r|d n|}t d |›d|›d|›d|›d ¡WW|rÈ| ¡|rÔ| ¡dSt yB}zR|rô|  ¡tjd|›d|›ddWYd}~W|r(| ¡|r6| ¡dSd}~00W|rT| ¡|r‚| ¡n|rr| ¡|r€| ¡0dS)a Award diamonds to player with tracking. Args: player_id: The player's ID reason: Reason for awarding diamonds (for logging) amount: Number of diamonds to award (must be positive) Returns: True if successful, False otherwise rú"Invalid diamond amount for player ú: ú (must be positive)FNT©Ú dictionaryz9UPDATE players SET diamonds = diamonds + %s WHERE id = %szˆINSERT INTO diamond_transactions (player_id, transaction_type, amount, reason) VALUES (%s, 'earn', %s, %s)ú*SELECT diamonds FROM players WHERE id = %sÚdiamondsúPlayer z earned ú diamonds: z (total: ú)z"Error awarding diamonds to player ©Úexc_info) ÚloggingÚerrorrÚcursorÚexecuteÚcommitÚfetchoneÚinfoÚcloseÚ ExceptionÚrollback)rrrÚconnrÚresultZ new_totalÚe©r#ú=/var/www/lichun.app/lichun/ws/monetization/diamond_economy.pyÚaward_diamonds sZ  þü$ ô ú ýr%c Cs|dkr*t d|›d|›d¡dddœSd}d}z°ztƒ}|jd d }| d |f¡| ¡}|sŒdd dœWW|r~| ¡|rŠ| ¡S|d }||krÎdd|›d|›dœWW|rÀ| ¡|rÌ| ¡S| d||f¡| d|||f¡| ¡||}t d|›d|›d|›d|›d ¡d d|dœWW|r@| ¡|rN| ¡St yÄ}z\|rn|  ¡tjd|›d|›d ddddœWYd}~W|r¬| ¡|rº| ¡Sd}~00W|rÖ| ¡|r| ¡n|rô| ¡|r| ¡0dS)a Deduct diamonds from player with validation. Args: player_id: The player's ID reason: Reason for deducting diamonds (for logging) amount: Number of diamonds to deduct (must be positive) Returns: dict with 'success' and 'message' keys rr r r FzInvalid amount)ÚsuccessÚmessageNTr rzPlayer not foundrzInsufficient diamonds. Need z, have z9UPDATE players SET diamonds = diamonds - %s WHERE id = %sz‰INSERT INTO diamond_transactions (player_id, transaction_type, amount, reason) VALUES (%s, 'spend', %s, %s)rz spent rz (remaining: rzDiamonds deducted successfully)r&r'Ú new_balancez%Error deducting diamonds from player rz Server error) rrrrrrrrrrr) rrrr rr!Zcurrent_balancer(r"r#r#r$Údeduct_diamondsKs‚    *Óþ%Øþü$ý ðú ýr))rr c Csðd}d}zÈzTtƒ}|jdd}| d|f¡| ¡}|r@|dndWW|rR| ¡|r^| ¡Sty´}z>t d|›d|›¡WYd}~W|rœ| ¡|r¨| ¡dSd}~00W|rÄ| ¡|rì| ¡n|rÞ| ¡|rê| ¡0dS)zŸ Get player's current diamond balance. Args: player_id: The player's ID Returns: Diamond balance or None if player not found NTr rrz)Error getting diamond balance for player r )rrrrrrrr)rr rr!r"r#r#r$Úget_diamond_balance˜s:  ö ú ýr*é2)rÚlimitr c Csäd}d}z¼zFtƒ}|jdd}| d||f¡| ¡WW|rD| ¡|rP| ¡Sty¨}z@t d|›d|›¡gWYd}~W|r’| ¡|rž| ¡Sd}~00W|r¸| ¡|rà| ¡n|rÒ| ¡|rÞ| ¡0dS)zÐ Get player's diamond transaction history. Args: player_id: The player's ID limit: Maximum number of transactions to return Returns: List of transaction dictionaries NTr zÀSELECT transaction_type, amount, reason, created_at FROM diamond_transactions WHERE player_id = %s ORDER BY created_at DESC LIMIT %sz-Error getting transaction history for player r )rrrÚfetchallrrrr)rr,r rr"r#r#r$Úget_diamond_transaction_history¹s>  ú öú ýr.é éééédéé) Zlevel_upZcomplete_school_yearZgraduate_high_schoolÚgraduate_collegeZ get_first_jobÚ promotionÚ become_ceoÚ get_marriedZhave_first_childZreach_age_milestoneZ daily_loginZ weekly_streakZmonthly_streak)rr cCs t |d¡S)zª Get the diamond reward amount for a specific reason. Args: reason: The reason/achievement key Returns: Diamond amount (0 if not found) r)ÚDIAMOND_REWARDSÚget)rr#r#r$Úget_reward_amountós r<)r+)Ú__doc__ÚtypingrrrrÚdatabaserÚintÚstrÚboolr%r)r*Úlistr.r:r<r#r#r#r$Ús, >M!*ó