o JhK@sddlmZddlmZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z dd lZdd lZdd lZdd lZdd lZdd lZdd lZdd lmZdd lmZdd lmZdd lmZe dZGdddZGdddeZGdddZddZ ddZ!d S))abc)datetime)timezone)contextmanager import_module)count) getLogger) entry_pointsN) exceptions)internalmigrations)utils)topological_sortzyoyo.migrationsc@sBeZdZdZdddZddZddZd d Zd d Zd dZ dS)TransactionManagerz Returned by the :meth:`~yoyo.backends.DatabaseBackend.transaction` context manager. If rollback is called, the transaction is flagged to be rolled back when the context manager block closes FcCs||_||_dSN)backendrollback_on_exit)selfrrrT/var/www/html/pro-man-master/venv/lib/python3.10/site-packages/yoyo/backends/base.py__init__1s zTransactionManager.__init__cCs ||Sr) _do_beginrrrr __enter__5szTransactionManager.__enter__cCs.|r|dS|jr|dS|dSr) _do_rollbackr _do_commit)rexc_typevalue tracebackrrr__exit__9s   zTransactionManager.__exit__cC|jdS)z= Instruct the backend to begin a transaction N)rbeginrrrrrCzTransactionManager._do_begincCr )z@ Instruct the backend to commit the transaction N)rcommitrrrrrIr"zTransactionManager._do_commitcCr )zC Instruct the backend to roll back the transaction N)rrollbackrrrrrOr"zTransactionManager._do_rollbackNF) __name__ __module__ __qualname____doc__rrrrrrrrrrr(s   rc@s0eZdZdZedZddZddZddZdS) SavepointTransactionManagerNcCs2|jdusJdt|j|_|j|jdS)Nzsp_{})idformatnext id_generatorr savepointrrrrrZsz%SavepointTransactionManager._do_begincCdS)z This does nothing. Trying to the release savepoint here could cause an database error in databases where DDL queries cause the transaction to be committed and all savepoints released. Nrrrrrr_z&SavepointTransactionManager._do_commitcCs|j|jdSr)rsavepoint_rollbackr,rrrrrhsz(SavepointTransactionManager._do_rollback) r&r'r(r,rr/rrrrrrrr*Vs  r*c@s.eZdZUdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZiZeeefed<ddZddZddZddZeddZeddZddZ dd Z!d!d"Z"d#d$Z#d%d&Z$d'd(Z%d)d*Z&d+d,Z'd-e(e)fd.d/Z*dud0d1Z+d2d3Z,d4d5Z-d6d7Z.d8d9Z/d:d;Z0dd?Z2e3d@dAZ4e3dvdCdDZ5dwdFdGZ6dHdIZ7dJdKZ8dxdLe9j:e;jdOdPZ?dQdRZ@dSdTZAdUdVZBdWdXZCdYdZZDdud[d\ZEdud]d^ZFdud_d`ZGdudadbZHdcddZIdedfZJdydhdiZKdudjdkZLdzdldmZMdzdndoZNdpdqZOd{dsdtZPdS)|DatabaseBackend _yoyo_log yoyo_lockz0SELECT table_name FROM information_schema.tables _yoyo_version_yoyo_migrationszW SELECT COUNT(1) FROM {0.migration_table_quoted} WHERE {quoted.id}=:id}zINSERT INTO {0.migration_table_quoted} ({quoted.migration_hash}, {quoted.migration_id}, {quoted.applied_at_utc}) VALUES (:migration_hash, :migration_id, :when)zVDELETE FROM {0.migration_table_quoted} WHERE {quoted.migration_hash} = :migration_hashz_SELECT {quoted.migration_hash} FROM {0.migration_table_quoted} ORDER by {quoted.applied_at_utc}z>CREATE TABLE {table_name_quoted} ({quoted.id} INT PRIMARY KEY)aINSERT INTO {0.log_table_quoted} ({quoted.id}, {quoted.migration_hash}, {quoted.migration_id}, {quoted.operation}, {quoted.username}, {quoted.hostname}, {quoted.created_at_utc}) VALUES (:id, :migration_hash, :migration_id, :operation, :username, :hostname, :created_at_utc)zCREATE TABLE {0.lock_table_quoted} ({quoted.locked} INT DEFAULT 1, {quoted.ctime} TIMESTAMP,{quoted.pid} INT NOT NULL,PRIMARY KEY ({quoted.locked}))NF_transactional_ddl_cachecCsJ||_|jj|_|||_||j||_|jt |jd|_ dS)NT) uridriver DatabaseErrorconnect _connectioninit_connectionmigration_tabler:getpickledumpshas_transactional_ddl)rdburirArrrrs     zDatabaseBackend.__init__cCs*|||_|j|jt|j<dSr)create_lock_table_check_transactional_ddlrEr:rCrDr;rrrr init_databases   zDatabaseBackend.init_databasecCst|j}t|j|S)zT Load the dbapi driver module and register the base exception class )get_dbapi_module driver_moduler registerr=)rr<rrr_load_driver_modules  z#DatabaseBackend._load_driver_modulec s0|jGfddd}|j|fd|i|S)zt Take a string in the format used by the various ``..._sql`` class variables and format it. cseZdZfddZdS)z*DatabaseBackend.format_sql..Quotercs|Srr)rsquote_identifierrr __getattr__sz6DatabaseBackend.format_sql..Quoter.__getattr__N)r&r'r(rQrrOrrQuotersrRquoted)rPr-)rrNkwargsrRrrOr format_sqlszDatabaseBackend.format_sqlcCs|jr|jS||_|jSr)_driverrMrrrrr<s zDatabaseBackend.drivercCs|jSr)r?rrrr connectionszDatabaseBackend.connectioncCr1)z Called when creating a connection or after a rollback. May do any db specific tasks required to make the connection ready for use. Nr)rrWrrrr@r2zDatabaseBackend.init_connectioncCs||j|jS)zX Return a copy of the backend with a independent db connection. ) __class__r;rArrrrcopyszDatabaseBackend.copycCs|SrrrrrrrszDatabaseBackend.__enter__cCs|jdSr)rWclose)rr exc_valuerrrrrzDatabaseBackend.__exit__cCs0|drt||dd}||St|)N_quotedr)endswithgetattrrsplitrPAttributeError)rattrnameunquotedrrrrQs  zDatabaseBackend.__getattr__cCstr)NotImplementedError)rrFrrrr>szDatabaseBackend.connectcCs$d|vsJ|dd}d|dS)N"z"")replace)rrNrSrrrrPs   z DatabaseBackend.quote_identifiercCsdtd}||}|j|j|d}z|jdd ||Wdn1s,wYWn |jy=YdSwz!||d|WdWdS1sXwYWdS|jyjYdSw) zz Return True if the database supports committing/rolling back DDL statements within a transaction z yoyo_tmp_{} )table_name_quotedTrNFz DROP TABLE ) r-r get_random_stringrPrUcreate_test_table_sql transactionexecuter=)r table_namerisqlrrrrHs.    z(DatabaseBackend._check_transactional_ddlreturncKs4||jtd|jjifi|}dd|DS)z Return a list of tables present in the backend. This is used by the test suite to clean up tables generated during testing databasecSg|]}|dqSrr.0rowrrr z/DatabaseBackend.list_tables..)rnlist_tables_sqldictr;rrfetchall)rrTcursorrrr list_tables s zDatabaseBackend.list_tablescCs|js t||dSt||dS)Nrj)_in_transactionrr*)rrrrrrms  zDatabaseBackend.transactioncCs |jSr)rWr}rrrrr}s zDatabaseBackend.cursorcCs|jd|_dSNF)rWr#rrrrrr#s  zDatabaseBackend.commitcCs |j||jd|_dSr)rWr$r@rrrrrr$#s   zDatabaseBackend.rollbackcCs|jrJd|_|ddS)z) Begin a new transaction TBEGINN)rrnrrrrr!(s zDatabaseBackend.begincC|d||dS)z: Create a new savepoint with the given id z SAVEPOINT NrnrPrr,rrrr00zDatabaseBackend.savepointcCr)zB Release (commit) the savepoint with the given id zRELEASE SAVEPOINT Nrrrrrsavepoint_release6rz!DatabaseBackend.savepoint_releasecCr)z: Rollback the savepoint with the given id zROLLBACK TO SAVEPOINT Nrrrrrr3<rz"DatabaseBackend.savepoint_rollbackccs|dVdS)z Disable the connection's transaction support, for example by setting the isolation mode to 'autocommit' N)r$rrrrdisable_transactionsBs z$DatabaseBackend.disable_transactionsrhc csV|jr dVdSt}|||zd|_dVd|_W||dS||w)z Create a lock to prevent concurrent migrations. :param timeout: duration in seconds before raising a LockTimeout error. NTF) _is_lockedosgetpid_insert_lock_row_delete_lock_row)rtimeoutpidrrrlockKs zDatabaseBackend.lock?c Cst||}t}|j} z>|/|d|jd|dd|dd|dd ttj j dd |d WdWdS1sDwYWdS|j y|rt||kr|d |dd |j}| }|r|t d |ddt dt|Ynwq )NTz% INSERT INTO z (lockedz, ctimerzJ) VALUES (1, :when, :pid) tzinfo)whenrzSELECT z FROM zProcess rzC has locked this database (run yoyo break-lock to remove this lock)z9Database locked (run yoyo break-lock to remove this lock))mintimerPrmrnlock_table_quotedrnowrutcrgr=fetchoner LockTimeoutsleep)rrr poll_intervalstartedqir}rwrrrr_sP   z DatabaseBackend._insert_lock_rowcCsX||j}|d|jd|ddd|iWddS1s%wYdS)N DELETE FROM z WHERE rz=:pid)rmrPrnr)rrrrrrrs "z DatabaseBackend._delete_lock_rowcCs@||d|jWddS1swYdS)Nr)rmrnrrrrr break_locks "zDatabaseBackend.break_lockparamscCs.|}t|jj||\}}||||S)a' Create a new cursor, execute a single statement and return the cursor object. :param sql: A single SQL statement, optionally with named parameters (eg 'SELECT * FROM foo WHERE :bar IS NULL') :param params: A dictionary of parameters )r}r change_param_styler< paramstylern)rrprr} queryparamsrrrrns  zDatabaseBackend.executecCs\z"||||jWdWdS1swYWdS|jy-YdSw)zE Create the lock table if it does not already exist. N)rmrnrUcreate_lock_table_sqlr=rrrrrGs &z!DatabaseBackend.create_lock_tablecCsj|jrdSt|r3|jrJ|t||jd|_WddS1s,wYdSdS)z; Check and upgrade yoyo's internal schema. NT)_internal_schema_updatedr needs_upgradingrrupgraderWr#rrrrensure_internal_schema_updateds     "z.DatabaseBackend.ensure_internal_schema_updatedcCs|j|vSr)hashget_applied_migration_hashes)r migrationrrr is_appliedr\zDatabaseBackend.is_appliedcCs,|||j}dd||DS)ze Return the list of migration hashes in the order in which they were applied cSrsrtrrurrrrxryz@DatabaseBackend.get_applied_migration_hashes..)rrUapplied_migrations_sqlrnr|)rrprrrrs z,DatabaseBackend.get_applied_migration_hashescs,|fdd|D}|t||jS)zF Return the subset of migrations not already applied. c3s|] }|jvr|VqdSrrrvmappliedrr z+DatabaseBackend.to_apply..)rrXr post_applyr migrationsmsrrrto_applyszDatabaseBackend.to_applycs4|fdd|D}|ttt||jS)z Return the subset of migrations already applied and which may be rolled back. The order of migrations will be reversed. c3s|] }|jvr|VqdSrrrrrrrrz.DatabaseBackend.to_rollback..)rrXreversedlistrrrrrr to_rollbacks zDatabaseBackend.to_rollbackcCs(|r|j||d|j||ddSdS)Nforce)apply_migrations_onlyrun_post_apply)rrrrrrapply_migrationssz DatabaseBackend.apply_migrationsc Cs<|sdS|D]}z |j||dWqtjyYqwdS)zd Apply the list of migrations, but do not run any post-apply hooks present. Nr) apply_oner BadMigrationrrrrrrrrsz%DatabaseBackend.apply_migrations_onlycCs |jD] }|j|d|dqdS)zI Run any post-apply migrations present in ``migrations`` F)markrN)rrrrrrrs zDatabaseBackend.run_post_applyc CsB||sdS|D]}z|||Wq tjyYq wdSr)r rollback_oner rrrrrrollback_migrationssz#DatabaseBackend.rollback_migrationsc Cb|||D]}z||Wq tjyYq wWddS1s*wYdSr)rrmmark_oner rrrrrrrmark_migrations "zDatabaseBackend.mark_migrationsc Crr)rrm unmark_oner rrrrrunmark_migrationsrz!DatabaseBackend.unmark_migrationsTcCstd|j||}|j|d|dWdn1s"wY||d|rM||j|ddWddS1sFwYdSdS)z* Apply a single migration z Applying %sapplyrNFlog) loggerinfor,rrY process_steps log_migrationrmr)rrrrmigration_backendrrrr s   "zDatabaseBackend.apply_onecCstd|j||}|j|d|dWdn1s"wY||d||j|ddWddS1sDwYdS)z- Rollback a single migration zRolling back %sr$rNFr) rrr,rrYrrrmr)rrrrrrrrs   "zDatabaseBackend.rollback_onecCs>|||j}||d|ji|r||ddSdS)Nmigration_hashunmark)rrUunmark_migration_sqlrnrrrrrrprrrr$s  zDatabaseBackend.unmark_onecCsb|td|j||j}|||j|jt t j j ddd|r/| |ddSdS)NzMarking %s appliedr)r migration_idrr)rrrr,rUmark_migration_sqlrnrrrrrrgrrrrrr+s zDatabaseBackend.mark_onecCs$||j}|||||dSr)rUlog_migration_sqlrn get_log_data)rr operationrprrrr:s zDatabaseBackend.log_migrationrcCsT|dvsJtt|r|jnd|r|jndttt t j j dd|dS)zR Return a dict of data for insertion into the ``_yoyo_log`` table >rrrr$Nr)r,rrusernamehostnamecreated_at_utcr)struuiduuid1r,rgetpassgetusersocketgetfqdnrrrrrg)rrrrrrr>s    zDatabaseBackend.get_log_datar%)rh)rr)FT)T)Nr)Qr&r'r(rK log_table lock_tablerz version_tablerAis_applied_sqlrrrrlrrrVrrrr:r{bytesbool__annotations__rrIrMrUpropertyr<rWr@rYrrrQr>rPrHrrr~rmr}r#r$r!r0rr3rrrrrrtUnionrMappingAnyrnrGrrrrrrrrrrrrrrrrrrrrrr4ls         %&       r4cCstdd}||S)Nz yoyo.backends)group)r load)name backend_epsrrrget_backend_classNs  rcCst|S)z: Import and return the named DB-API driver module r)rrrrrJSsrJ)" collectionsrrr contextlibr importlibr itertoolsrloggingr importlib_metadatar rrrCrrtypingrryoyor r r yoyo.migrationsrrrr*r4rrJrrrrs8            .e