U
    8ÄT_«[  ã                
   @   s�  d 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 ddlmZ dd	lmZmZ dd
lmZ dd„ Zeddgƒ zddlZW n. ek
rÖ Z ze
de ƒ‚W 5 dZ[X Y 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'm(Z( ddl)m*Z* edd„ ƒZ+G dd„ dƒZ,G dd„ deƒZ-G dd „ d ƒZ.G d!d"„ d"ƒZ/G d#d$„ d$ƒZ0dS )%ze
Oracle database backend for Django.

Requires cx_Oracle: https://oracle.github.io/python-cx_Oracle/
é    N)Úcontextmanager)Úsettings)ÚImproperlyConfigured)ÚIntegrityError)ÚBaseDatabaseWrapper)Úasync_unsafe)Úforce_bytesÚ	force_str)Úcached_propertyc              
   C   s„   t  ¡  ¡  d¡rtzdd l}W n. tk
rL } ztd| ƒ‚W 5 d }~X Y nX | d¡}| D ]\}}| ||¡ q\nt	j
 | ¡ d S )NZCYGWINr   z_Error loading ctypes: %s; the Oracle backend requires ctypes to operate correctly under Cygwin.Úkernel32)ÚplatformÚsystemÚupperÚ
startswithÚctypesÚImportErrorr   ÚCDLLZSetEnvironmentVariableAÚosÚenvironÚupdate)r   r   Úer   ÚnameÚvalue© r   úB/tmp/pip-unpacked-wheel-dv63caxw/django/db/backends/oracle/base.pyÚ_setup_environment   s    þ
r   )ZNLS_LANGz	.AL32UTF8)ZORA_NCHAR_LITERAL_REPLACEÚTRUEz"Error loading cx_Oracle module: %sé   )ÚDatabaseClient)ÚDatabaseCreation)ÚDatabaseFeatures)ÚDatabaseIntrospection)ÚDatabaseOperations)ÚDatabaseSchemaEditor)ÚOracle_datetime)ÚDatabaseValidationc               
   c   s€   z
d V  W np t jk
rz }  zP| jd }t|dƒrht|dƒrh|jdkrhd|jksZd|jkrhtt| jƒŽ ‚‚ W 5 d } ~ X Y nX d S )Nr   ÚcodeÚmessagei+  z	ORA-02291z	ORA-00001)ÚDatabaseÚDatabaseErrorÚargsÚhasattrr&   r'   r   Útuple)r   Úxr   r   r   Úwrap_oracle_errors?   s     

