XZ @s5ddlZddlZddlZddlmZddlmZddlmZddl m Z ddl m Z ddl mZddlmZdd lmZdd lmZmZdd lmZdd lmZdd lmZyddlZWnek rdZYnXdZGddde Z!dS)N)deque)contextmanager)settings)ImproperlyConfigured)DEFAULT_DB_ALIAS)utils)connection_created)TransactionManagementError) DatabaseErrorDatabaseErrorWrapper)timezone)cached_property)_threadZ __no_db__c@seZdZdZiZiZiZdZdZdZ dZ e dddZ e dd Ze d d Zed d ZeddZddZddZddZddZddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*d+Z d,d-Z!d.d/Z"d0d1Z#d2d3Z$d4d5Z%d6d7Z&d8d9Z'd:d;Z(d<d=Z)d>d?Z*d@dAZ+ddBdCZ,dDdEZ-dFdGZ.dHdIZ/dJdKZ0e1dLdMZ2dNdOZ3dPdQZ4ddRdSZ5dTdUZ6dVdWZ7dXdYZ8dZd[Z9e d\d]Z:d^d_Z;d`daZ<e1dbdcZ=edddeZ>dfdgZ?dhdiZ@djdkZAdldmZBdddndoZCdS)pBaseDatabaseWrapperz+ Represents a database connection. Nunknowni(#FcCsd|_||_||_td|j|_d|_d|_d|_d|_ g|_ d|_ d|_ d|_ d|_d|_||_tj|_g|_d|_dS)NmaxlenFrT) connection settings_dictaliasr queries_limit queries_logforce_debug_cursor autocommitin_atomic_blocksavepoint_state savepoint_idsZcommit_on_exitneeds_rollbackclose_atclosed_in_transactionerrors_occurredallow_thread_sharingthread get_ident _thread_ident run_on_commit%run_commit_hooks_on_set_autocommit_on)selfrrr r'J/home/ubuntu/projects/ifolica/build/django/django/db/backends/base/base.py__init__*s$               zBaseDatabaseWrapper.__init__cCsOtjs dS|jjrdS|jddkr7tjStj|jdSdS)a_ Time zone for datetimes stored as naive values in the database. Returns a tzinfo object or None. This is only needed when time zone support is enabled and the database doesn't support time zones. (When the database supports time zones, the adapter handles aware datetimes so Django doesn't need to.) N TIME_ZONE)rUSE_TZfeaturessupports_timezonesrr utcpytz)r&r'r'r(r [s  zBaseDatabaseWrapper.timezonecCs6tjstjS|jddkr'dS|jdSdS)zC Name of the time zone of the database connection. r*NUTC)rr+r*r)r&r'r'r( timezone_nameps  z!BaseDatabaseWrapper.timezone_namecCs|jptjS)N)rrDEBUG)r&r'r'r(queries_logged|sz"BaseDatabaseWrapper.queries_loggedcCsGt|j|jjkr:tjdj|jjnt|jS)NzLLimit for query logging exceeded, only the last {} queries will be returned.)lenrrwarningswarnformatlist)r&r'r'r(queriess zBaseDatabaseWrapper.queriescCstddS)z=Returns a dict of parameters suitable for get_new_connection.zNsubclasses of BaseDatabaseWrapper may require a get_connection_params() methodN)NotImplementedError)r&r'r'r(get_connection_paramssz)BaseDatabaseWrapper.get_connection_paramscCstddS)z#Opens a connection to the database.zKsubclasses of BaseDatabaseWrapper may require a get_new_connection() methodN)r:)r& conn_paramsr'r'r(get_new_connectionsz&BaseDatabaseWrapper.get_new_connectioncCstddS)z-Initializes the database connection settings.zOsubclasses of BaseDatabaseWrapper may require an init_connection_state() methodN)r:)r&r'r'r(init_connection_statesz)BaseDatabaseWrapper.init_connection_statecCstddS)z;Creates a cursor. Assumes that a connection is established.zFsubclasses of BaseDatabaseWrapper may require a create_cursor() methodN)r:)r&r'r'r( create_cursorsz!BaseDatabaseWrapper.create_cursorcCs|jd|_g|_d|_|jd}|dkrDdn tj||_d|_d|_|j }|j ||_ |j |jd|j tjd|jd|g|_dS)z@Connects to the database. Assumes that the connection is closed.FZ CONN_MAX_AGEN AUTOCOMMITZsenderr)check_settingsrrrrtimerrrr;r=rset_autocommitr>rsend __class__r$)r&Zmax_ager<r'r'r(connects     %    zBaseDatabaseWrapper.connectcCs}|jddk rytjs2td|jqy|jjrTtd|jqytdkrytd|jqyndS)Nr*z=Connection '%s' cannot set TIME_ZONE because USE_TZ is False.z`Connection '%s' cannot set TIME_ZONE because its engine handles time zones conversions natively.zBConnection '%s' cannot set TIME_ZONE because pytz isn't installed.)rrr+rrr,r-r/)r&r'r'r(rAs   z"BaseDatabaseWrapper.check_settingscCs0|jdkr,|j|jWdQXndS)zN Guarantees that a connection to the database is established. N)rwrap_database_errorsrF)r&r'r'r(ensure_connections z%BaseDatabaseWrapper.ensure_connectioncCs(|j|j|jSWdQXdS)N)rHrGr?)r&r'r'r(_cursors  zBaseDatabaseWrapper._cursorcCs3|jdk r/|j|jjSWdQXndS)N)rrGcommit)r&r'r'r(_commits zBaseDatabaseWrapper._commitcCs3|jdk r/|j|jjSWdQXndS)N)rrGrollback)r&r'r'r( _rollbacks zBaseDatabaseWrapper._rollbackcCs3|jdk r/|j|jjSWdQXndS)N)rrGclose)r&r'r'r(_closes zBaseDatabaseWrapper._closecCsD|j|jr+|j|j}n|j|j}|S)zF Creates a cursor, opening a connection if necessary. )validate_thread_sharingr3make_debug_cursorrI make_cursor)r&cursorr'r'r(rSs   zBaseDatabaseWrapper.cursorcCs4|j|j|jd|_d|_dS)zB Commits a transaction and resets the dirty flag. FTN)rPvalidate_no_atomic_blockrKrr%)r&r'r'r(rJs     zBaseDatabaseWrapper.commitcCs4|j|j|jd|_g|_dS)zE Rolls back a transaction and resets the dirty flag. FN)rPrTrMrr$)r&r'r'r(rLs     zBaseDatabaseWrapper.rollbackc Csl|jg|_|js+|jdkr/dSz|jWd|jr^d|_d|_n d|_XdS)z8 Closes the connection to the database. NT)rPr$rrrOrr)r&r'r'r(rNs     zBaseDatabaseWrapper.closec Cs2|j }|j|jj|WdQXdS)N)rSexecuteopsZsavepoint_create_sql)r&sidrSr'r'r( _savepointszBaseDatabaseWrapper._savepointc Cs2|j }|j|jj|WdQXdS)N)rSrUrVZsavepoint_rollback_sql)r&rWrSr'r'r(_savepoint_rollbacksz'BaseDatabaseWrapper._savepoint_rollbackc Cs2|j }|j|jj|WdQXdS)N)rSrUrVZsavepoint_commit_sql)r&rWrSr'r'r(_savepoint_commit"sz%BaseDatabaseWrapper._savepoint_commitcCs|jjo|j S)N)r,Zuses_savepointsget_autocommit)r&r'r'r(_savepoint_allowed&sz&BaseDatabaseWrapper._savepoint_allowedcCsq|jsdStj}t|jdd}|jd7_d||jf}|j|j||S)z Creates a savepoint inside the current transaction. Returns an identifier for the savepoint that will be used for the subsequent rollback or commit. Does nothing if savepoints are not supported. N-zs%s_x%d)r\r!r"strreplacerrPrX)r&Z thread_identtidrWr'r'r( savepoint,s    zBaseDatabaseWrapper.savepointcsJ|jsdS|j|jfdd|jD|_dS)zZ Rolls back to a savepoint. Does nothing if savepoints are not supported. Ncs.g|]$\}}|kr||fqSr'r').0sidsfunc)rWr'r( Ls z:BaseDatabaseWrapper.savepoint_rollback..)r\rPrYr$)r&rWr')rWr(savepoint_rollback@s    z&BaseDatabaseWrapper.savepoint_rollbackcCs+|jsdS|j|j|dS)zU Releases a savepoint. Does nothing if savepoints are not supported. N)r\rPrZ)r&rWr'r'r(savepoint_commitOs  z$BaseDatabaseWrapper.savepoint_commitcCs d|_dS)zZ Resets the counter used to generate unique savepoint ids in this thread. rN)r)r&r'r'r(clean_savepointsYsz$BaseDatabaseWrapper.clean_savepointscCstddS)zR Backend-specific implementation to enable or disable autocommit. zHsubclasses of BaseDatabaseWrapper may require a _set_autocommit() methodN)r:)r&rr'r'r(_set_autocommitasz#BaseDatabaseWrapper._set_autocommitcCs|j|jS)z- Check the autocommit state. )rHr)r&r'r'r(r[is z"BaseDatabaseWrapper.get_autocommitcCs|j|j|o*| o*|jj}|r@|jn |j|||_|r{|jr{|jd|_ndS)a Enable or disable autocommit. The usual way to start a transaction is to turn autocommit off. SQLite does not properly start a transaction when disabling autocommit. To avoid this buggy behavior and to actually enter a new transaction, an explcit BEGIN is required. Using force_begin_transaction_with_broken_autocommit=True will issue an explicit BEGIN with SQLite. This option will be ignored for other backends. FN) rTrHr,Z"autocommits_when_autocommit_is_off#_start_transaction_under_autocommitrkrr%run_and_clear_commit_hooks)r&rZ.force_begin_transaction_with_broken_autocommitZ"start_transaction_under_autocommitr'r'r(rCps        z"BaseDatabaseWrapper.set_autocommitcCs|jstdn|jS)zK Get the "needs rollback" flag -- for *advanced use* only. z Raise an error if an atomic block is active. z3This is forbidden when an 'atomic' block is active.N)rr )r&r'r'r(rTs z,BaseDatabaseWrapper.validate_no_atomic_blockcCs|jrtdndS)NzlAn error occurred in the current transaction. You can't execute queries until the end of the 'atomic' block.)rr )r&r'r'r(validate_no_broken_transactions z2BaseDatabaseWrapper.validate_no_broken_transactionc cs0|j}z dVWd|r+|jnXdS)zP Context manager that disables foreign key constraint checking. N)disable_constraint_checkingenable_constraint_checking)r&disabledr'r'r(constraint_checks_disableds   z.BaseDatabaseWrapper.constraint_checks_disabledcCsdS)z Backends can implement as needed to temporarily disable foreign key constraint checking. Should return True if the constraints were disabled and will need to be reenabled. Fr')r&r'r'r(rqsz/BaseDatabaseWrapper.disable_constraint_checkingcCsdS)zh Backends can implement as needed to re-enable foreign key constraint checking. Nr')r&r'r'r(rrsz.BaseDatabaseWrapper.enable_constraint_checkingcCsdS)z Backends can override this method if they can apply constraint checking (e.g. via "SET CONSTRAINTS ALL IMMEDIATE"). Should raise an IntegrityError if any invalid foreign key references are encountered. Nr')r&Z table_namesr'r'r(check_constraintssz%BaseDatabaseWrapper.check_constraintscCstddS)a Tests if the database connection is usable. This function may assume that self.connection is not None. Actual implementations should take care not to raise exceptions as that may prevent Django from recycling unusable connections. zCsubclasses of BaseDatabaseWrapper may require an is_usable() methodN)r:)r&r'r'r( is_usables zBaseDatabaseWrapper.is_usablecCs|jdk r|j|jdkr6|jdS|jrh|jrWd|_qh|jdSn|jdk rtj|jkr|jdSndS)z Closes the current connection if unrecoverable errors have occurred, or if it outlived its maximum age. Nr@F)rr[rrNrrvrrB)r&r'r'r(close_if_unusable_or_obsoletes     $ z1BaseDatabaseWrapper.close_if_unusable_or_obsoletecCsJ|jp|jtjksFtd|j|jtjfndS)a7 Validates that the connection isn't accessed by another thread than the one which originally created it, unless the connection was explicitly authorized to be shared between threads (via the `allow_thread_sharing` property). Raises an exception if the validation fails. zDatabaseWrapper objects created in a thread can only be used in that same thread. The object with alias '%s' was created in thread id %s and this is thread id %s.N)r r#r!r"r r)r&r'r'r(rPsz+BaseDatabaseWrapper.validate_thread_sharingcCsdS)z~ Hook to do any database check or preparation, generally called before migrating a project or an app. Nr')r&r'r'r(prepare_databasesz$BaseDatabaseWrapper.prepare_databasecCs t|S)z Context manager and decorator that re-throws backend-specific database exceptions using Django's common wrappers. )r )r&r'r'r(rGsz(BaseDatabaseWrapper.wrap_database_errorscCstj||S)zM Creates a cursor that logs all queries in self.queries_log. )rZCursorDebugWrapper)r&rSr'r'r(rQsz%BaseDatabaseWrapper.make_debug_cursorcCstj||S)z9 Creates a cursor without debug logging. )rZ CursorWrapper)r&rSr'r'r(rR$szBaseDatabaseWrapper.make_cursorc csI|jdk}|j}z |VWd|j|rD|jnXdS)a9 Context manager that ensures that a connection is established, and if it opened one, closes it to avoid leaving a dangling connection. This is useful for operations outside of the request-response cycle. Provides a cursor: with self.temporary_connection() as cursor: ... N)rrSrN)r&Z must_closerSr'r'r(temporary_connection*s    z(BaseDatabaseWrapper.temporary_connectioncCs8|jj}d|d<|j|dtdd}|S)ac Return an alternative connection to be used when there is no need to access the main database, specifically for test db creation/deletion. This also prevents the production database from being exposed to potential child threads while (or after) the test database is destroyed. Refs #10868, #17786, #16969. NNAMErr F)rcopyrE NO_DB_ALIAS)r&rZnodb_connectionr'r'r(_nodb_connection<s   z$BaseDatabaseWrapper._nodb_connectioncCstddS)zO Only required when autocommits_when_autocommit_is_off = True. z\subclasses of BaseDatabaseWrapper may require a _start_transaction_under_autocommit() methodN)r:)r&r'r'r(rlMsz7BaseDatabaseWrapper._start_transaction_under_autocommitcOs1|jdkrtdn|j|||S)zH Returns a new instance of this backend's SchemaEditor. NzFThe SchemaEditorClass attribute of this database wrapper is still None)SchemaEditorClassr:)r&argskwargsr'r'r( schema_editorVs z!BaseDatabaseWrapper.schema_editorcCsQ|jr+|jjt|j|fn"|jsFtdn|dS)Nz;on_commit() cannot be used in manual transaction management)rr$appendsetrr[r )r&rfr'r'r( on_commit_s  " zBaseDatabaseWrapper.on_commitcCsI|j|j}g|_x&|rD|jd\}}|qWdS)Nr)rTr$pop)r&Zcurrent_run_on_commitrerfr'r'r(rmjs     z.BaseDatabaseWrapper.run_and_clear_commit_hookscCsXtj|j}|dkr*|j}n|dkrB|j}nt||||S)zy Return a copy of this connection. For tests that require two connections to the same database. N)r{deepcopyrrr type)r&rr rr'r'r(r{rs     zBaseDatabaseWrapper.copy)D__name__ __module__ __qualname____doc__Z data_typesZdata_types_suffixZdata_type_check_constraintsrVvendorr~rrr)r r r1propertyr3r9r;r=r>r?rFrArHrIrKrMrOrSrJrLrNrXrYrZr\rcrhrirjrkr[rCrnrorTrprrtrqrrrurvrwrPrxrGrQrRryr}rlrrrmr{r'r'r'r(rs| 0                               r)"r{rBr5 collectionsr contextlibrZ django.confrZdjango.core.exceptionsrZ django.dbrZdjango.db.backendsrZdjango.db.backends.signalsrZdjango.db.transactionr Zdjango.db.utilsr r Z django.utilsr Zdjango.utils.functionalr Zdjango.utils.six.movesrr!r/ ImportErrorr|objectrr'r'r'r(s(