3
(ha                 @   s  d Z ddlmZ 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 ddlmZ ddlmZ ddlmZmZ dd	lmZ dd
lmZ ddlmZmZ ddlmZ dd Zed-d.g yddlZW n< e k
r Z! zddl"m#Z# e#de! W Y ddZ![!X nX ddl$m%Z% ddl&m'Z' ddl(m)Z) ddl*m+Z+ ddl,m-Z- ddl.m/Z/ ddlm0Z0m1Z1 ej2Z2ej3Z3G dd de4Z5G dd  d eZ6G d!d" d"e4Z7G d#d$ d$e4Z8G d%d& d&e4Z9G d'd( d(ej:Z;d)d* Z<d+d, Z=dS )/z\
Oracle database backend for Django.

Requires cx_Oracle: http://cx-oracle.sourceforge.net/
    )unicode_literalsN)settings)utils)BaseDatabaseWrapper)BaseDatabaseValidation)sixtimezone)RemovedInDjango20Warning)duration_string)force_bytes
force_text)cached_propertyc             C   s   t j j jdrydd l}W n: tk
rX } zddlm} |d| W Y d d }~X nX |jd}x*| D ]\}}|j	|| qjW nt
jj|  d S )NZCYGWINr   )ImproperlyConfiguredz_Error loading ctypes: %s; the Oracle backend requires ctypes to operate correctly under Cygwin.kernel32)platformsystemupper
startswithctypesImportErrordjango.core.exceptionsr   CDLLZSetEnvironmentVariableAosenvironupdate)r   r   er   r   namevalue r   B/tmp/pip-install-q3hcpn_q/Django/django/db/backends/oracle/base.py_setup_environment   s    
r    NLS_LANG.UTF8ORA_NCHAR_LITERAL_REPLACETRUE)r   z"Error loading cx_Oracle module: %s   )DatabaseClient)DatabaseCreation)DatabaseFeatures)DatabaseIntrospection)DatabaseOperations)DatabaseSchemaEditor)Oracle_datetimeconvert_unicodec               @   s   e Zd ZdddZdS )!_UninitializedOperatorsDescriptorNc             C   s&   |d krt d|j j  |jd S )Nz*operators not available as class attribute	operators)AttributeErrorcursorclose__dict__)selfinstanceclsr   r   r   __get__I   s    z)_UninitializedOperatorsDescriptor.__get__)N)__name__
__module____qualname__r7   r   r   r   r   r.   G   s   r.   c                   sZ  e Zd ZdZdddddddd	d
dddddddddddddddd	dddZdddddZe ZdddddddddddddZej	 Z
e
jdd ddd d d! d"Zd#d$d%d&d'd(d)Zd*d+ ej D Zd,d+ ej D ZeZeZ fd-d.Zd/d0 Zd1d2 Zd3d4 Zd5d6 Zd7d8 Zd9d: Zd;d< Zd=d> ZdHd@dAZdBdC ZedDdE Z edFdG Z!  Z"S )IDatabaseWrapperoraclez
NUMBER(11)z
NUMBER(19)ZBLOBz	NUMBER(1)zNVARCHAR2(%(max_length)s)zVARCHAR2(%(max_length)s)ZDATEZ	TIMESTAMPz*NUMBER(%(max_digits)s, %(decimal_places)s)zINTERVAL DAY(9) TO SECOND(6)zDOUBLE PRECISIONzVARCHAR2(15)zVARCHAR2(39)ZNCLOBzVARCHAR2(32))Z	AutoFieldZBigAutoFieldZBinaryFieldBooleanFieldZ	CharFieldZCommaSeparatedIntegerFieldZ	DateFieldZDateTimeFieldZDecimalFieldZDurationFieldZ	FileFieldZFilePathFieldZ
FloatFieldZIntegerFieldZBigIntegerFieldZIPAddressFieldZGenericIPAddressFieldNullBooleanFieldZOneToOneFieldPositiveIntegerFieldPositiveSmallIntegerFieldZ	SlugFieldZSmallIntegerFieldZ	TextFieldZ	TimeFieldZURLFieldZ	UUIDFieldz%(qn_column)s IN (0,1)z3(%(qn_column)s IN (0,1)) OR (%(qn_column)s IS NULL)z%(qn_column)s >= 0)r=   r>   r?   r@   z= %sz= UPPER(%s)zFLIKE TRANSLATE(%s USING NCHAR_CS) ESCAPE TRANSLATE('\' USING NCHAR_CS)zMLIKE UPPER(TRANSLATE(%s USING NCHAR_CS)) ESCAPE TRANSLATE('\' USING NCHAR_CS)z> %sz>= %sz< %sz<= %s)exactZiexactcontains	icontainsgtZgteltZlter   endswithistartswith	iendswithzLIKEC %s ESCAPE '\'zLIKEC UPPER(%s) ESCAPE '\')rB   rC   r   rF   rG   rH   z@REPLACE(REPLACE(REPLACE({}, '\', '\\'), '%%', '\%%'), '_', '\_')z'%%' || {} || '%%'z'%%' || UPPER({}) || '%%'z
{} || '%%'zUPPER({}) || '%%'z
'%%' || {}z'%%' || UPPER({}))rB   rC   r   rG   rF   rH   c             C   s   i | ]\}}d | d |qS )zLIKE TRANSLATE( z5 USING NCHAR_CS) ESCAPE TRANSLATE('\' USING NCHAR_CS)r   ).0kvr   r   r   
<dictcomp>   s   zDatabaseWrapper.<dictcomp>c             C   s   i | ]\}}d | d |qS )zLIKEC z ESCAPE '\'r   )rI   rJ   rK   r   r   r   rL      s   c                sl   t t| j|| t| | _| jd jdd}|| j_t| | _	t
| | _t| | _t| | _t| | _d S )NOPTIONSuse_returning_intoT)superr;   __init__r(   featuressettings_dictgetZcan_return_id_from_insertr*   opsr&   clientr'   creationr)   introspectionr   Z
validation)r4   argskwargsrN   )	__class__r   r   rP      s    




