U
    Ha                      @   s   d dl Z 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	 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dZdd Zdd Zdd ZdS )    N)contextmanager)NotSupportedErrorzdjango.db.backendsc                   @   sx   e Zd Zdd Zeddddg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	   </tmp/pip-unpacked-wheel-3jxiddxt/django/db/backends/utils.py__init__   s    zCursorWrapper.__init__ZfetchoneZ	fetchmanyZfetchallZnextsetc                 C   s*   t | j|}|tjk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   s$   | j j | jE d H  W 5 Q R X 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k
r&   Y nX d S r   )closer   ZDatabaseError)r   typevalue	tracebackr	   r	   r
   __exit__#   s    zCursorWrapper.__exit__Nc              
   C   s   |d k	r| j jjstd| j   | j jt |d krV|d krV| j|W  5 Q R  S |d krx| j||W  5 Q R  S |p~d}| j|||W  5 Q R  S W 5 Q R X 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   /   s    

zCursorWrapper.callprocc                 C   s   | j ||d| jdS )NFmanyexecutor)_execute_with_wrappers_executer   sqlr   r	   r	   r
   executeA   s    zCursorWrapper.executec                 C   s   | j ||d| jdS )NTr   )r!   _executemanyr   r$   
param_listr	   r	   r
   executemanyD   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!   G   s    z$CursorWrapper._execute_with_wrappersc              
   G   s\   | j   | j j@ |d kr4| j|W  5 Q R  S | j||W  5 Q R  S W 5 Q R X d S r   )r   r   r   r   r%   )r   r$   r   ignored_wrapper_argsr	   r	   r
   r"   M   s
    

zCursorWrapper._executec              
   G   s8   | j   | j j | j||W  5 Q R  S Q R X d S r   )r   r   r   r   r)   )r   r$   r(   r0   r	   r	   r
   r&   V   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              
      s6   | j ||dd t ||W  5 Q R  S Q R X d S )NT)use_last_executed_query)	debug_sqlsuperr%   r#   	__class__r	   r
   r%   `   s    zCursorDebugWrapper.executec              
      s6   | j ||dd t ||W  5 Q R  S Q R X d S )NT)r   )r7   r8   r)   r'   r9   r	   r
   r)   d   s    zCursorDebugWrapper.executemanyFc           	      c   s   t  }z
d V  W 5 t  }|| }|r<| jj| j||}z|rJt|nd}W n tk
rh   d}Y nX | jj	|rd||f n|d| d t
jd||||||dd X d S )	N ?z%s times: %sz%.3f)r$   timez(%.3f) %s; args=%s)durationr$   r   )extra)r=   	monotonicr   opsZlast_executed_queryr   len	TypeErrorZqueries_logappendloggerdebug)	r   r$   r   r6   r   startstopr>   timesr	   r	   r
   r7   h   s,    


zCursorDebugWrapper.debug_sql)N)NNFF)r1   r2   r3   r%   r)   r   r7   __classcell__r	   r	   r9   r
   r5   \   s   r5   c                 C   s   | rt jtt| d S d S )N-)datetimedatemapintsplit)sr	   r	   r
   typecast_date   s    rR   c              	   C   s`   | sd S |  d\}}}d|kr0| d\}}nd}tt|t|t|t|d d d S )N:.0000000   )rP   rL   r=   rO   )rQ   hourminutessecondsmicrosecondsr	   r	   r
   typecast_time   s    r\   c                 C   s   | sd S d| krt | S |  \}}d|kr>|dd\}}nd|krV|dd\}}|d}|d}|d }d|kr|d\}}nd}tt|d	 t|d t|d t|d	 t|d t|t|d
 d d S )N rK      +rS      rT   rU   r   rV   rW   )rR   rP   rL   rO   )rQ   dt_datesrI   rZ   r[   r	   r	   r
   typecast_timestamp   s0    


 
 

 
 re   c                 C   sF   z|  d\}}W n tk
r0   d|  }}Y nX |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;   ")rP   
ValueErrorstrip)
identifier	namespacenamer	   r	   r
   split_identifier   s
    rl      c                 C   sV   t | \}}|dk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;   )rl   rB   names_digest)ri   rn   Zhash_lenrj   rk   digestr	   r	   r
   truncate_name   s
    rq   c                 G   s0   t  }|D ]}||  q| d|  S )zk
    Generate a 32-bit digest of a set of arguments that can be used to shorten
    identifying names.
    N)hashlibmd5updateencode	hexdigest)rn   argshargr	   r	   r
   ro      s    ro   c                 C   sl   | dkrdS t   }|dk	r&||_|dk	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.
    Nr^   )r.   z{:f})decimal
getcontextcopyprecquantizeDecimalscalebZtrapsRoundedcreate_decimalformat)r   Z
max_digitsZ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'.
    rf   r^   )
startswithendswith)Z
table_nameZ
has_quotesr	   r	   r
   strip_quotes   s    r   )Nrm   )rL   rz   r,   rr   loggingr=   
contextlibr   Z	django.dbr   	getLoggerrE   r   r5   rR   r\   re   rl   rq   ro   r   r   r	   r	   r	   r
   <module>   s$   
N,
