a
    Lcòf'  ã                   @   sÚ   d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlmZ d dl	m
Z
 d dlmZ e d¡ZG dd„ dƒZG d	d
„ d
eƒZedd„ ƒZdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zd dd„Zdd„ Zdd„ Zdd„ ZdS )!é    N)Úcontextmanager)ÚNotSupportedError)Úmd5)Ú
parse_timezdjango.db.backendsc                   @   st   e Zd Zdd„ Zeg d¢ƒZdd„ Zdd„ Zdd	„ Zd
d„ Z	ddd„Z
ddd„Zdd„ Zdd„ Zdd„ Zdd„ ZdS )ÚCursorWrapperc                 C   s   || _ || _d S ©N)ÚcursorÚdb)Úselfr   r	   © r   úU/home/httpd/docs/test/DocsMgr/lib/python3.9/site-packages/django/db/backends/utils.pyÚ__init__   s    zCursorWrapper.__init__)ZfetchoneZ	fetchmanyZfetchallZnextsetc                 C   s*   t | j|ƒ}|tjv r"| j |¡S |S d S r   )Úgetattrr   r   ÚWRAP_ERROR_ATTRSr	   Úwrap_database_errors)r
   ÚattrZcursor_attrr   r   r   Ú__getattr__   s    
zCursorWrapper.__getattr__c                 c   s8   | j j� | jE d H  W d   ƒ n1 s*0    Y  d S r   )r	   r   r   ©r
   r   r   r   Ú__iter__   s    
zCursorWrapper.__iter__c                 C   s   | S r   r   r   r   r   r   Ú	__enter__!   s    zCursorWrapper.__enter__c                 C   s*   z|   ¡  W n | jjjy$   Y n0 d S r   )Úcloser	   ZDatabaseÚError)r
   ÚtypeÚvalueÚ	tracebackr   r   r   Ú__exit__$   s    zCursorWrapper.__exit__Nc                 C   sÄ   |d ur| j jjstdƒ‚| j  ¡  | j j�„ |d u rX|d u rX| j |¡W  d   ƒ S |d u r|| j ||¡W  d   ƒ S |p‚d}| j |||¡W  d   ƒ S W d   ƒ n1 s¶0    Y  d S )NzKKeyword parameters for callproc are not supported on this database backend.r   )r	   ÚfeaturesZsupports_callproc_kwargsr   Úvalidate_no_broken_transactionr   r   Úcallproc)r
   ZprocnameÚparamsZkparamsr   r   r   r   0   s    ÿ

zCursorWrapper.callprocc                 C   s   | j ||d| jd�S )NF©ÚmanyÚexecutor)Ú_execute_with_wrappersÚ_execute©r
   Úsqlr   r   r   r   ÚexecuteB   s    
ÿzCursorWrapper.executec                 C   s   | j ||d| jd�S )NTr    )r#   Ú_executemany©r
   r&   Ú
param_listr   r   r   ÚexecutemanyG   s    
ÿzCursorWrapper.executemanyc                 C   s8   | j | dœ}t| j jƒD ]}t ||¡}q|||||ƒS )N)Ú
connectionr   )r	   ÚreversedZexecute_wrappersÚ	functoolsÚpartial)r
   r&   r   r!   r"   ÚcontextÚwrapperr   r   r   r#   L   s    z$CursorWrapper._execute_with_wrappersc                 G   st   | j  ¡  | j j�N |d u r6| j |¡W  d   ƒ S | j ||¡W  d   ƒ S W d   ƒ n1 sf0    Y  d S r   )r	   r   r   r   r'   )r
   r&   r   Úignored_wrapper_argsr   r   r   r$   R   s
    

zCursorWrapper._executec                 G   sD   | j  ¡  | j j� | j ||¡W  d   ƒ S 1 s60    Y  d S r   )r	   r   r   r   r+   )r
   r&   r*   r2   r   r   r   r(   [   s    

zCursorWrapper._executemany)NN)N)Ú__name__Ú
__module__Ú__qualname__r   Ú	frozensetr   r   r   r   r   r   r'   r+   r#   r$   r(   r   r   r   r   r      s   

	r   c                       s8   e Zd Zd	‡ fdd„	Z‡ fdd„Zed
dd„ƒZ‡  ZS )ÚCursorDebugWrapperNc                    sB   | j ||dd�� tƒ  ||¡W  d   ƒ S 1 s40    Y  d S )NT)Úuse_last_executed_query)Ú	debug_sqlÚsuperr'   r%   ©Ú	__class__r   r   r'   d   s    zCursorDebugWrapper.executec                    sB   | j ||dd�� tƒ  ||¡W  d   ƒ S 1 s40    Y  d S )NT)r!   )r9   r:   r+   r)   r;   r   r   r+   h   s    zCursorDebugWrapper.executemanyFc           	      c   sf  t  ¡ }z®d V  W t  ¡ }|| }|r:| jj | j||¡}z|rHt|ƒnd}W n tyd   d}Y n0 | jj 	|r~d||f n|d| dœ¡ t
jd|||| jj|||| jjdœd� nªt  ¡ }|| }|rà| jj | j||¡}z|rît|ƒnd}W n t�y   d}Y n0 | jj 	|�r(d||f n|d| dœ¡ t
jd|||| jj|||| jjdœd� 0 d S )	NÚ ú?z%s times: %sú%.3f©r&   Útimeú(%.3f) %s; args=%s; alias=%s)Údurationr&   r   Úalias©Úextra)rA   Ú	monotonicr	   ÚopsZlast_executed_queryr   ÚlenÚ	TypeErrorÚqueries_logÚappendÚloggerÚdebugrD   )	r
   r&   r   r8   r!   ÚstartÚstoprC   Útimesr   r   r   r9   l   sj    