ÿþýüür.   c                   @   s   e Zd Zddd„ZdS )Ú!_UninitializedOperatorsDescriptorNc                 C   s&   |d krt dƒ‚| ¡  ¡  |jd S )Nz*operators not available as class attributeÚ	operators)ÚAttributeErrorÚcursorÚcloseÚ__dict__)ÚselfÚinstanceÚclsr   r   r   Ú__get__[   s    z)_UninitializedOperatorsDescriptor.__get__)N)Ú__name__Ú
__module__Ú__qualname__r8   r   r   r   r   r/   Y   s   r/   c                       sŠ  e Zd 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ddœZdddddddœZdZeƒ Z	dddddd d!d"ddddd#œZ
e
d$d%d$d$d%d%d&œ–Zd'Zd(d)d*d+d,d-d.œZd/d0„ e ¡ D ƒZd1d0„ e ¡ D ƒZeZeZeZeZeZeZeZeZ‡ fd2d3„Z d4d5„ Z!d6d7„ Z"d8d9„ Z#e$d:d;„ ƒZ%d<d=„ Z&e$dOd?d@„ƒZ'dAdB„ Z(dCdD„ Z)dEdF„ Z*dPdGdH„Z+dIdJ„ Z,e-dKdL„ ƒZ.e-dMdN„ ƒZ/‡  Z0S )QÚDatabaseWrapperÚoracleZOraclez3NUMBER(11) GENERATED BY DEFAULT ON NULL AS IDENTITYz3NUMBER(19) GENERATED BY DEFAULT ON NULL AS IDENTITYZBLOBz	NUMBER(1)zNVARCHAR2(%(max_length)s)ZDATEÚ	TIMESTAMPz*NUMBER(%(max_digits)s, %(decimal_places)s)zINTERVAL DAY(9) TO SECOND(6)zDOUBLE PRECISIONz
NUMBER(11)ZNCLOBz
NUMBER(19)zVARCHAR2(15)zVARCHAR2(39)z2NUMBER(5) GENERATED BY DEFAULT ON NULL AS IDENTITYzVARCHAR2(%(max_length)s)zVARCHAR2(32))Z	AutoFieldZBigAutoFieldZBinaryFieldÚBooleanFieldZ	CharFieldZ	DateFieldZDateTimeFieldZDecimalFieldZDurationFieldZ	FileFieldZFilePathFieldZ
FloatFieldZIntegerFieldÚ	JSONFieldZBigIntegerFieldZIPAddressFieldZGenericIPAddressFieldÚNullBooleanFieldZOneToOneFieldÚPositiveBigIntegerFieldÚPositiveIntegerFieldÚPositiveSmallIntegerFieldZ	SlugFieldZSmallAutoFieldZSmallIntegerFieldZ	TextFieldZ	TimeFieldZURLFieldZ	UUIDFieldz%(qn_column)s IN (0,1)z%(qn_column)s IS JSONz%(qn_column)s >= 0)r?   r@   rA   rB   rC   rD   )ZclobZnclobZblobz= %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 '\')rF   rG   r   rJ   rK   rL   z@REPLACE(REPLACE(REPLACE({}, '\', '\\'), '%%', '\%%'), '_', '\_')z'%%' || {} || '%%'z'%%' || UPPER({}) || '%%'z
{} || '%%'zUPPER({}) || '%%'z
'%%' || {}z'%%' || UPPER({}))rF   rG   r   rK   rJ   rL   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   rM   r   r   r   rQ   Ë   s   ÿ c                    s,   t ƒ j||Ž | jd  dd¡}|| j_d S )NÚOPTIONSÚuse_returning_intoT)ÚsuperÚ__init__Úsettings_dictÚgetÚfeaturesZcan_return_columns_from_insert)r5   r*   ÚkwargsrS   ©Ú	__class__r   r   rU   Ø   s    zDatabaseWrapper.__init__c                 C   sH   | j }|d  ¡ sd|d< |d r@t |d t|d ƒ|d ¡S |d S )NZHOSTÚ	localhostZPORTÚNAME)rV   Ústripr(   ZmakedsnÚint)r5   rV   r   r   r   Ú_dsnÝ   s    zDatabaseWrapper._dsnc                 C   s   d| j d | j d |  ¡ f S )Nz
%s/"%s"@%sÚUSERÚPASSWORD)rV   r`   ©r5   r   r   r   Ú_connect_stringå   s    zDatabaseWrapper._connect_stringc                 C   s    | j d  ¡ }d|kr|d= |S )NrR   rS   )rV   Úcopy©r5   Zconn_paramsr   r   r   Úget_connection_paramsè   s    z%DatabaseWrapper.get_connection_paramsc                 C   s(   t jf | jd | jd |  ¡ dœ|—ŽS )Nra   rb   )ÚuserÚpasswordZdsn)r(   ÚconnectrV   r`   rf   r   r   r   Úget_new_connectionî   s    ýüz"DatabaseWrapper.get_new_connectionc                 C   s¼   |   ¡ }| d¡ | dtjr"dnd ¡ | ¡  d| jkr |   ¡ }z| d| jd  dg¡ W n& tjk
r†   | j	| _
| j| _Y nX | j| _
| j| _| ¡  d	| j_|  ¡ s¸|  ¡  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'Ú r0   z!SELECT 1 FROM DUAL WHERE DUMMY %srF   ÚXé   )Úcreate_cursorÚexecuter   ÚUSE_TZr3   r4   Ú_standard_operatorsr(   r)   Ú_likec_operatorsr0   Ú_likec_pattern_opsZpattern_opsÚ_standard_pattern_opsÚ
connectionZstmtcachesizeZget_autocommitÚcommit©r5   r2   r   r   r   Úinit_connection_state÷   s2    
þÿ
ÿþz%DatabaseWrapper.init_connection_stateNc                 C   s
   t | jƒS ©N)ÚFormatStylePlaceholderCursorrv   )r5   r   r   r   r   ro     s    zDatabaseWrapper.create_cursorc              
   C   s2   | j d k	r.tƒ � | j  ¡ W  5 Q R £ S Q R X d S rz   )rv   r.   rw   rc   r   r   r   Ú_commit#  s    