zDatabaseWrapper.__init__c             C   sd   | j }|d j sd|d< |d j rFtj|d t|d |d }n|d }d|d |d |f S )NZHOST	localhostZPORTNAMEz%s/%s@%sUSERZPASSWORD)rR   stripDatabaseZmakedsnint)r4   rR   Zdsnr   r   r   _connect_string   s    

zDatabaseWrapper._connect_stringc             C   s    | j d j }d|kr|d= |S )NrM   rN   )rR   copy)r4   conn_paramsr   r   r   get_connection_params   s    z%DatabaseWrapper.get_connection_paramsc             C   s   t | j }tj|f|S )N)r-   ra   r_   connect)r4   rc   Zconn_stringr   r   r   get_new_connection   s    z"DatabaseWrapper.get_new_connectionc             C   s   | j  }|jd |jdtjr"dnd  |j  d| jkr| j  }y|jd| jd  dg W n$ tk
r   | j| _	| j
| _Y nX | j| _	| j| _|j  yd	| j_W n tk
r   Y nX | j s| j  d S )
Nz+ALTER SESSION SET NLS_TERRITORY = 'AMERICA'zmALTER SESSION SET NLS_DATE_FORMAT = 'YYYY-MM-DD HH24:MI:SS' NLS_TIMESTAMP_FORMAT = 'YYYY-MM-DD HH24:MI:SS.FF'z TIME_ZONE = 'UTC' r/   z!SELECT 1 FROM DUAL WHERE DUMMY %srB   X   )create_cursorexecuter   USE_TZr2   r3   _standard_operatorsDatabaseError_likec_operatorsr/   _likec_pattern_opsZpattern_ops_standard_pattern_ops
connectionZstmtcachesizer0   Zget_autocommitcommit)r4   r1   r   r   r   init_connection_state   s0    


z%DatabaseWrapper.init_connection_statec             C   s
   t | jS )N)FormatStylePlaceholderCursorrr   )r4   r   r   r   rj     s    zDatabaseWrapper.create_cursorc             C   s   | j d k	ry
| j j S  tjk
r } z^|jd }t|dr~t|dr~|jdkr~d|jkr~tj	t
jt
jt|j tj d   W Y d d }~X nX d S )Nr   codemessagei+  z	ORA-02291   )rr   rs   r_   rn   rX   hasattrrv   rw   r   reraiser   IntegrityErrortuplesysexc_info)r4   r   xr   r   r   _commit  s    


