a %ib @sdZddlmZmZmZmZddlmZddlZddlZddl m Z e e dddZ d+e ee e d d d Ze ee e d ddZe ee e d ddZe eeefdddZd,e eeeeefee eeee dddZd-e e e eeeefdddZe e dddZe eeeeefdddZe e dd d!d"Ze e dd d#d$Ze ddd%d&Ze ddd'd(Ze ddd)d*ZdS).z Player Statistics and Photo Album System Tracks lifetime statistics and captures memorable life moments. Supports 15+ different statistics and integrates with achievement system. )DictAnyListOptional)datetimeN)get_database_connection) player_idreturnc Csd}d}zzTt}|}|d|f|td|WW|rP||r\|dSty}zN|rz|tj d|d|ddWYd}~W|r||r|dSd}~00W|r||r|n|r||r|0dS) z Create initial statistics record for new player. Called when player account is created. Args: player_id: The player's ID Returns: bool: True if successful, False otherwise NINSERT INTO player_statistics (player_id) VALUES (%s) ON DUPLICATE KEY UPDATE player_id = player_idz"Initialized statistics for player Tz)Error initializing statistics for player : exc_infoF) rcursorexecutecommitlogginginfoclose Exceptionrollbackerror)rconnrer5/var/www/lichun.app/lichun/ws/retention/statistics.pyinitialize_player_statisticssF    r)r stat_nameamountr c Cs|dkr$td|d|ddSd}d}gd}||vrPtd|dSzzt}|}|d |fd |d |d }||||f|td |d|d|WW|r||r|dStyH}zX|r| tjd|d|d|ddWYd}~W|r.||r<|dSd}~00W|rZ||r|n|rx||r|0dS)a Increment a player statistic by a specified amount. Args: player_id: The player's ID stat_name: Name of the statistic column (e.g., 'total_relationships') amount: Amount to increment by (default: 1, must be positive) Returns: bool: True if successful, False otherwise Valid stat_name values: - lifetime_earnings, lifetime_spending - total_relationships, total_activities, total_conversations - total_years_lived, total_deaths - highest_job_level, max_affinity_reached - people_dated, times_fired, years_married - job_count, children_count, friends_count rz$Invalid increment amount for player r z (must be positive)FNlifetime_earningslifetime_spendingtotal_relationshipstotal_activitiestotal_conversationsZtotal_years_livedZ total_deathsZhighest_job_levelZmax_affinity_reached people_dated times_fired years_married job_countchildren_count friends_countInvalid stat_name: r UPDATE player_statistics SET z = z + %s WHERE player_id = %sPlayer z incremented by TzError incrementing stat for player r rrrrrrdebugrrr)rrrrr valid_statsqueryrrrrincrement_stat=sX     r3)rrvaluer c Csbd}d}gd}||vr,td|dSzz|t}|}|d|fd|d}||||f|td|d |d |WW|r||r|d Sty}zX|r| tjd |d |d |d dWYd}~W|r||r|dSd}~00W|r0||r^|n|rN||r\|0dS)a Set a player statistic to a specific value. Used for 'highest' or 'max' type stats that track records. Args: player_id: The player's ID stat_name: Name of the statistic column value: Value to set Returns: bool: True if successful, False otherwise Example: # Update highest job level reached update_stat(player_id, 'highest_job_level', 5) Nrr+Fr r, = %s WHERE player_id = %sr-r  set to TzError updating stat r.r r/)rrr4rrr1r2rrrr update_statsR      r7c Cs`d}d}dg}||vr*td|dSzz|t}|}|d|fd|d}||||f|td|d |d |WW|r||r|d Sty}zX|r| tjd |d |d |d dWYd}~W|r||r|dSd}~00W|r.||r\|n|rL||rZ|0dS)a Set a boolean statistic (e.g., ever_married). Args: player_id: The player's ID stat_name: Name of the boolean statistic column value: Boolean value to set Returns: bool: True if successful, False otherwise N ever_marriedzInvalid boolean stat_name: Fr r,r5r-r r6TzError setting boolean stat r.r r/)rrr4rrZvalid_bool_statsr2rrrrset_boolean_statsR      r9c Csd}d}zzrt}|jdd}|d|f|}|sVt||d|f|}|r^|niWW|rp||r||Sty}zDtjd|d|ddiWYd}~W|r||r|Sd}~00W|r||r|n|r||r|0dS)z Get all statistics for a player. Args: player_id: The player's ID Returns: Dict with all statistic values, or empty dict if not found NT dictionaryz4SELECT * FROM player_statistics WHERE player_id = %sz$Error getting statistics for player r r ) rrrfetchonerrrrr)rrrstatsrrrrget_player_statisticssN   r>)r event_type description snapshot_data character_age game_dater c Cstd}d}zFzt}|jdd}|durN|d|f|}|rJ|dnd}|dur^t}|d|||||t|f|j} | t d|d |d || WW|r| |r| St y0} zR|r|t jd |d | dd WYd} ~ W|r| |r$| dSd} ~ 00W|rB| |rp| n|r`| |rn| 0dS) a= Capture a memorable life moment in the photo album. Args: player_id: The player's ID event_type: Type of event (e.g., 'graduation', 'marriage', 'birth') description: Human-readable description snapshot_data: Dictionary of relevant data to store character_age: Character's age at time of event (optional, will fetch from player if not provided) game_date: Game date/time (optional, will use current datetime if not provided) Returns: int: Photo album entry ID if successful, None otherwise Example: capture_photo_memory( player_id=1, event_type='graduation', description='Graduated from Harvard University', snapshot_data={'level': 'college', 'gpa': 3.8, 'major': 'Computer Science'} ) NTr:z%SELECT age FROM players WHERE id = %sagerzINSERT INTO player_photo_album (player_id, event_type, event_description, game_date, character_age, snapshot_data) VALUES (%s, %s, %s, %s, %s, %s)z!Captured photo memory for player r z - z(Error capturing photo memory for player r )rrrr<rnowjsondumps lastrowidrrrrrrr) rr?r@rArBrCrrplayerZphoto_idrrrrcapture_photo_memory3sT    rJ2)rlimitoffsetr c Cszd}d}zLzt}|jdd}|d|||f|}|D]v}|drzt|d|d<Wntjy~i|d<Yn0|dr|d|d<|dr>|d|d<q>|WW|r| |r| St y6}zHt j d|d |dd gWYd}~W|r| |r,| Sd}~00W|rH| |rv| n|rf| |rt| 0dS) a7 Get photo memories for a player. Args: player_id: The player's ID limit: Maximum number of photos to return (default: 50) offset: Number of photos to skip (for pagination, default: 0) Returns: List of photo memory dictionaries, ordered by game_date descending NTr:zSELECT id, event_type, event_description, game_date, character_age, snapshot_data, created_at FROM player_photo_album WHERE player_id = %s ORDER BY game_date DESC LIMIT %s OFFSET %srArC created_atz%Error getting photo album for player r r rrrfetchallgetrFloadsJSONDecodeError isoformatrrrr)rrLrMrrphotosphotorrrrget_photo_album}sT      rWc Csd}d}zzTt}|jdd}|d|f|}|r@|dndWW|rR||r^|Sty}zBtjd|d|dd WYd}~W|r||r|dSd}~00W|r||r|n|r||r|0dS) z Get total count of photo memories for a player. Args: player_id: The player's ID Returns: int: Total number of photos NTr:zESELECT COUNT(*) as count FROM player_photo_album WHERE player_id = %scountrz+Error getting photo album count for player r r )rrrr<rrrr)rrrresultrrrrget_photo_album_counts@    rZ)rr?r c Csxd}d}zJzt}|jdd}|d||f|}|D]v}|dr~zt|d|d<Wntjy|i|d<Yn0|dr|d|d<|dr<|d|d<q<|WW|r| |r| St y4}zHt j d|d |dd gWYd}~W|r| |r*| Sd}~00W|rF| |rt| n|rd| |rr| 0dS) z Get all photo memories of a specific type for a player. Args: player_id: The player's ID event_type: Type of event to filter by Returns: List of photo memory dictionaries NTr:zSELECT id, event_type, event_description, game_date, character_age, snapshot_data, created_at FROM player_photo_album WHERE player_id = %s AND event_type = %s ORDER BY game_date DESCrArCrNz(Error getting photos by type for player r r rO)rr?rrrUrVrrrrget_photos_by_typesT      r[)rrr cCst|d|dS)z Track money earned by player. Call this whenever player receives money (salary, job, gifts, etc.) Args: player_id: The player's ID amount: Amount of money earned r Nr3rrrrrtrack_money_earneds r^cCst|d|dS)z Track money spent by player. Call this whenever player spends money (purchases, expenses, etc.) Args: player_id: The player's ID amount: Amount of money spent r!Nr\r]rrrtrack_money_spents r_cCst|dddS)z Track new relationship formed. Call this when player forms a new relationship (friend, romantic, family). Args: player_id: The player's ID r"rNr\rrrrtrack_relationship_formed+sracCst|dddS)z Track activity completion. Call this when player completes any activity. Args: player_id: The player's ID r#rNr\r`rrrtrack_activity_completed6srbcCst|dddS)z Track conversation with NPC. Call this when player has a conversation. Args: player_id: The player's ID r$rNr\r`rrrtrack_conversationAsrc)r)NN)rKr)__doc__typingrrrrrrFrdatabaserintboolrstrr3r7r9r>rJrWrZr[r^r_rarbrcrrrrs<  *KC76  J"8$: