a i @sUddlmZddlmZmZmZddlmZmZddl m Z ddl m Z ddl mZddlmZeejdd Zeed <e eedd Zeeed d ddZd dddZd dddZd dddZeed fdddZd S))AsyncGenerator)deletetextupdate) AsyncEnginecreate_async_engine) sessionmaker)SQLModel) AsyncSession)settingsF)echoengine)class_Zexpire_on_commitN) table_name column_name column_sqlreturncsV|d|dIdH}dd|D}||vr6dS|d|d|IdHdS)NzPRAGMA table_info('z')cSsh|]}t|dqS))str).0rowrz)_add_column_if_missing..z ALTER TABLE z ADD COLUMN )Zexec_driver_sqlZfetchall)connrrrrowsZexisting_columnsrrr_add_column_if_missings r)rcsttjdsdSt|dddIdHt|dddIdHt|dddIdHt|d d d IdHt|d d d IdHt|d ddIdHt|d ddIdHt|d ddIdHt|d ddIdH|tdIdH|tdIdH|tdIdHdS)NZsqlitepoll winner_sourcezwinner_source TEXT DEFAULT ''Z config_jsonzconfig_json TEXT DEFAULT '{}' runtime_jsonzruntime_json TEXT DEFAULT '{}'ZlineeventrecordZraw_jsonzraw_json TEXT DEFAULT ''room_idzroom_id TEXT DEFAULT ''Zparticipant_idzparticipant_id TEXT DEFAULT ''Zline_user_id_hashz!line_user_id_hash TEXT DEFAULT ''Z reply_statuszreply_status TEXT DEFAULT ''Z reply_errorzreply_error TEXT DEFAULT ''zRCREATE INDEX IF NOT EXISTS ix_lineeventrecord_room_id ON lineeventrecord (room_id)z`CREATE INDEX IF NOT EXISTS ix_lineeventrecord_participant_id ON lineeventrecord (participant_id)zfCREATE INDEX IF NOT EXISTS ix_lineeventrecord_line_user_id_hash ON lineeventrecord (line_user_id_hash))rr database_url startswithrZexecuter)rrrr_run_sqlite_migrationss,  r$c snddlm}t4IdH8}|tjjIdHt|IdHWdIdHqj1IdHs`0YdS)Nr)entities) Z app.modelsr%r beginrun_syncr metadataZ create_allr$)r%rrrrinit_db9s r)c sddlm}m}m}m}m}m}m}t4IdH}| t |IdH| t |IdH| t |IdH| t |IdH| t |IdH| t |j |j dddddIdH|IdHWdIdHq1IdHs0YdS)Nr)AudienceSession JoinTokenLineEventRecord ParticipantPoll PollStatusVotez{})statusZ winner_optionrZ deadline_atr )Zapp.models.entitiesr*r+r,r-r.r/r0async_session_factoryexecrrvaluesZdraftcommit)r*r+r,r-r.r/r0sessionrrrreset_runtime_stateAs"$ r8c CsDt4IdH}|VWdIdHq@1IdHs60YdS)N)r3)r7rrr get_sessionVsr9)collections.abcrZ sqlalchemyrrrZsqlalchemy.ext.asynciorrZsqlalchemy.ormrZsqlmodelr Zsqlmodel.ext.asyncio.sessionr app.configr r"r __annotations__r3rrr$r)r8r9rrrrs(