$zDatabaseWrapper._commitc             C   s(   | j r$| jjd| jj| dd d S )Nz-- RELEASE SAVEPOINT %s (faked)z0.000)Zsqltime)Zqueries_loggedZqueries_logappendrT   Z
quote_name)r4   Zsidr   r   r   _savepoint_commit  s    z!DatabaseWrapper._savepoint_commitc          	   C   s   | j  || j_W d Q R X d S )N)Zwrap_database_errorsrr   
autocommit)r4   r   r   r   r   _set_autocommit&  s    zDatabaseWrapper._set_autocommitNc             C   s    | j  jd | j  jd dS )z
        To check constraints, we set constraints to immediate. Then, when, we're done we must ensure they
        are returned to deferred.
        zSET CONSTRAINTS ALL IMMEDIATEzSET CONSTRAINTS ALL DEFERREDN)r1   rk   )r4   Ztable_namesr   r   r   check_constraints*  s    z!DatabaseWrapper.check_constraintsc             C   s.   y| j j  W n tjk
r$   dS X dS d S )NFT)rr   Zpingr_   Error)r4   r   r   r   	is_usable2  s
    zDatabaseWrapper.is_usablec          	   C   s   | j  
 | jjS Q R X d S )N)Ztemporary_connectionrr   version)r4   r   r   r   oracle_full_version:  s    