zDatabaseWrapper._commitc                 C   s(   | j r$| j d| j |¡ ddœ¡ d S )Nz-- RELEASE SAVEPOINT %s (faked)z0.000)ZsqlÚtime)Zqueries_loggedZqueries_logÚappendÚopsZ
quote_name)r5   Zsidr   r   r   Ú_savepoint_commit*  s
    þz!DatabaseWrapper._savepoint_commitc              	   C   s   | j � || j_W 5 Q R X d S rz   )Zwrap_database_errorsrv   Ú
autocommit)r5   r�   r   r   r   Ú_set_autocommit1  s    zDatabaseWrapper._set_autocommitc              	   C   s,   |   ¡ �}| d¡ | d¡ W 5 Q R X dS )zl
        Check constraints by setting them to immediate. Return them to deferred
        afterward.
        zSET CONSTRAINTS ALL IMMEDIATEzSET CONSTRAINTS ALL DEFERREDN)r2   rp   )r5   Ztable_namesr2   r   r   r   Úcheck_constraints5  s    

z!DatabaseWrapper.check_constraintsc                 C   s0   z| j  ¡  W n tjk
r&   Y dS X dS d S )NFT)rv   Zpingr(   ÚErrorrc   r   r   r   Ú	is_usable>  s
    zDatabaseWrapper.is_usablec                 C   s   t dd„ tj d¡D ƒƒS )Nc                 s   s   | ]}t |ƒV  qd S rz   ©r_   ©rN   r-   r   r   r   Ú	<genexpr>H  s     z4DatabaseWrapper.cx_oracle_version.<locals>.<genexpr>Ú.)r,   r(   ÚversionÚsplitrc   r   r   r   Úcx_oracle_versionF  s    z!DatabaseWrapper.cx_oracle_versionc              
   C   s<   |   ¡ �* tdd„ | jj d¡D ƒƒW  5 Q R £ S Q R X d S )Nc                 s   s   | ]}t |ƒV  qd S rz   r†   r‡   r   r   r   rˆ   M  s     z1DatabaseWrapper.oracle_version.<locals>.<genexpr>r‰   )Ztemporary_connectionr,   rv   rŠ   r‹   rc   r   r   r   Úoracle_versionJ  s    
zDatabaseWrapper.oracle_version)N)N)1r9   r:   r;   ÚvendorZdisplay_nameZ
data_typesZdata_type_check_constraintsZ_limited_data_typesr/   r0   rr   rs   Zpattern_escZ_pattern_opsÚitemsru   rt   r(   r#   ZSchemaEditorClassr   Zclient_classr   Zcreation_classr    Zfeatures_classr!   Zintrospection_classr"   Z	ops_classr%   Zvalidation_classrU   r`   rd   rg   r   rk   ry   ro   r|   r€   r‚   rƒ   r…   r
   rŒ   r�   Ú__classcell__r   r   rZ   r   r<   f   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r&t|tjƒr&t|tƒs&t |¡}d}|dkr8d}n|dkrDd}t|dƒr\| |¡| _nHt|t	j
tjfƒrv|| _n.t||j|ƒ| _t| jtƒr¤tt||j|ƒƒ}t|dƒr¸|j| _n.|dkrÊt	j| _nt|tjƒràt	j| _nd | _d S )Nr   Tr   FÚbind_parameterÚ
input_sizei   )r   rq   Ú
isinstanceÚdatetimer$   Zfrom_datetimer+   r’   r   r(   ÚBinaryÚ	timedeltar	   ÚcharsetÚstrÚlenr“   ZCLOBr>   )r5   Úparamr2   Zstrings_onlyZstring_sizer   r   r   rU   Z  s.    ÿ





zOracleParam.__init__N)F)r9   r:   r;   Ú__doc__rU   r   r   r   r   r‘   P  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 rz   ©Úvar)r5   rŸ   r   r   r   rU   †  s    zVariableWrapper.__init__c                 C   s   | j S rz   rž   rx   r   r   r   r’   ‰  s    zVariableWrapper.bind_parameterc                 C   s   t | j|ƒS rz   )ÚgetattrrŸ   )r5   Úkeyr   r   r   Ú__getattr__Œ  s    zVariableWrapper.__getattr__c                 C   s&   |dkr|| j |< nt| j||ƒ d S )NrŸ   )r4   ÚsetattrrŸ   )r5   r¡   r   r   r   r   Ú__setattr__�  s    zVariableWrapper.__setattr__N)r9   r:   r;   rœ   rU   r’   r¢   r¤   r   r   r   r   r�   ~  s
   r�   c                   @   sž   e Zd ZdZdZdd„ Zedd„ ƒZedd„ ƒZed	d
