v^r!@s*ddlZddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z ddl m Z ejdZGdddZGd d d eZd d Zd dZddZddZddddZddZddZddZdS)N)contextmanager)settings)NotSupportedError)utczdjango.db.backendsc@seZdZddZeddddgZddZd d Zd d Zd dZ ddddZ dddZ ddZ ddZ ddZddZdS) CursorWrappercCs||_||_dS)N)cursordb)selfrrr :/tmp/pip-build-8lau8j11/django/django/db/backends/utils.py__init__s zCursorWrapper.__init__ZfetchoneZ fetchmanyZfetchallZnextsetcCs9t|j|}|tjkr1|jj|S|SdS)N)getattrrrWRAP_ERROR_ATTRSrwrap_database_errors)r attrZ cursor_attrr r r __getattr__szCursorWrapper.__getattr__c cs$|jj|jEdHWdQRXdS)N)rrr)r r r r __iter__s zCursorWrapper.__iter__cCs|S)Nr )r r r r __enter__"szCursorWrapper.__enter__c Cs0y|jWn|jjjk r+YnXdS)N)closerZDatabaseError)r typevalue tracebackr r r __exit__%szCursorWrapper.__exit__Nc Cs|dk r(|jjj r(td|jj|jjn|dkrj|dkrj|jj|S|dkr|jj||S|pf}|jj|||SWdQRXdS)NzKKeyword parameters for callproc are not supported on this database backend.)rfeaturesZsupports_callproc_kwargsrvalidate_no_broken_transactionrrcallproc)r ZprocnameparamsZkparamsr r r r1s     zCursorWrapper.callproccCs|j||ddd|jS)NmanyFexecutor)_execute_with_wrappers_execute)r sqlrr r r executeCszCursorWrapper.executecCs|j||ddd|jS)NrTr)r _executemany)r r" param_listr r r executemanyFszCursorWrapper.executemanycCsWd|jd|i}x,t|jjD]}tj||}q(W|||||S)N connectionr)rreversedZexecute_wrappers functoolspartial)r r"rrrcontextwrapperr r r r Isz$CursorWrapper._execute_with_wrappersc GsT|jj|jj4|dkr6|jj|S|jj||SWdQRXdS)N)rrrrr#)r r"rignored_wrapper_argsr r r r!Os    zCursorWrapper._executec Gs8|jj|jj|jj||SWdQRXdS)N)rrrrr&)r r"r%r-r r r r$Xs  zCursorWrapper._executemany)__name__ __module__ __qualname__r frozensetrrrrrrr#r&r r!r$r r r r rs         rcsUeZdZdfddZfddZeddddddZS) CursorDebugWrapperNc s7|j||ddtj||SWdQRXdS)Nuse_last_executed_queryT) debug_sqlsuperr#)r r"r) __class__r r r#bszCursorDebugWrapper.executec s7|j||ddtj||SWdQRXdS)NrT)r4r5r&)r r"r%)r6r r r&fszCursorDebugWrapper.executemanyFc cstj}z dVWdtj}||}|rR|jjj|j||}y|rgt|nd}Wntk rd}YnX|jjj d|rd||fn|dd|it j d|||dd |d|d |iXdS) N?r"z %s times: %stimez%.3fz(%.3f) %s; args=%sextradurationr) r9 monotonicropsZlast_executed_queryrlen TypeErrorZ queries_logappendloggerdebug) r r"rr3rstartstopr;timesr r r r4js(       zCursorDebugWrapper.debug_sql)r.r/r0r#r&rr4r r )r6r r2^s r2cCs)|r%tjtt|jdSdS)N-)datetimedatemapintsplit)sr r r typecast_datesrMcCs|s dS|jd\}}}d|krF|jd\}}nd}tjt|t|t|t|dddS)N:.0000000)rKrGr9rJ)rLhourminutesseconds microsecondsr r r typecast_times rWc CsR|s dSd|kr t|S|j\}}d|krY|jdd\}}n$d|kr}|jdd\}}|jd}|jd}|d}d|kr|jd\}}nd}tjrtnd}tjt|d t|dt|dt|d t|dt|t|d dd |S) N rF+rNrOrPrrQrR)rMrKrZUSE_TZrrGrJ) rLdt_datesrErUrVtzinfor r r typecast_timestamps(      '#rac CsWy|jd\}}Wntk r:d|}}YnX|jd|jdfS)z Split a SQL identifier into a two element tuple of (namespace, name). The identifier could be a table, column, or sequence name might be prefixed by a namespace. z"."r7")rK ValueErrorstrip) identifier namespacenamer r r split_identifiers  rhcCsut|\}}|dks0t||kr4|St|d|}d|rYd|nd|d|||fS)z Shorten a SQL identifier to a repeatable mangled version with the given length. If a quote stripped name contains a namespace, e.g. USERNAME"."TABLE, truncate the table portion only. Nlengthz%s%s%sz%s"."r7)rhr> names_digest)rerjZhash_lenrfrgdigestr r r truncate_names rmcGsDtj}x!|D]}|j|jqW|jd|S)zk Generate a 32-bit digest of a set of arguments that can be used to shorten identifying names. N)hashlibmd5updateencode hexdigest)rjargshargr r r rks  rkcCs|dkrdStjj}|dk r7||_|dk rn|jtjdj| d|}nd|jtj<|j |}dj |S)zc Format a number into a string with the requisite number of digits and decimal places. NrYr+z{:f}) decimalZ getcontextcopyZprecZquantizeZDecimalZscalebZtrapsZRoundedZcreate_decimalformat)rZ max_digitsZdecimal_placesr+r r r format_numbers    +rycCs6|jdo|jd}|r2|ddS|S)z Strip quotes off of quoted table names to make them safe for use in index names, sequence names, etc. For example '"USER"."TABLE"' (an Oracle naming scheme) becomes 'USER"."TABLE'. rbrY) startswithendswith)Z table_nameZ has_quotesr r r strip_quotessr})rGrvr)rnloggingr9 contextlibrZ django.confrZdjango.db.utilsrZdjango.utils.timezoner getLoggerrArr2rMrWrarhrmrkryr}r r r r s(      N,