a w^i>@sdZddlZddlmZddlZddlZddlZddlZddlZddl m Z ddl m Z e e ZGdddejZdd Zd d Zd d ZddZddZedddZddZddZddZddZddZdS)a Database Operations Module This module contains all database-related functions for the BaoLife game. Extracted from functions.py to provide a centralized database layer. Functions: - connect_to_database(): Create database connection - get_database_connection(): Get or reconnect to database - insertGame(player): Insert/replace game save - saveGameAsync(player): Save game state (async version) - loadGameAsync(user_id): Load game state (async version) - saveGame(player): Save game state (legacy sync version) - loadGame(id): Load game state (sync version) - loadGames(): Load all game IDs - saveConversationMessage(...): Save conversation message - markConversationAsRead(...): Mark conversation as read N)Error)config)ComplexHandlerc s>eZdZdZdddddddd d d d d d ZfddZZS)RefactoringUnpicklera Custom unpickler to handle classes that were moved during refactoring. During the refactoring, classes were moved from functions.py to organized submodules. This unpickler redirects pickle to find classes in their new locations. Class migrations: - functions.playerClass -> core.models.playerClass - functions.personClass -> core.models.personClass - functions.locationClass -> core.models.locationClass - functions.ActivityRecord -> core.models.ActivityRecord - functions.EducationRecord -> core.models.EducationRecord - functions.dailyEvent -> core.models.dailyEvent - functions.oneTimeEvent -> core.models.oneTimeEvent - functions.scheduleDays -> core.models.scheduleDays - functions.scheduler -> core.models.scheduler - functions.relationshipClass -> core.models.relationshipClass - functions.healthCondition -> health.health_manager.HealthCondition - functions.HealthCondition -> health.health_manager.HealthCondition ) core.models playerClass)r personClass)r locationClass)rActivityRecord)rEducationRecord)r dailyEvent)r oneTimeEvent)r scheduleDays)r scheduler)rrelationshipClass)zhealth.health_managerHealthCondition) ) functionsr)rr)rr )rr )rr )rr )rr )rr)rr)rr)rhealthCondition)rrc sV|j||f}|rH|\}}td|d|d|d|||}}t||S)a  Override find_class to redirect old class locations to new ones. Args: module: Original module name from pickle name: Original class name from pickle Returns: The class object from the new location zRedirecting pickle: .z -> ) CLASS_RENAMESgetloggerdebugsuper find_class)selfmodulenameZrenamedZ new_modulenew_name __class__7/var/www/lichun.app/lichun/ws/database/db_operations.pyrPs " zRefactoringUnpickler.find_class)__name__ __module__ __qualname____doc__rr __classcell__r!r!rr"r)src Cstt|}t|drLt|jtrLt dt |jdt |j|_t|drt|j trt dt |j dt |j |_ ddl m}m}m}t|d r|jrt|jd st d ||j_||j_t|d r.d}|jD]&}t|d s||_||_|d 7}q|dkr.t d|dt|drd}|jD]"}t|dsD||_|d 7}qD|dkrt d|d|S)z Load pickled data with refactoring compatibility. Args: data: Pickled byte data Returns: Unpickled object with class references updated to new locations eventsz*Migrating player.events from list to set (z events)askedQuestionsz2Migrating player.askedQuestions from list to set (z questions)r)initialize_messaging_traitsinitialize_messaging_patterns!initialize_relationship_modifierscmessaging_traitsz4Migrating player character - adding messaging traitsrz Migrated z( NPC characters - added messaging traitsrelDatamessaging_modifiersz* relationships - added messaging modifiers)rioBytesIOloadhasattr isinstancer(listrinfolensetr)Zmessaging_styler*r+r,r-r.Zmessaging_patternsr/r1r2) dataplayerr*r+r,Zmigrated_count characterZmigrated_rel_countrelr!r!r"pickle_loads_compatfs>                 r@cCs,tjtjtjtjtjd}tjjfi|S)z Connect to database using configuration from environment variables. NOTE: This uses mysql.connector (sync), which does NOT support pool_timeout. For async connections with pool_timeout, use database_async.initialize_pool() instead. )hostportuserpasswordZdatabase) rDB_HOSTDB_PORTDB_USER DB_PASSWORDDB_NAMEmysqlZ connectorconnect)Z conn_paramsr!r!r"connect_to_databases  rLc CsDztWSty>}ztd|WYd}~n d}~00dS)a Create a new database connection for synchronous operations. IMPORTANT: Caller is responsible for closing the connection. Use with try/finally or context manager pattern. DEPRECATED: Use async pool (database_async.py) for new code. This creates a new connection on each call to avoid module-level connection issues. Returns: mysql.connector.connection: New database connection zFailed to connect to database: N)rLrrerror)er!r!r"get_database_connections rOc Csd}d}z|t}|}t|}d}|j|jj|jj|jj|jj t j|j t d|f}| |||W|rx||r|n|r||r|0dS)z[ Insert/replace game save in database. DEPRECATED: Use saveGameAsync instead. NzREPLACE INTO lifesim_savegames (id, firstname, lastname, ageDays, ageYears, json, pickle_data) VALUES (%s, %s, %s, %s, %s, %s, %s))default)rOcursorpickledumpsidr- firstnamelastnameageDaysageYearsjson__dict__rexecutecommitclose)r=mydbmycursorZserialized_playersqlvalr!r!r" insertGames$ 0   rbc sDddlm}ddl}z|j||t|jdr6|jjndt|jdrL|jjndt|jdrb|jj ndt|jdrx|jj ndt|jd r|jj nd d }d }|||d |d|d|d|d|dfIdHt d|jd|jjd|jjdWdSty>}z*t jd|jd|ddWYd}~dSd}~00dS)z Save game state to database (async version). Args: player: playerClass instance Returns: bool: True if successful r) execute_queryNrUrVrXrWstatusalive)rT pickle_datarUrVrXrWrea INSERT INTO lifesim_savegames (id, pickle_data, firstname, lastname, ageYears, ageDays, json) VALUES (%s, %s, %s, %s, %s, %s, '{}') AS new_data ON DUPLICATE KEY UPDATE pickle_data = new_data.pickle_data, firstname = new_data.firstname, lastname = new_data.lastname, ageYears = new_data.ageYears, ageDays = new_data.ageDays, lastUpdated = CURRENT_TIMESTAMP rTrgz Game saved:  ( )TzFailed to save game for : exc_infoF)database_asyncrcrRrTrSr6r-rUrVrXrWrerr ExceptionrM)r=rcrR player_dataqueryrNr!r!r" saveGameAsyncs6    (rr)user_idc sddlm}ddl}zh|d|fIdH}|durFtd|WdSt|d}td|d|jjd|jjd |WSt y}z(tj d |d |d d WYd}~dSd}~00dS)z Load game state from database (async version). Args: user_id: User ID to load Returns: playerClass instance or None if not found r) fetch_oneN7SELECT pickle_data FROM lifesim_savegames WHERE id = %szNo saved game found for z Game loaded: rhrirjzFailed to load game for rkTrl) rnrtrRrrr@r-rUrVrorM)rsrtrRresultr=rNr!r!r" loadGameAsyncs   &rwc Csddl}d}d}zLzt}|}|j||t|jdrD|jjndt|jdrZ|jjndt|jdrp|jj ndt|jdr|jj ndd }d }| ||d |d |d|d|d |df| WW|r| |r| dSty>}ztd|d|WYd}~W|r||r|dSd}~00W|r ||r8|n|r(||r6|0dS)zs Load game using parameterized query to prevent SQL injection. DEPRECATED: Use loadGameAsync instead. NrurFzError loading game rk) rOrQr[fetchallr@r]rorrM)rTr^r_r`myresultr=rNr!r!r"loadGamevsL  r|c Csd}d}zzlt}|}d}|||}|rT|WW|rF||rR|SWW|rd||rp|dSWnPty}z8td|WYd}~W|r||r|dSd}~00W|r||r|n|r||r|0dS)zM Load all game IDs. DEPRECATED: Create async version if needed. Nz SELECT id FROM lifesim_savegamesFzError loading games: )rOrQr[rzr]rorrM)r^r_stringr{rNr!r!r" loadGamessJ   r~c Csd}zrt}|D}tj}d}||||||jf} ||| |Wdn1s^0YW|r|n|r|0dS)zg Save a conversation message to the database. DEPRECATED: Create async version if needed. NzaINSERT INTO messages (id, partner, player, message, sender, date) VALUES (%s, %s, %s, %s, %s, %s)) rOrQuuiduuid4hexdater[r\r]) messager>ZplayerIDsenderr=r^r_rTr`rar!r!r"saveConversationMessages   ( rcCs6tdt|jD] }|j|j|krd|j|_qdS)NrFT)ranger: conversationsr>unread)r= characterIDir!r!r"markConversationAsReadsr)r&Zmysql.connectorrJrrRrYrloggingr3rserver.websocket_messagingr getLoggerr#rZ Unpicklerrr@rLrOrbrrstrrwryr|r~rrr!r!r!r"s,    =7;#: