o
    ÕèFh[^  ã                   @   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 dd
lmZ ddlmZmZ ddlmZ ddlmZ dd„ Zeddgƒ 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. edd„ ƒZ/G dd„ dƒZ0G dd „ d eƒZ1G d!d"„ d"ƒZ2G d#d$„ d$ƒZ3G d%d&„ d&ƒZ4dS )'zc
Oracle database backend for Django.

Requires oracledb: https://oracle.github.io/python-oracledb/
é    N)Úcontextmanager)Úsettings)ÚImproperlyConfigured)ÚIntegrityError)ÚBaseDatabaseWrapper)Úoracledb)Údebug_transaction)Úasync_unsafe)Úforce_bytesÚ	force_str)Úcached_property)Úget_version_tuplec              
   C   s|   t  ¡  ¡  d¡r6zdd l}W n ty! } ztd| ƒ‚d }~ww | d¡}| D ]
\}}| ||¡ q)d S t	j
 | ¡ d S )NÚCYGWINr   z_Error loading ctypes: %s; the Oracle backend requires ctypes to operate correctly under Cygwin.Úkernel32)ÚplatformÚsystemÚupperÚ
startswithÚctypesÚImportErrorr   ÚCDLLÚSetEnvironmentVariableAÚosÚenvironÚupdate)r   r   Úer   ÚnameÚvalue© r   úW/var/www/html/loop/nvenv/lib/python3.10/site-packages/django/db/backends/oracle/base.pyÚ_setup_environment   s    þÿ€ÿ
ÿr    )ÚNLS_LANGz	.AL32UTF8)ÚORA_NCHAR_LITERAL_REPLACEÚTRUEé   )ÚDatabaseClient)ÚDatabaseCreation)ÚDatabaseFeatures)ÚDatabaseIntrospection)ÚDatabaseOperations)ÚDatabaseSchemaEditor)ÚOracle_datetimeÚdsn)ÚDatabaseValidationc               
   c   sv   � zd V  W d S  t jy: }  z&| jd }t|dƒr5t|dƒr5|jdkr5d|jv s.d|jv r5tt| jƒŽ ‚‚ d } ~ ww )Nr   ÚcodeÚmessagei+  z	ORA-02291z	ORA-00001)ÚDatabaseÚDatabaseErrorÚargsÚhasattrr.   r/   r   Útuple)r   Úxr   r   r   Úwrap_oracle_errorsC   s   €
ÿþ
€ír6   c                   @   s   e Zd Zddd„ZdS )Ú!_UninitializedOperatorsDescriptorNc                 C   s&   |d u 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__r@   r   r   r   r   r7   ]   s    r7   c                       sÎ  e Zd ZdZdZi 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“dd “d!d"“dddddd#dddd$d%d&œ¥Zd'd(d)d)d)d*œZd+Zeƒ Z	d,d-d.d/d0d1d2d3d.d.d/d/d4œZ
i e
¥d5d6d5d5d6d6d7œ¥Zd8Zd9d:d;d<d=d>d?œZd@dA„ e ¡ D ƒZdBdA„ e ¡ D ƒZeZeZeZeZeZeZeZeZ‡ fdCdD„Z dEdF„ Z!dGdH„ Z"e#dIdJ„ ƒZ$‡ fdKdL„Z%e#d^dNdO„ƒZ&dPdQ„ Z'dRdS„ Z(dTdU„ Z)d^dVdW„Z*dXdY„ Z+e,dZd[„ ƒZ-e,d\d]„ ƒZ.‡  Z/S )_ÚDatabaseWrapperÚoracleÚOracleÚ	AutoFieldz3NUMBER(11) GENERATED BY DEFAULT ON NULL AS IDENTITYÚBigAutoFieldz3NUMBER(19) GENERATED BY DEFAULT ON NULL AS IDENTITYÚBinaryFieldÚBLOBÚBooleanFieldz	NUMBER(1)Ú	CharFieldzNVARCHAR2(%(max_length)s)Ú	DateFieldÚDATEÚDateTimeFieldÚ	TIMESTAMPÚDecimalFieldz*NUMBER(%(max_digits)s, %(decimal_places)s)ÚDurationFieldzINTERVAL DAY(9) TO SECOND(6)Ú	FileFieldÚFilePathFieldÚ
FloatFieldzDOUBLE PRECISIONÚIntegerFieldz
NUMBER(11)Ú	JSONFieldÚNCLOBÚBigIntegerFieldz
NUMBER(19)ÚIPAddressFieldzVARCHAR2(15)ÚGenericIPAddressFieldzVARCHAR2(39)z2NUMBER(5) GENERATED BY DEFAULT ON NULL AS IDENTITYzVARCHAR2(%(max_length)s)zVARCHAR2(32))ÚOneToOneFieldÚPositiveBigIntegerFieldÚPositiveIntegerFieldÚPositiveSmallIntegerFieldÚ	SlugFieldÚSmallAutoFieldÚSmallIntegerFieldÚ	TextFieldÚ	TimeFieldÚURLFieldÚ	UUIDFieldz%(qn_column)s IN (0,1)z%(qn_column)s IS JSONz%(qn_column)s >= 0)rL   rX   r^   r_   r`   )ÚclobÚnclobÚ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)ÚexactÚiexactÚcontainsÚ	icontainsÚgtÚgteÚltÚlter   ÚendswithÚistartswithÚ	iendswithzLIKEC %s ESCAPE '\'zLIKEC UPPER(%s) ESCAPE '\')rm   rn   r   rs   rt   ru   z@REPLACE(REPLACE(REPLACE({}, '\', '\\'), '%%', '\%%'), '_', '\_')z'%%' || {} || '%%'z'%%' || UPPER({}) || '%%'z
{} || '%%'zUPPER({}) || '%%'z
'%%' || {}z'%%' || UPPER({}))rm   rn   r   rt   rs   ru   c                 C   ó   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   rv   )zLIKEC z ESCAPE '\'r   rw   r   r   r   r{   Ý   s    ÿc                    s0   t ƒ j|i |¤Ž | jd  dd¡}|| j_d S )NÚOPTIONSÚuse_returning_intoT)ÚsuperÚ__init__Úsettings_dictÚgetÚfeaturesÚcan_return_columns_from_insert)r=   r2   Úkwargsr}   ©Ú	__class__r   r   r   ë   s
   
ÿzDatabaseWrapper.__init__c                 C   ó   | j S rA   )Úoracle_version©r=   r   r   r   Úget_database_versionò   ó   z$DatabaseWrapper.get_database_versionc                 C   s    | j d  ¡ }d|v r|d= |S )Nr|   r}   )r€   Úcopy©r=   Úconn_paramsr   r   r   Úget_connection_paramsõ   s   z%DatabaseWrapper.get_connection_paramsc                 C   s*   t jd| jd | jd t| jƒdœ|¤ŽS )NÚUSERÚPASSWORD)ÚuserÚpasswordr,   r   )r0   Úconnectr€   r,   r�   r   r   r   Úget_new_connectionû   s   ýüz"DatabaseWrapper.get_new_connectionc                    sÈ   t ƒ  ¡  |  ¡ }| d¡ | dtjrdnd ¡ | ¡  d| jvrT|  ¡ }z| d| jd  dg¡ W n t	j
yG   | j| _| j| _Y n	w | j| _| j| _| ¡  d	| j_|  ¡ sb|  ¡  d 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'Ú r8   z!SELECT 1 FROM DUAL WHERE DUMMY %srm   ÚXé   )r~   Úinit_connection_stateÚcreate_cursorÚexecuter   ÚUSE_TZr;   r<   Ú_standard_operatorsr0   r1   Ú_likec_operatorsr8   Ú_likec_pattern_opsÚpattern_opsÚ_standard_pattern_opsÚ
connectionÚstmtcachesizeÚget_autocommitÚcommit©r=   r:   r…   r   r   r™     s:   

þÿ
ÿýþÿz%DatabaseWrapper.init_connection_stateNc                 C   s   t | j| ƒS rA   )ÚFormatStylePlaceholderCursorr¢   )r=   r   r   r   r   rš   /  s   zDatabaseWrapper.create_cursorc              	   C   s|   | j d ur<t| dƒ�( tƒ � | j  ¡ W  d   ƒ W  d   ƒ S 1 s%w   Y  W d   ƒ d S 1 s5w   Y  d S d S )NÚCOMMIT)r¢   r   r6   r¥   r‰   r   r   r   Ú_commit3  s
   
RÿÿzDatabaseWrapper._commitc                 C   s,   | j r| j d| j |¡ ddœ¡ d S d S )Nz-- RELEASE SAVEPOINT %s (faked)z0.000)ÚsqlÚtime)Úqueries_loggedÚqueries_logÚappendÚopsÚ
quote_name)r=   Úsidr   r   r   Ú_savepoint_commit:  s   þÿÿz!DatabaseWrapper._savepoint_commitc                 C   s4   | j � || j_W d   ƒ d S 1 sw   Y  d S rA   )Úwrap_database_errorsr¢   Ú
autocommit)r=   r´   r   r   r   Ú_set_autocommitC  s   
"ÿzDatabaseWrapper._set_autocommitc                 C   sB   |   ¡ �}| d¡ | d¡ W d  ƒ dS 1 sw   Y  dS )zl
        Check constraints by setting them to immediate. Return them to deferred
        afterward.
        zSET CONSTRAINTS ALL IMMEDIATEzSET CONSTRAINTS ALL DEFERREDN)r:   r›   )r=   Útable_namesr:   r   r   r   Úcheck_constraintsG  s   

"þz!DatabaseWrapper.check_constraintsc                 C   s(   z| j  ¡  W dS  tjy   Y dS w )NFT)r¢   Úpingr0   ÚErrorr‰   r   r   r   Ú	is_usableP  s   ýÿzDatabaseWrapper.is_usablec                 C   sH   |   ¡ � tdd„ | jj d¡D ƒƒW  d   ƒ S 1 sw   Y  d S )Nc                 s   s   � | ]}t |ƒV  qd S rA   )Úint)rx   r5   r   r   r   Ú	<genexpr>[  s   € z1DatabaseWrapper.oracle_version.<locals>.<genexpr>Ú.)Útemporary_connectionr4   r¢   ÚversionÚsplitr‰   r   r   r   rˆ   X  s   
$ÿzDatabaseWrapper.oracle_versionc                 C   s
   t tjƒS rA   )r   r0   Ú__version__r‰   r   r   r   Úoracledb_version]  s   
z DatabaseWrapper.oracledb_versionrA   )0rB   rC   rD   ÚvendorÚdisplay_nameÚ
data_typesÚdata_type_check_constraintsÚ_limited_data_typesr7   r8   r�   rž   Úpattern_escÚ_pattern_opsÚitemsr¡   rŸ   r0   r*   ÚSchemaEditorClassr%   Úclient_classr&   Úcreation_classr'   Úfeatures_classr(   Úintrospection_classr)   Ú	ops_classr-   Úvalidation_classr   rŠ   r�   r	   r•   r™   rš   r©   r²   rµ   r·   rº   r   rˆ   rÂ   Ú__classcell__r   r   r…   r   rE   i   sð    ÿþýüûúùø	÷
öõôóòñðïäû	çÿùú	ýÿ
+	
	
rE   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}|jjj}|s)|du r#d}n|du r)d}t	|dƒr5| 
|¡| _n$t|tjtjfƒrB|| _nt||j|ƒ| _t| jtƒrYtt||j|ƒƒ}t	|dƒrd|j| _d S |dkrntj| _d S t|tjƒrztj| _d S |r‡t|tƒr‡tj| _d S d | _d S )Nr   Tr$   FÚbind_parameterÚ
input_sizei   )r   rœ   Ú
isinstanceÚdatetimer+   Úfrom_datetimeÚdatabaser‚   Ú&supports_boolean_expr_in_select_clauser3   rÔ   r
   r0   ÚBinaryÚ	timedeltar   ÚcharsetÚstrÚlenrÕ   ÚCLOBrQ   ÚboolÚDB_TYPE_BOOLEAN)r=   Úparamr:   Ústrings_onlyÚstring_sizeÚhas_boolean_data_typer   r   r   r   l  s<   
ÿþ
ÿ


zOracleParam.__init__N©F)rB   rC   rD   Ú__doc__r   r   r   r   r   rÓ   b  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 rA   ©Úvar)r=   rë   r   r   r   r      ó   
zVariableWrapper.__init__c                 C   r‡   rA   rê   r¦   r   r   r   rÔ   £  r‹   zVariableWrapper.bind_parameterc                 C   ó   t | j|ƒS rA   )Úgetattrrë   )r=   Úkeyr   r   r   Ú__getattr__¦  ó   zVariableWrapper.__getattr__c                 C   s(   |dkr|| j |< d S t| j||ƒ d S )Nrë   )r<   Úsetattrrë   )r=   rï   r   r   r   r   Ú__setattr__©  s   zVariableWrapper.__setattr__N)rB   rC   rD   rè   r   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 rA   )r:   Ú_output_type_handlerÚoutputtypehandlerrÙ   )r=   r¢   rÙ   r   r   r   r   ¹  s   


z%FormatStylePlaceholderCursor.__init__c                 C   s   d| v 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)rz   ©rú   Ú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   sz   |t jkr-|dkr|dkrtj}nt}n|dkrt ||¡}ntj}| jt jd| j|d�S |t j	kr;| jt j	| jd�S dS )z˜
        Called for each db column fetched from cursors. Return numbers as the
        appropriate Python type, and NCLOB with JSON as strings.
        i�ÿÿÿr   éÿ   )ÚsizeÚ	arraysizeÚoutconverter)r  N)
r0   ÚNUMBERr§   rø   Úfloatr  rë   ÚSTRINGr  ÚDB_TYPE_NCLOB)r:   r   ÚdefaultTypeÚlengthr  r  r  r   r   r   rô   Ê  s&   
ÿü
ÿz1FormatStylePlaceholderCursor._output_type_handlerc                    sB   z‡ fdd„|  ¡ D ƒW S  ty    t‡ fdd„|D ƒƒ Y S w )Nc                    s   i | ]\}}|t |ˆ d ƒ“qS )T©rÓ   rw   r‰   r   r   r{   ó  s    z?FormatStylePlaceholderCursor._format_params.<locals>.<dictcomp>c                 3   s   � | ]	}t |ˆ d ƒV  qdS )TNr  ©rx   Úpr‰   r   r   r¼   õ  s   € z>FormatStylePlaceholderCursor._format_params.<locals>.<genexpr>)rÊ   r9   r4   ©r=   Úparamsr   r‰   r   Ú_format_paramsñ  s
   ÿz+FormatStylePlaceholderCursor._format_paramsc                 C   s®   t |d dƒr-i }|D ]}| ¡ D ]\}}|jr|j||< qq|r+| jdi |¤Ž d S d S d gt|d ƒ }|D ]}t|ƒD ]\}}|jrJ|j||< q>q8|rU| j|Ž  d S d S )Nr   Úkeysr   )r3   rÊ   rÕ   Úsetinputsizesrß   Ú	enumerate)r=   Úparams_listÚsizesr  ry   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 )NrÊ   c                 S   s   i | ]\}}||j “qS r   ©r
   rw   r   r   r   r{     ó    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>)r3   rÊ   r  r   r   r   Ú_param_generator  s   
z-FormatStylePlaceholderCursor._param_generatorFc                    sà   |  d¡s
|  d¡r|d d… }|d u rg }nRt|dƒr(dd„ |D ƒ}||; }nA|rX|rXdd„ |D ƒ}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   )rx   ry   r   r   r   r{     ó    z@FormatStylePlaceholderCursor._fix_for_params.<locals>.<dictcomp>c                 S   s   g | ]}t |ƒ|f‘qS r   )Útype)rx   rã   r   r   r   r  -  r  z@FormatStylePlaceholderCursor._fix_for_params.<locals>.<listcomp>c                 S   s   i | ]	\}}|d | “qS ©z:arg%dr   )rx   r  Ú
param_typer   r   r   r{   .  s    ÿÿc                    s   g | ]}ˆ | ‘qS r   r   )rx   r&  ©Úparams_dictr   r   r  2  rÿ   c                 S   s   i | ]	\\}}}||“qS r   r   )rx   Ú_rã   Úplaceholderr   r   r   r{   3  s    ÿc                 S   s   g | ]}d | ‘qS r%  r   )rx   r  r   r   r   r  9  rÿ   )
rs   r3   r  ÚdictÚfromkeysrÊ   r4   Úrangerß   r  )r=   Úqueryr  Úunify_by_valuesr2   Úparam_typesr   r'  r   Ú_fix_for_params  s(   

þÿz,FormatStylePlaceholderCursor._fix_for_paramsNc                 C   s^   | j ||dd�\}}|  |g¡ tƒ � | j ||  |¡¡W  d   ƒ S 1 s(w   Y  d S )NT)r/  )r1  r  r6   r:   r›   r  )r=   r.  r  r   r   r   r›   =  s
   $ÿz$FormatStylePlaceholderCursor.executec                    sŒ   |sd S t |ƒ}ˆ  |t|ƒ¡\}}|g‡ fdd„|D ƒ }ˆ  |¡ tƒ � ˆ j |‡ fdd„|D ƒ¡W  d   ƒ S 1 s?w   Y  d S )Nc                    ó   g | ]}ˆ   |¡‘qS r   )r  r  r‰   r   r   r  L  r#  z<FormatStylePlaceholderCursor.executemany.<locals>.<listcomp>c                    r2  r   )r  r  r‰   r   r   r  P  r#  )Úiterr1  Únextr  r6   r:   Úexecutemany)r=   r.  r  Úparams_iterÚfirstparamsÚ	formattedr   r‰   r   r5  C  s   
ÿ$ÿz(FormatStylePlaceholderCursor.executemanyc                 C   s(   z| j  ¡  W d S  tjy   Y d S w rA   )r:   r;   r0   ÚInterfaceErrorr‰   r   r   r   r;   S  s
   þz"FormatStylePlaceholderCursor.closec                 G   ó   t | jj|Ž ƒS rA   )ré   r:   rë   ©r=   r2   r   r   r   rë   Z  ó   z FormatStylePlaceholderCursor.varc                 G   r:  rA   )ré   r:   Úarrayvarr;  r   r   r   r=  ]  r<  z%FormatStylePlaceholderCursor.arrayvarc                 C   rí   rA   )rî   r:   )r=   Úattrr   r   r   rð   `  rñ   z(FormatStylePlaceholderCursor.__getattr__c                 C   s
   t | jƒS rA   )r3  r:   r‰   r   r   r   Ú__iter__c  rì   z%FormatStylePlaceholderCursor.__iter__rç   rA   )rB   rC   rD   rè   rÝ   r   Ústaticmethodrø   r  rô   r  r  r  r1  r›   r5  r;   rë   r=  rð   r?  r   r   r   r   r§   °  s*    


&

+
r§   )5rè   r×   rö   r   r   Ú
contextlibr   Údjango.confr   Údjango.core.exceptionsr   Ú	django.dbr   Údjango.db.backends.base.baser   Ú&django.db.backends.oracle.oracledb_anyr   r0   Údjango.db.backends.utilsr   Údjango.utils.asyncior	   Údjango.utils.encodingr
   r   Údjango.utils.functionalr   Údjango.utils.versionr   r    Úclientr%   Úcreationr&   r‚   r'   Úintrospectionr(   Ú
operationsr)   Úschemar*   Úutilsr+   r,   Ú
validationr-   r6   r7   rE   rÓ   ré   r§   r   r   r   r   Ú<module>   sL    ûÿ
 z6