„ ƒZ	dd„ Z
dd„ Zdd„ Zd#dd„Zd$dd„Zd%dd„Zdd„ Zdd„ Zdd„ Zdd „ Zd!d"„ ZdS )&r{   zË
    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".
    zutf-8c                 C   s   |  ¡ | _ | j| j _d S rz   )r2   Ú_output_type_handlerZoutputtypehandler)r5   rv   r   r   r   rU   ž  s    
z%FormatStylePlaceholderCursor.__init__c                 C   s   d| krt  | ¡S t| ƒS )Nr‰   )ÚdecimalÚDecimalr_   )r   r   r   r   Ú_output_number_converter¢  s    z5FormatStylePlaceholderCursor._output_number_converterc                    s8   |dkrt S tj| d�‰ t d¡ | ¡‰‡ ‡fdd„S )Nr   )Úprecr   c                    s   t  | ¡jˆˆ d�S )N)Úcontext)r¦   r§   Úquantize)rP   ©rª   Zquantize_valuer   r   Ú<lambda>¬  ó    zEFormatStylePlaceholderCursor._get_decimal_converter.<locals>.<lambda>)r_   r¦   ÚContextr§   Úscaleb)Ú	precisionÚscaler   r¬   r   Ú_get_decimal_converter¦  s
    z3FormatStylePlaceholderCursor._get_decimal_converterc                 C   s^   |t jkrZ|dkr(|dkr"tj}qDt}n|dkr>t ||¡}ntj}| jt jd| j|d�S dS )zx
        Called for each db column fetched from cursors. Return numbers as the
        appropriate Python type.
        i�ÿÿÿr   éÿ   )ÚsizeÚ	arraysizeÚoutconverterN)	r(   ÚNUMBERr{   r¨   Úfloatr³   rŸ   ÚSTRINGr¶   )r2   r   ZdefaultTypeÚlengthr±   r²   r·   r   r   r   r¥   ®  s    
üz1FormatStylePlaceholderCursor._output_type_handlerc                    sH   z‡ fdd„|  ¡ D ƒW S  tk
rB   t‡ fdd„|D ƒƒ Y S X d S )Nc                    s   i | ]\}}|t |ˆ d ƒ“qS )T©r‘   rM   rc   r   r   rQ   Ñ  s      z?FormatStylePlaceholderCursor._format_params.<locals>.<dictcomp>c                 3   s   | ]}t |ˆ d ƒV  qdS )TNr¼   ©rN   Úprc   r   r   rˆ   Ó  s     z>FormatStylePlaceholderCursor._format_params.<locals>.<genexpr>)r�   r1   r,   ©r5   Úparamsr   rc   r   Ú_format_paramsÏ  s    z+FormatStylePlaceholderCursor._format_paramsc                 C   s    t |d dƒrPi }|D ]&}| ¡ D ]\}}|jr"|j||< q"q|rœ| jf |Ž nLd gt|d ƒ }|D ]&}t|ƒD ]\}}|jrr|j||< qrqf|rœ| j|Ž  d S )Nr   Úkeys)r+   r�   r“   Zsetinputsizesrš   Ú	enumerate)r5   Zparams_listZsizesrÀ   rO   r   Úir   r   r   Ú_guess_input_sizesÕ  s    z/FormatStylePlaceholderCursor._guess_input_sizesc                 C   s.   t |dƒrdd„ | ¡ D ƒS dd„ |D ƒS d S )Nr�   c                 S   s   i | ]\}}||j “qS r   ©r   rM   r   r   r   rQ   ì  s      zAFormatStylePlaceholderCursor._param_generator.<locals>.<dictcomp>c                 S   s   g | ]
}|j ‘qS r   rÆ   r½   r   r   r   Ú
<listcomp>î  s     zAFormatStylePlaceholderCursor._param_generator.<locals>.<listcomp>)r+   r�   r¿   r   r   r   Ú_param_generatoré  s    
z-FormatStylePlaceholderCursor._param_generatorFc                    sÒ   |  d¡s|  d¡r |d d… }|d kr.g }n–t|dƒrPdd„ |D ƒ}|| }nt|r¢|r¢dd„ tt |¡ƒD ƒ‰ ‡ fdd	„|D ƒ}d
d„ ˆ  ¡ D ƒ}|t|ƒ }n"dd	„ tt|ƒƒD ƒ}|t|ƒ }||  	|¡fS )Nú;ú/éÿÿÿÿrÂ   c                 S   s   i | ]}|d | “qS )z:%sr   )rN   rO   r   r   r   rQ   û  s      z@FormatStylePlaceholderCursor._fix_for_params.<locals>.<dictcomp>c                 S   s   i | ]\}}|d | “qS ©z:arg%dr   )rN   rÄ   r›   r   r   r   rQ     s   ÿ c                    s   g | ]}ˆ | ‘qS r   r   )rN   r›   ©Úparams_dictr   r   rÇ   	  s     z@FormatStylePlaceholderCursor._fix_for_params.<locals>.<listcomp>c                 S   s   i | ]\}}||“qS r   r   )rN   r¡   r   r   r   r   rQ   
  s      c                 S   s   g | ]}d | ‘qS rÌ   r   )rN   rÄ   r   r   r   rÇ     s     )