þÿüúñ
þÿüúzCursorDebugWrapper.debug_sql)N)NNFF)r3   r4   r5   r'   r+   r   r9   Ú__classcell__r   r   r;   r   r7   a   s
    ÿr7   c                 c   s¾   t  ¡ }z\d V  W | jrºt  ¡ }|| }| j d| d| dœ¡ tjd||d | j||| jdœd� nT| jr¸t  ¡ }|| }| j d| d| dœ¡ tjd||d | j||| jdœd� 0 d S )Nz%sr?   r@   rB   )rC   r&   rD   rE   )rA   rG   Zqueries_loggedrK   rL   rM   rN   rD   )r,   r&   rO   rP   rC   r   r   r   Údebug_transaction’   sR    þÿýú÷þÿýúrS   c                 C   sF   dD ]6}|| v r|   |d¡\}}|rt|ƒr|||f  S q| ddfS )zH
    Split a time zone name into a 3-tuple of (name, sign, offset).
    )ú+ú-é   N)Úrsplitr   )ÚtznameÚsignÚnameÚoffsetr   r   r   Úsplit_tzname_delta¯   s    r\   c                 C   s   | rt jtt|  d¡ƒŽ S d S )NrU   )ÚdatetimeÚdateÚmapÚintÚsplit)Úsr   r   r   Útypecast_dateÀ   s    ÿrc   c              	   C   s`   | sd S |   d¡\}}}d|v r0|  d¡\}}nd}t t|ƒt|ƒt|ƒt|d d d… ƒ¡S )Nú:Ú.Ú0Ú000000é   )ra   r]   rA   r`   )rb   ÚhourÚminutesÚsecondsÚmicrosecondsr   r   r   Útypecast_timeÆ   s    $ÿrm   c                 C   sà   | sd S d| vrt | ƒS |  ¡ \}}d|v r>| dd¡\}}nd|v rV| dd¡\}}| d¡}| d¡}|d }d|v rŠ| d¡\}}nd}t t|d	 ƒt|d ƒt|d ƒt|d	 ƒt|d ƒt|ƒt|d
 d d… ƒ¡S )Nú rU   rV   rT   rd   é   re   rf   r   rg   rh   )rc   ra   r]   r`   )rb   ÚdÚtÚ_ÚdatesrQ   rk   rl   r   r   r   Útypecast_timestampÓ   s0    






ùrt   c                 C   sD   z|   d¡\}}W n ty.   d|  }}Y n0 | d¡| d¡fS )zµ
    Split an 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"."r=   ú")ra   Ú
ValueErrorÚstrip)Ú
identifierÚ	namespacerZ   r   r   r   Úsplit_identifier÷   s
    rz   é   c                 C   sV   t | ƒ\}}|du s t|ƒ|kr$| S t||d�}d|r>d| nd|d|| … |f S )zÎ
    Shorten an 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.
    N)Úlengthz%s%s%sz%s"."r=   )rz   rI   Únames_digest)rx   r|   Zhash_lenry   rZ   Údigestr   r   r   Útruncate_name  s    ýr   c                 G   s2   t dd�}|D ]}| | ¡ ¡ q| ¡ d| … S )zk
    Generate a 32-bit digest of a set of arguments that can be used to shorten
    identifying names.
    F)ÚusedforsecurityN)r   ÚupdateÚencodeÚ	hexdigest)r|   ÚargsÚhÚargr   r   r   r}     s    
r}   c                 C   sl   | du rdS t  ¡  ¡ }|dur&||_|durL| jt  d¡ | ¡|d�} nd|jt j< | 	| ¡} d 
| ¡S )zc
    Format a number into a string with the requisite number of digits and
    decimal places.
    NrV   )r0   z{:f})ÚdecimalÚ
getcontextÚcopyÚprecÚquantizeÚDecimalÚscalebZtrapsÚRoundedÚcreate_decimalÚformat)r   Z
max_digitsZdecimal_placesr0   r   r   r   Úformat_number%  s    ÿ
r‘   c                 C   s(   |   d¡o|  d¡}|r$| dd… S | 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'.
    ru   rV   éÿÿÿÿ)Ú
startswithÚendswith)Z
table_nameZ
has_quotesr   r   r   Ústrip_quotes9  s    r•   )Nr{   )r]   r‡   r.   ÚloggingrA   Ú
contextlibr   Z	django.dbr   Zdjango.utils.cryptor   Zdjango.utils.dateparser   Ú	getLoggerrM   r   r7   rS   r\   rc   rm   rt   rz   r   r}   r‘   r•   r   r   r   r   Ú<module>   s,   
R1
$