z#DatabaseWrapper.oracle_full_versionc             C   s.   yt | jjdd S  tk
r(   d S X d S )N.r   )r`   r   split
ValueError)r4   r   r   r   oracle_version?  s    zDatabaseWrapper.oracle_version)N)#r8   r9   r:   vendorZ
data_typesZdata_type_check_constraintsr.   r/   rm   rb   ro   r   Zpattern_escZ_pattern_opsitemsrq   rp   r_   r+   ZSchemaEditorClassrP   ra   rd   rf   rt   rj   r   r   r   r   r   r   r   r   __classcell__r   r   )rZ   r   r;   T   s   	
.
r;   c               @   s   e Zd ZdZdddZdS )OracleParama  
    Wrapper object for formatting parameters for Oracle. If the string
    representation of the value is large enough (greater than 4000 characters)
    the input size needs to be set as CLOB. Alternatively, if the parameter
    has an `input_size` attribute, then the value of the `input_size` attribute
    will be used instead. Otherwise, no input size will be set for the
    parameter when executing the query.
    Fc             C   s$  t jrRt|tjrRt|t rRtj|rHtjdt	 |j
tjjd d}tj|}t|tjrvt|}d|krvd| }d}|dkrd}n|dkrd}t|d	r|j|| _nDt|tjr|| _n0t||j|| _t| jtjrtt||j|}t|d
r|j| _n|dkrtj| _nd | _d S )NzThe Oracle database adapter received an aware datetime (%s), probably from cursor.execute(). Update your code to pass a naive datetime in the database connection's time zone (UTC by default).)tzinfo z0 r   Tr%   Fbind_parameter
input_sizei  )r   rl   
isinstancedatetimer,   r   Zis_awarewarningswarnr	   
astimezoneutcreplaceZfrom_datetime	timedeltar
   ry   r   r   r_   Binaryr-   charsetr   string_typeslenr   ZCLOB)r4   paramr1   Zstrings_onlyZstring_sizer   r   r   rP   Q  s:    





zOracleParam.__init__N)F)r8   r9   r:   __doc__rP   r   r   r   r   r   G  s   r   c               @   s0   e Zd ZdZdd Zdd Zdd Zdd	 Zd
S )VariableWrappera	  
    An adapter class for cursor variables that prevents the wrapped object
    from being converted into a string when used to instantiate an OracleParam.
    This can be used generally for any other object that should be passed into
    Cursor.execute as-is.
    c             C   s
   || _ d S )N)var)r4   r   r   r   r   rP     s    zVariableWrapper.__init__c             C   s   | j S )N)r   )r4   r1   r   r   r   r     s    zVariableWrapper.bind_parameterc             C   s   t | j|S )N)getattrr   )r4   keyr   r   r   __getattr__  s    zVariableWrapper.__getattr__c             C   s&   |dkr|| j |< nt| j|| d S )Nr   )r3   setattrr   )r4   r   r   r   r   r   __setattr__  s    zVariableWrapper.__setattr__N)r8   r9   r:   r   rP   r   r   r   r   r   r   r   r     s
   r   c               @   s   e Zd ZdZdZd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dZdd Zdd Zdd Zdd Zdd Zd d! ZdS )%ru   aF  
    Django uses "format" (e.g. '%s') style placeholders, but Oracle uses ":var"
    style. This fixes it -- but note that if you want to use a literal "%s" in
    a query, you'll need to use "%%s".

    We also do automatic conversion between Unicode on the Python side and
    UTF-8 -- for talking to Oracle -- in here.
    zutf-8c             C   s   |j  | _ d| j _d| j _d S )NTd   )r1   ZnumbersAsStrings	arraysize)r4   rr   r   r   r   rP     s    
z%FormatStylePlaceholderCursor.__init__c                sB   y fdd|j  D S  tk
r<   t fdd|D S X d S )Nc                s   i | ]\}}t | d |qS )T)r   )rI   rJ   rK   )r4   r   r   rL     s    z?FormatStylePlaceholderCursor._format_params.<locals>.<dictcomp>c             3   s   | ]}t | d V  qdS )TN)r   )rI   p)r4   r   r   	<genexpr>  s    z>FormatStylePlaceholderCursor._format_params.<locals>.<genexpr>)r   r0   r|   )r4   paramsr   )r4   r   _format_params  s    z+FormatStylePlaceholderCursor._format_paramsc             C   s   t |d drTi }x2|D ]*}x$|j D ]\}}|jr&|j||< q&W qW | jf | nPd gt|d  }x2|D ]*}x$t|D ]\}}|jrz|j||< qzW qlW | j|  d S )Nr   keys)ry   r   r   Zsetinputsizesr   	enumerate)r4   Zparams_listsizesr   rJ   r   ir   r   r   _guess_input_sizes  s    

z/FormatStylePlaceholderCursor._guess_input_sizesc             C   s.   t |drdd |j D S dd |D S d S )Nr   c             S   s   i | ]\}}|j |qS r   )r   )rI   rJ   rK   r   r   r   rL     s    zAFormatStylePlaceholderCursor._param_generator.<locals>.<dictcomp>c             S   s   g | ]
}|j qS r   )r   )rI   r   r   r   r   
<listcomp>  s    zAFormatStylePlaceholderCursor._param_generator.<locals>.<listcomp>)ry   r   )r4   r   r   r   r   _param_generator  s    
z-FormatStylePlaceholderCursor._param_generatorc             C   s   |j ds|j dr |d d	 }|d kr:g }t|| j}nXt|drhdd |j D }t|| | j}n*dd tt|D }t|t| | j}|| j|fS )
N;/r%   r   c             S   s   i | ]}d | |qS )z:%sr   )rI   rJ   r   r   r   rL     s    z@FormatStylePlaceholderCursor._fix_for_params.<locals>.<dictcomp>c             S   s   g | ]}d | qS )z:arg%dr   )rI   r   r   r   r   r     s    z@FormatStylePlaceholderCursor._fix_for_params.<locals>.<listcomp>)	rF   r-   r   ry   r   ranger   r|   r   )r4   queryr   rX   r   r   r   _fix_for_params  s    
z,FormatStylePlaceholderCursor._fix_for_paramsNc             C   s   | j ||\}}| j|g y| jj|| j|S  tjk
r } zXt|jd dr|jd j	dkrt
|t rtjtjtjt|j tj d   W Y d d }~X nX d S )Nr   rv   ix  rx   )r   r   r1   rk   r   r_   rn   ry   rX   rv   r   r{   r   rz   r   r|   r}   r~   )r4   r   r   r   r   r   r   rk     s    ,$z$FormatStylePlaceholderCursor.executec                s   |sd S t |} j|t|\}}|g fdd|D  } j| y jj| fdd|D S  tjk
r } zXt|j	d dr|j	d j
dkrt|t rtjtjtjt|j	 tj d   W Y d d }~X nX d S )Nc                s   g | ]} j |qS r   )r   )rI   r   )r4   r   r   r     s    z<FormatStylePlaceholderCursor.executemany.<locals>.<listcomp>c                s   g | ]} j |qS r   )r   )rI   r   )r4   r   r   r     s    r   rv   ix  rx   )iterr   nextr   r1   executemanyr_   rn   ry   rX   rv   r   r{   r   rz   r   r|   r}   r~   )r4   r   r   Zparams_iterZfirstparams	formattedr   r   )r4   r   r     s    
,$z(FormatStylePlaceholderCursor.executemanyc             C   s"   | j j }|d kr|S t|| j S )N)r1   fetchone_rowfactory)r4   rowr   r   r   r     s    
z%FormatStylePlaceholderCursor.fetchonec                s,   |d kr j }t fdd jj|D S )Nc             3   s   | ]}t | jV  qd S )N)r   r1   )rI   r)r4   r   r   r     s    z9FormatStylePlaceholderCursor.fetchmany.<locals>.<genexpr>)r   r|   r1   	fetchmany)r4   sizer   )r4   r   r     s    z&FormatStylePlaceholderCursor.fetchmanyc                s   t  fdd jj D S )Nc             3   s   | ]}t | jV  qd S )N)r   r1   )rI   r   )r4   r   r   r     s    z8FormatStylePlaceholderCursor.fetchall.<locals>.<genexpr>)r|   r1   fetchall)r4   r   )r4   r   r     s    z%FormatStylePlaceholderCursor.fetchallc             C   s*   y| j j  W n tjk
r$   Y nX d S )N)r1   r2   r_   ZInterfaceError)r4   r   r   r   r2   	  s    z"FormatStylePlaceholderCursor.closec             G   s   t | jj| S )N)r   r1   r   )r4   rX   r   r   r   r     s    z FormatStylePlaceholderCursor.varc             G   s   t | jj| S )N)r   r1   arrayvar)r4   rX   r   r   r   r     s    z%FormatStylePlaceholderCursor.arrayvarc             C   s$   || j kr| j | S t| j|S d S )N)r3   r   r1   )r4   attrr   r   r   r     s    

z(FormatStylePlaceholderCursor.__getattr__c             C   s
   t | jS )N)CursorIteratorr1   )r4   r   r   r   __iter__  s    z%FormatStylePlaceholderCursor.__iter__)N)N)N)r8   r9   r:   r   r   rP   r   r   r   r   rk   r   r   r   r   r2   r   r   r   r   r   r   r   r   ru     s"   


ru   c               @   s(   e Zd ZdZdd Zdd Zdd ZdS )	r   zF
    Cursor iterator wrapper that invokes our custom row factory.
    c             C   s   || _ t|| _d S )N)r1   r   )r4   r1   r   r   r   rP   %  s    zCursorIterator.__init__c             C   s   | S )Nr   )r4   r   r   r   r   )  s    zCursorIterator.__iter__c             C   s   t t| j| jS )N)r   r   r   r1   )r4   r   r   r   __next__,  s    zCursorIterator.__next__N)r8   r9   r:   r   rP   r   r   r   r   r   r   r      s   r   c             C   s   g }xt | |jD ]\}}|d k	r|d tjkr|dd \}}|dkrx|dkrnd|krdtj|}qvt|}qt|}q|dkr|dkrt|}qtj|}qd|krtj|}qt|}n |d tjtj	tj
fkrt|}|j| qW t|S )Nr%            r   r   i)zipdescriptionr_   NUMBERdecimalDecimalr`   floatSTRINGZ
FIXED_CHARZLONG_STRING
to_unicoder   r|   )r   r1   Zcastedr   descZ	precisionZscaler   r   r   r   0  s,    




r   c             C   s   t | tjrt| S | S )z]
    Convert strings to Unicode objects (and return all other data types
    unchanged).
    )r   r   r   r   )sr   r   r   r   Y  s    r   )r!   r"   )r#   r$   )>r   
__future__r   r   r   r   r   r}   r   Zdjango.confr   Z	django.dbr   Zdjango.db.backends.base.baser   Z"django.db.backends.base.validationr   Zdjango.utilsr   r   Zdjango.utils.deprecationr	   Zdjango.utils.durationr
   Zdjango.utils.encodingr   r   Zdjango.utils.functionalr   r    Z	cx_Oracler_   r   r   r   r   rU   r&   rV   r'   rQ   r(   rW   r)   
operationsr*   Zschemar+   r,   r-   rn   r{   objectr.   r;   r   r   ru   Iteratorr   r   r   r   r   r   r   <module>   sX    t9 	)