rJ   r+   rÃ   ÚdictÚfromkeysr�   r,   Úrangerš   rÁ   )r5   ÚqueryrÀ   Úunify_by_valuesr*   r   rÍ   r   Ú_fix_for_paramsð  s"    

þz,FormatStylePlaceholderCursor._fix_for_paramsNc              
   C   sR   | j ||dd�\}}|  |g¡ tƒ �" | j ||  |¡¡W  5 Q R £ S Q R X d S )NT)rÓ   )rÔ   rÅ   r.   r2   rp   rÈ   )r5   rÒ   rÀ   r   r   r   rp     s    z$FormatStylePlaceholderCursor.executec              
      s€   |sd S t |ƒ}ˆ  |t|ƒ¡\}}|g‡ fdd„|D ƒ }ˆ  |¡ tƒ �* ˆ j |‡ fdd„|D ƒ¡W  5 Q R £ S Q R X d S )Nc                    s   g | ]}ˆ   |¡‘qS r   )rÁ   r½   rc   r   r   rÇ   !  s     z<FormatStylePlaceholderCursor.executemany.<locals>.<listcomp>c                    s   g | ]}ˆ   |¡‘qS r   )rÈ   r½   rc   r   r   rÇ   $  s     )ÚiterrÔ   ÚnextrÅ   r.   r2   Úexecutemany)r5   rÒ   rÀ   Zparams_iterZfirstparamsÚ	formattedr   rc   r   r×     s    
z(FormatStylePlaceholderCursor.executemanyc                 C   s*   z| j  ¡  W n tjk
r$   Y nX d S rz   )r2   r3   r(   ZInterfaceErrorrc   r   r   r   r3   &  s    z"FormatStylePlaceholderCursor.closec                 G   s   t | jj|Ž ƒS rz   )r�   r2   rŸ   ©r5   r*   r   r   r   rŸ   -  s    z FormatStylePlaceholderCursor.varc                 G   s   t | jj|Ž ƒS rz   )r�   r2   ÚarrayvarrÙ   r   r   r   rÚ   0  s    z%FormatStylePlaceholderCursor.arrayvarc                 C   s   t | j|ƒS rz   )r    r2   )r5   Úattrr   r   r   r¢   3  s    z(FormatStylePlaceholderCursor.__getattr__c                 C   s
   t | jƒS rz   )rÕ   r2   rc   r   r   r   Ú__iter__6  s    z%FormatStylePlaceholderCursor.__iter__)F)N)N)r9   r:   r;   rœ   r˜   rU   Ústaticmethodr¨   r³   r¥   rÁ   rÅ   rÈ   rÔ   rp   r×   r3   rŸ   rÚ   r¢   rÜ   r   r   r   r   r{   –  s(   


 
"

r{   )1rœ   r•   r¦   r   r   Ú
contextlibr   Zdjango.confr   Zdjango.core.exceptionsr   Z	django.dbr   Zdjango.db.backends.base.baser   Zdjango.utils.asyncior   Zdjango.utils.encodingr   r	   Zdjango.utils.functionalr
   r   Z	cx_Oracler(   r   r   Úclientr   Zcreationr   rX   r    Zintrospectionr!   Ú
operationsr"   Zschemar#   Úutilsr$   Z
validationr%   r.   r/   r<   r‘   r�   r{   r   r   r   r   Ú<module>   sJ   û	
 k.