o
    ˜¨Êh$  ã                   @   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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   úJ/var/www/html/env/lib/python3.10/site-packages/django/db/backends/utils.pyÚ__init__   s   
zCursorWrapper.__init__)ÚfetchoneÚ	fetchmanyÚfetchallÚnextsetc                 C   s&   t | j|ƒ}|tjv r| j |¡S |S r   )Úgetattrr   r   ÚWRAP_ERROR_ATTRSr	   Úwrap_database_errors)r
   ÚattrÚcursor_attrr   r   r   Ú__getattr__   s   
zCursorWrapper.__getattr__c                 c   s<   � | j j� | jE d H  W d   ƒ d S 1 sw   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 d S  | jjjy   Y d S w r   )Úcloser	   Ú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 r,|d u r,| j |¡W  d   ƒ S |d u r>| j ||¡W  d   ƒ S |pAd}| j |||¡W  d   ƒ S 1 sTw   Y  d S )NzKKeyword parameters for callproc are not supported on this database backend.r   )r	   ÚfeaturesÚsupports_callproc_kwargsr   Úvalidate_no_broken_transactionr   r   Úcallproc)r
   ÚprocnameÚparamsÚkparamsr   r   r   r%   0   s   ÿ


þü$ùzCursorWrapper.callprocc                 C   ó   | j ||d| jd�S )NF©ÚmanyÚexecutor)Ú_execute_with_wrappersÚ_execute©r
   Úsqlr'   r   r   r   ÚexecuteB   ó   
ÿzCursorWrapper.executec                 C   r)   )NTr*   )r-   Ú_executemany©r
   r0   Ú
param_listr   r   r   ÚexecutemanyG   r2   zCursorWrapper.executemanyc                 C   s8   | j | dœ}t| j jƒD ]}t ||¡}q|||||ƒS )N)Ú
connectionr   )r	   ÚreversedÚexecute_wrappersÚ	functoolsÚpartial)r
   r0   r'   r+   r,   ÚcontextÚwrapperr   r   r   r-   L   s   z$CursorWrapper._execute_with_wrappersc                 G   sf   | j  ¡  | j j�  |d u r| j |¡W  d   ƒ S | j ||¡W  d   ƒ S 1 s,w   Y  d S r   )r	   r$   r   r   r1   )r
   r0   r'   Úignored_wrapper_argsr   r   r   r.   R   s   


ý$ûzCursorWrapper._executec                 G   sD   | j  ¡  | j j� | j ||¡W  d   ƒ S 1 sw   Y  d S r   )r	   r$   r   r   r6   )r
   r0   r5   r>   r   r   r   r3   [   s   

$ÿzCursorWrapper._executemany)NNr   )Ú__name__Ú
__module__Ú__qualname__r   Ú	frozensetr   r   r   r   r!   r%   r1   r6   r-   r.   r3   r   r   r   r   r      s    

	r   c                       s:   e Zd Zd	‡ fdd„	Z‡ fdd„Ze	d
dd„ƒZ‡  ZS )ÚCursorDebugWrapperNc                    óB   | j ||dd�� tƒ  ||¡W  d   ƒ S 1 sw   Y  d S )NT)Úuse_last_executed_query)Ú	debug_sqlÚsuperr1   r/   ©Ú	__class__r   r   r1   e   ó   $ÿzCursorDebugWrapper.executec                    rD   )NT)r+   )rF   rG   r6   r4   rH   r   r   r6   i   rJ   zCursorDebugWrapper.executemanyFc           	      c   sb  � t  ¡ }zXd V  W t  ¡ }|| }|r| jj | j||¡}z
|r%t|ƒnd}W n ty3   d}Y nw | jj 	|r@d||f n|d| dœ¡ t
jd|||| jj|||| jjdœd� d S t  ¡ }|| }|rr| jj | j||¡}z
|ryt|ƒnd}W n ty‡   d}Y nw | jj 	|r”d||f n|d| dœ¡ t
jd|||| jj|||| jjdœd� w )	NÚ ú?z%s times: %sz%.3f)r0   Útimez(%.3f) %s; args=%s; alias=%s)Údurationr0   r'   Úalias)Úextra)rM   Ú	monotonicr	   ÚopsÚlast_executed_queryr   ÚlenÚ	TypeErrorÚqueries_logÚappendÚloggerÚdebugrO   )	r
   r0   r'   rE   r+   ÚstartÚstoprN   Útimesr   r   r   rF   m   sp   €þþÿü
úñþþÿüúzCursorDebugWrapper.debug_sqlr   )NNFF)r?   r@   rA   r1   r6   r   rF   Ú__classcell__r   r   rH   r   rC   a   s    ÿrC   c                 C   sF   dD ]}|| 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   €
rf   c                 C   s   | rt jtt|  d¡ƒŽ S d S )Nr_   )ÚdatetimeÚdateÚmapÚintÚsplit)Úsr   r   r   Útypecast_date¤   s   ÿÿrm   c              	   C   s`   | sd S |   d¡\}}}d|v r|  d¡\}}nd}t t|ƒt|ƒt|ƒt|d d d… ƒ¡S )Nú:Ú.Ú0Ú000000é   )rk   rg   rM   rj   )rl   ÚhourÚminutesÚsecondsÚmicrosecondsr   r   r   Útypecast_timeª   s   $ÿrw   c                 C   sà   | sd S d| vrt | ƒS |  ¡ \}}d|v r| dd¡\}}nd|v r+| dd¡\}}| d¡}| d¡}|d }d|v rE| d¡\}}nd}t t|d	 ƒt|d ƒt|d ƒt|d	 ƒt|d ƒt|ƒt|d
 d d… ƒ¡S )Nú r_   r`   r^   rn   é   ro   rp   r   rq   rr   )rm   rk   rg   rj   )rl   ÚdÚtÚ_Údatesr\   ru   rv   r   r   r   Útypecast_timestamp·   s0   






ùr~   c                 C   sD   z	|   d¡\}}W n ty   d| }}Y nw | 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"."rK   ú")rk   Ú
ValueErrorÚstrip)Ú
identifierÚ	namespacerd   r   r   r   Úsplit_identifierÛ   s   ÿr„   é   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"."rK   )r„   rT   Únames_digest)r‚   r†   Úhash_lenrƒ   rd   Ú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r&| 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.
    Nr`   )r<   z{:f})ÚdecimalÚ
getcontextÚcopyÚprecÚquantizeÚDecimalÚscalebÚtrapsÚRoundedÚcreate_decimalÚformat)r   Ú
max_digitsÚdecimal_placesr<   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'.
    r   r`   éÿÿÿÿ)Ú
startswithÚendswith)Ú
table_nameÚ
has_quotesr   r   r   Ústrip_quotes  s   r¥   )Nr…   )rg   r’   r:   ÚloggingrM   Ú
contextlibr   Ú	django.dbr   Údjango.utils.cryptor   Údjango.utils.dateparser   Ú	getLoggerrX   r   rC   rf   rm   rw   r~   r„   rŠ   r‡   rŸ   r¥   r   r   r   r   Ú<module>   s*    
R2$
