o
    uvXj!c  ã                   @   s°   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
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mZ G dd„ deƒZdS )é    N)Ú	lru_cache)Úsettings)ÚBaseDatabaseOperations)Ústrip_quotesÚtruncate_name)ÚDatabaseError)Útimezone)Úforce_bytes)Úcached_propertyé   )ÚDatabase)ÚBulkInsertMapperÚInsertIdVarÚOracle_datetimec                       s"  e Zd ZddddddœZi ej¥ddi¥ZdZdZd	d
edœZdd„ Z	dd„ Z
dd„ Ze d¡Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Z‡ fdd „Zd!d"„ Zd#d$„ Zd%d&„ Zd'd(„ Zd)d*„ Zd+d,„ Zd-d.„ Zed/d0„ ƒZed1d2„ ƒZd3d4„ Z d5d6„ Z!d7d8„ Z"d9d:„ Z#d;d<„ Z$‡ fd=d>„Z%d?d@„ Z&dydBdC„Z'dDdE„ Z(dFdG„ Z)dHdI„ Z*dJdK„ Z+dLdM„ Z,dNdO„ Z-dPdQ„ Z.dRdS„ Z/dTdU„ Z0dVdW„ Z1e2dXdY„ ƒZ3dzd[d\„Z4d]d^„ Z5d_d`„ Z6dadb„ Z7dzdcdd„Z8dedf„ Z9dgdh„ Z:didj„ Z;‡ fdkdl„Z<dmdn„ Z=dodp„ Z>dqdr„ Z?‡ fdsdt„Z@dudv„ ZAe2‡ fdwdx„ƒZB‡  ZCS ){ÚDatabaseOperations)lýÿÿÿÿgí] ì   ÿgí] )lûÿÿÿÿÏ9V l   ÿÏ9V )r   r   )ÚSmallIntegerFieldÚIntegerFieldÚBigIntegerFieldÚPositiveSmallIntegerFieldÚPositiveIntegerFieldÚ
differenceÚMINUSaï  
DECLARE
    table_value integer;
    seq_value integer;
    seq_name user_tab_identity_cols.sequence_name%%TYPE;
BEGIN
    BEGIN
        SELECT sequence_name INTO seq_name FROM user_tab_identity_cols
        WHERE  table_name = '%(table_name)s' AND
               column_name = '%(column_name)s';
        EXCEPTION WHEN NO_DATA_FOUND THEN
            seq_name := '%(no_autofield_sequence_name)s';
    END;

    SELECT NVL(MAX(%(column)s), 0) INTO table_value FROM %(table)s;
    SELECT NVL(last_number - cache_size, 0) INTO seq_value FROM user_sequences
           WHERE sequence_name = seq_name;
    WHILE table_value > seq_value LOOP
        EXECUTE IMMEDIATE 'SELECT "'||seq_name||'".nextval FROM DUAL'
        INTO seq_value;
    END LOOP;
END;
/zNVARCHAR2(2000)z
NUMBER(11)z
NUMBER(19))Ú	AutoFieldÚBigAutoFieldÚ	TextFieldc                 C   ó   dS )NzSSELECT cache_key FROM %s ORDER BY cache_key OFFSET %%s ROWS FETCH FIRST 1 ROWS ONLY© ©Úselfr   r   úa/var/www/html/myproject/venv/lib/python3.10/site-packages/django/db/backends/oracle/operations.pyÚcache_key_culling_sql>   ó   z(DatabaseOperations.cache_key_culling_sqlc                 C   sP   |dkrd| S |dkrd| S |dkrd| S |dkr d| S d	|  ¡ |f S )
NÚweek_dayzTO_CHAR(%s, 'D')ÚweekzTO_CHAR(%s, 'IW')ÚquarterzTO_CHAR(%s, 'Q')Úiso_yearzTO_CHAR(%s, 'IYYY')zEXTRACT(%s FROM %s)©Úupper©r   Úlookup_typeÚ
field_namer   r   r    Údate_extract_sqlA   s   z#DatabaseOperations.date_extract_sqlc                 C   s@   |dv rd||  ¡ f S |dkrd| S |dkrd| S d| S )N©ÚyearÚmonthúTRUNC(%s, '%s')r%   úTRUNC(%s, 'Q')r$   úTRUNC(%s, 'IW')ú	TRUNC(%s)r'   r)   r   r   r    Údate_trunc_sqlP   s   z!DatabaseOperations.date_trunc_sqlz^[\w/:+-]+$c                 C   s.   t js|S | j |¡std| ƒ‚d||f S )NzInvalid time zone name: %sz:CAST((FROM_TZ(%s, '0:00') AT TIME ZONE '%s') AS TIMESTAMP))r   ÚUSE_TZÚ
_tzname_reÚmatchÚ
ValueError©r   r+   Útznamer   r   r    Ú_convert_field_to_tza   s
   z'DatabaseOperations._convert_field_to_tzc                 C   s   |   ||¡}d| S )Nr3   ©r;   r9   r   r   r    Údatetime_cast_date_sqlj   s   z)DatabaseOperations.datetime_cast_date_sqlc                 C   s   |   ||¡S ©Nr<   r9   r   r   r    Údatetime_cast_time_sqln   s   z)DatabaseOperations.datetime_cast_time_sqlc                 C   s   |   ||¡}|  ||¡S r>   )r;   r,   )r   r*   r+   r:   r   r   r    Údatetime_extract_sqls   s   z'DatabaseOperations.datetime_extract_sqlc                 C   s˜   |   ||¡}|dv rd|| ¡ f }|S |dkrd| }|S |dkr(d| }|S |dkr2d| }|S |d	kr<d
| }|S |dkrFd| }|S d| }|S )Nr-   r0   r%   r1   r$   r2   Údayr3   ÚhourúTRUNC(%s, 'HH24')ÚminuteúTRUNC(%s, 'MI')úCAST(%s AS DATE))r;   r(   )r   r*   r+   r:   Úsqlr   r   r    Údatetime_trunc_sqlw   s*   ôö	øúüÿz%DatabaseOperations.datetime_trunc_sqlc                 C   s<   |dkr
d| }|S |dkrd| }|S |dkrd| }|S )NrB   rC   rD   rE   ÚsecondrF   r   )r   r*   r+   rG   r   r   r    Útime_trunc_sqlŠ   s   üþz!DatabaseOperations.time_trunc_sqlc                    sÚ   t ƒ  |¡}|j ¡ }|dkr| | j¡ nD|dkr!| | j¡ n9|dv r,| | j¡ n.|dkr:tj	r9| | j
¡ n |dkrE| | j¡ n|dkrP| | j¡ n
|dkrZ| | j¡ |jjrk| |dkrg| jn| j¡ |S )Nr   ÚBinaryField)ÚBooleanFieldÚNullBooleanFieldÚDateTimeFieldÚ	DateFieldÚ	TimeFieldÚ	UUIDField)ÚsuperÚget_db_convertersÚoutput_fieldÚget_internal_typeÚappendÚconvert_textfield_valueÚconvert_binaryfield_valueÚconvert_booleanfield_valuer   r5   Úconvert_datetimefield_valueÚconvert_datefield_valueÚconvert_timefield_valueÚconvert_uuidfield_valueÚfieldÚempty_strings_allowedÚconvert_empty_bytesÚconvert_empty_string)r   Ú
expressionÚ
convertersÚinternal_type©Ú	__class__r   r    rS   –   s2   
€ÿýz$DatabaseOperations.get_db_convertersc                 C   ó   t |tjƒr
| ¡ }|S r>   )Ú
isinstancer   ÚLOBÚread©r   Úvaluerb   Ú
connectionr   r   r    rW   ³   ó   z*DatabaseOperations.convert_textfield_valuec                 C   s   t |tjƒrt| ¡ ƒ}|S r>   )rh   r   ri   r	   rj   rk   r   r   r    rX   ¸   s   z,DatabaseOperations.convert_binaryfield_valuec                 C   s   |dv rt |ƒ}|S )N)r   r   )Úboolrk   r   r   r    rY   ½   s   z-DatabaseOperations.convert_booleanfield_valuec                 C   s   |d urt  || jj ¡}|S r>   )r   Ú
make_awarerm   rk   r   r   r    rZ   Æ   s   z.DatabaseOperations.convert_datetimefield_valuec                 C   rg   r>   )rh   r   Ú	TimestampÚdaterk   r   r   r    r[   Ë   rn   z*DatabaseOperations.convert_datefield_valuec                 C   rg   r>   )rh   r   rq   Útimerk   r   r   r    r\   Ð   rn   z*DatabaseOperations.convert_timefield_valuec                 C   s   |d ur	t  |¡}|S r>   )ÚuuidÚUUIDrk   r   r   r    r]   Õ   s   
z*DatabaseOperations.convert_uuidfield_valuec                 C   ó   | d u rdS | S ©NÚ r   ©rl   rb   rm   r   r   r    ra   Ú   ó   z'DatabaseOperations.convert_empty_stringc                 C   rv   )Nó    r   ry   r   r   r    r`   Þ   rz   z&DatabaseOperations.convert_empty_bytesc                 C   r   )Nz DEFERRABLE INITIALLY DEFERREDr   r   r   r   r    Údeferrable_sqlâ   r"   z!DatabaseOperations.deferrable_sqlc              	   C   sF   z|j  ¡ }tt|tƒr|d ƒW S |ƒW S  ttfy"   tdƒ‚w )Nr   z¾The database did not return a new row id. Probably "ORA-1403: no data found" was raised internally but was hidden by the Oracle OCI library (see https://code.djangoproject.com/ticket/28859).)Ú_insert_id_varÚgetvalueÚintrh   ÚlistÚ
IndexErrorÚ	TypeErrorr   )r   Úcursorrl   r   r   r    Úfetch_returned_insert_idå   s   
 ÿþz+DatabaseOperations.fetch_returned_insert_idc                 C   s   |r	|  d¡r	dS dS )Nri   zDBMS_LOB.SUBSTR(%s)ú%s)Úendswith)r   Údb_typerd   r   r   r    Úfield_cast_sqlò   s   z!DatabaseOperations.field_cast_sqlc                 C   s   d S r>   r   r   r   r   r    Úno_limit_valueø   r"   z!DatabaseOperations.no_limit_valuec                 C   s8   |   ||¡\}}d|rd| nd|rd| f S df S )Nz%s%sz OFFSET %d ROWSrx   z FETCH FIRST %d ROWS ONLY)Ú_get_limit_offset_params)r   Úlow_markÚ	high_markÚfetchÚoffsetr   r   r    Úlimit_offset_sqlû   s   
þþz#DatabaseOperations.limit_offset_sqlc                    s   |j }tƒ  |||¡S r>   )Ú	statementrR   Úlast_executed_query)r   rƒ   rG   Úparamsr�   re   r   r    r‘     s   z&DatabaseOperations.last_executed_queryc                 C   s,   |   |t|ƒ|¡}| d| ¡ | ¡ d S )Nz"%s".currvalr   )Ú_get_sequence_namer   ÚexecuteÚfetchone)r   rƒ   Ú
table_nameÚpk_nameÚsq_namer   r   r    Úlast_insert_id
  s   z!DatabaseOperations.last_insert_idNc                 C   s   |dv rdS dS )N)ÚiexactÚ	icontainsÚistartswithÚ	iendswithz	UPPER(%s)r…   r   )r   r*   rd   r   r   r    Úlookup_cast  s   zDatabaseOperations.lookup_castc                 C   r   )Niè  r   r   r   r   r    Úmax_in_list_size  r"   z#DatabaseOperations.max_in_list_sizec                 C   r   )Né   r   r   r   r   r    Úmax_name_length  r"   z"DatabaseOperations.max_name_lengthc                 C   r   )NÚNULLr   r   r   r   r    Úpk_default_value  r"   z#DatabaseOperations.pk_default_valuec                 C   ó   |S r>   r   )r   Úxr   r   r    Úprep_for_iexact_query  r"   z(DatabaseOperations.prep_for_iexact_queryc                 C   s   |d u rdS |  ¡ S rw   )rj   ©r   rl   r   r   r    Úprocess_clob   s   zDatabaseOperations.process_clobc                 C   s>   |  d¡s| d¡sdt| ¡ |  ¡ ƒ }| dd¡}| ¡ S )Nú"z"%s"ú%ú%%)Ú
startswithr†   r   r(   r¡   Úreplace)r   Únamer   r   r    Ú
quote_name%  s   zDatabaseOperations.quote_namec                 C   r   )NzDBMS_RANDOM.RANDOMr   r   r   r   r    Úrandom_function_sql2  r"   z&DatabaseOperations.random_function_sqlc                 C   s    |dkr
d}d| S d}d| S )NÚregexz'c'z'i'zREGEXP_LIKE(%%s, %%s, %s)r   )r   r*   Úmatch_optionr   r   r    Úregex_lookup5  s
   ÿzDatabaseOperations.regex_lookupc                 C   s   dt ƒ ffS )NzRETURNING %s INTO %%s)r   r   r   r   r    Úreturn_insert_id<  s   z#DatabaseOperations.return_insert_idc                 C   sZ   | j  ¡ �}|r| d||f¡ n| d|f¡ | ¡ W  d   ƒ S 1 s&w   Y  d S )Na†  
                    SELECT
                        user_tables.table_name, rcons.constraint_name
                    FROM
                        user_tables
                    JOIN
                        user_constraints cons
                        ON (user_tables.table_name = cons.table_name AND cons.constraint_type = ANY('P', 'U'))
                    LEFT JOIN
                        user_constraints rcons
                        ON (user_tables.table_name = rcons.table_name AND rcons.constraint_type = 'R')
                    START WITH user_tables.table_name = UPPER(%s)
                    CONNECT BY NOCYCLE PRIOR cons.constraint_name = rcons.r_constraint_name
                    GROUP BY
                        user_tables.table_name, rcons.constraint_name
                    HAVING user_tables.table_name != UPPER(%s)
                    ORDER BY MAX(level) DESC
                a6  
                    SELECT
                        cons.table_name, cons.constraint_name
                    FROM
                        user_constraints cons
                    WHERE
                        cons.constraint_type = 'R'
                        AND cons.table_name = UPPER(%s)
                )rm   rƒ   r”   Úfetchall)r   r–   Ú	recursiverƒ   r   r   r    Ú__foreign_key_constraints?  s   ïø	$âz,DatabaseOperations.__foreign_key_constraintsc                 C   s   t dd�| jƒS )Ni   )Úmaxsize)r   Ú,_DatabaseOperations__foreign_key_constraintsr   r   r   r    Ú_foreign_key_constraints`  s   z+DatabaseOperations._foreign_key_constraintsFc                    s¬   |rTdd„ |D ƒ}t ƒ }|D ]}ˆ j||d�D ]\}}	|r"| |¡ | ||	f¡ qq‡ ‡fdd„|D ƒ‡ ‡fdd„|D ƒ ‡ ‡fdd„|D ƒ }
|
 ˆ  ˆ|¡¡ |
S g S )Nc                 S   s   h | ]}|  ¡ ’qS r   r'   ©Ú.0Útabler   r   r    Ú	<setcomp>h  s    z/DatabaseOperations.sql_flush.<locals>.<setcomp>)r¶   c                    sd   g | ].\}}d ˆ  d¡ˆ  d¡ˆ ˆ  |¡¡ˆ  d¡ˆ  d¡ˆ ˆ  |¡¡ˆ  d¡ˆ  d¡f ‘qS )z%s %s %s %s %s %s %s %s;ÚALTERÚTABLEÚDISABLEÚ
CONSTRAINTÚKEEPÚINDEX©ÚSQL_KEYWORDÚ	SQL_FIELDr¯   ©r¼   r½   Ú
constraint©r   Ústyler   r    Ú
<listcomp>t  s    
÷øÿz0DatabaseOperations.sql_flush.<locals>.<listcomp>c              
      s2   g | ]}d ˆ  d¡ˆ  d¡ˆ ˆ  |¡¡f ‘qS )z	%s %s %s;ÚTRUNCATErÀ   rÅ   r»   rÊ   r   r    rÌ     s    üýÿc                    sT   g | ]&\}}d ˆ  d¡ˆ  d¡ˆ ˆ  |¡¡ˆ  d¡ˆ  d¡ˆ ˆ  |¡¡f ‘qS )z%s %s %s %s %s %s;r¿   rÀ   ÚENABLErÂ   rÅ   rÈ   rÊ   r   r    rÌ   …  s    ùúÿ)Úsetrº   ÚaddÚextendÚsequence_reset_by_name_sql)r   rË   ÚtablesÚ	sequencesÚallow_cascadeÚtruncated_tablesÚconstraintsr½   Úforeign_tablerÉ   rG   r   rÊ   r    Ú	sql_flushf  s.   
ý
öûõøïzDatabaseOperations.sql_flushc           	      C   sh   g }|D ]-}|   |d ¡}|  |d ¡}|  |d pd¡}| j|||t|ƒt|ƒdœ }| |¡ q|S )Nr½   ÚcolumnÚid©Úno_autofield_sequence_namer½   rÚ   r–   Úcolumn_name)Ú_get_no_autofield_sequence_namer¯   Ú_sequence_reset_sqlr   rV   )	r   rË   rÔ   rG   Úsequence_inforÝ   r½   rÚ   Úqueryr   r   r    rÒ   –  s   ûz-DatabaseOperations.sequence_reset_by_name_sqlc                 C   sæ   ddl m} g }| j}|D ]c}|jjD ]/}t||jƒrB|  |jj¡}|  	|jj¡}	|  	|j
¡}
| |||	|
t|	ƒt|
ƒdœ ¡  nq|jjD ](}|jjso|  | ¡ ¡}|  	| ¡ ¡}	|  	d¡}
| |||	|
t|	ƒddœ ¡ qGq|S )Nr   )ÚmodelsrÜ   rÛ   ÚID)Ú	django.dbrã   rà   Ú_metaÚlocal_fieldsrh   r   rß   Údb_tabler¯   rÚ   rV   r   Úmany_to_manyÚremote_fieldÚthroughÚm2m_db_table)r   rË   Ú
model_listrã   Úoutputrâ   ÚmodelÚfrÝ   r½   rÚ   r   r   r    Úsequence_reset_sql¦  sB   
û	ó

û€ûz%DatabaseOperations.sequence_reset_sqlc                 C   r   rw   r   r   r   r   r    Ústart_transaction_sqlÈ  r"   z(DatabaseOperations.start_transaction_sqlc                 C   s    |r	d|   |¡ S d|   |¡ S )NzUSING INDEX TABLESPACE %szTABLESPACE %s)r¯   )r   Ú
tablespaceÚinliner   r   r    Útablespace_sqlË  s   z!DatabaseOperations.tablespace_sqlc                 C   r¤   )zô
        Transform a date value to an object compatible with what is expected
        by the backend driver for date columns.
        The default implementation transforms the date to text, but that is not
        necessary for Oracle.
        r   r§   r   r   r    Úadapt_datefield_valueÑ  s   z(DatabaseOperations.adapt_datefield_valuec                 C   sN   |du rdS t |dƒr|S t |¡r"tjrt || jj¡}ntdƒ‚t 	|¡S )aE  
        Transform a datetime value to an object compatible with what is expected
        by the backend driver for datetime columns.

        If naive datetime is passed assumes that is in UTC. Normally Django
        models.DateTimeField makes sure that if USE_TZ is True passed datetime
        is timezone aware.
        NÚresolve_expressionzNOracle backend does not support timezone-aware datetimes when USE_TZ is False.)
Úhasattrr   Úis_awarer   r5   Ú
make_naiverm   r8   r   Úfrom_datetimer§   r   r   r    Úadapt_datetimefield_valueÚ  s   



z,DatabaseOperations.adapt_datetimefield_valuec                 C   s`   |d u rd S t |dƒr|S t|tƒrtj |d¡S t |¡r"tdƒ‚tddd|j	|j
|j|jƒS )Nr÷   z%H:%M:%Sz5Oracle backend does not support timezone-aware times.il  r   )rø   rh   ÚstrÚdatetimeÚstrptimer   rù   r8   r   rB   rD   rI   Úmicrosecondr§   r   r   r    Úadapt_timefield_valueô  s   


ÿz(DatabaseOperations.adapt_timefield_valuec                    sš   |\}}|dkrdd  |¡ S |dkrdd  |¡ S |dkr%d||dœ S |d	kr0d
||dœ S |dkr;d||dœ S |dkrFdd  |¡ S tƒ  ||¡S )Nr«   zMOD(%s)ú,ú&z
BITAND(%s)ú|z"BITAND(-%(lhs)s-1,%(rhs)s)+%(lhs)s)ÚlhsÚrhsz<<z(%(lhs)s * POWER(2, %(rhs)s))z>>z"FLOOR(%(lhs)s / POWER(2, %(rhs)s))ú^z	POWER(%s))ÚjoinrR   Úcombine_expression)r   Ú	connectorÚsub_expressionsr  r  re   r   r    r	    s   z%DatabaseOperations.combine_expressionc                 C   s"   |   ¡ d }dtt|ƒ|ƒ ¡  S )z‹
        Manually created sequence name to keep backward compatibility for
        AutoFields that aren't Oracle identity columns.
        é   z%s_SQ)r¡   r   r   r(   )r   r½   Úname_lengthr   r   r    rß     s   z2DatabaseOperations._get_no_autofield_sequence_namec                 C   s2   |  d||g¡ | ¡ }|d u r|  |¡S |d S )Nzš
            SELECT sequence_name
            FROM user_tab_identity_cols
            WHERE table_name = UPPER(%s)
            AND column_name = UPPER(%s)r   )r”   r•   rß   )r   rƒ   r½   r—   Úrowr   r   r    r“     s
   üz%DatabaseOperations._get_sequence_namec           	      C   s’   g }|D ]=}g }t |ƒD ]*\}}|| r)t|| d|| ƒ ¡ }tj |d¡| }|s1d||f }| |¡ q| dd |¡ ¡ qdd |¡ S )NÚtarget_fieldr…   z	%s col_%szSELECT %s FROM DUALz, zSELECT * FROM (%s)z UNION ALL )Ú	enumerateÚgetattrrU   r   ÚtypesÚgetrV   r  )	r   ÚfieldsÚplaceholder_rowsrâ   r  ÚselectÚiÚplaceholderrd   r   r   r    Úbulk_insert_sql'  s   z"DatabaseOperations.bulk_insert_sqlc                    s<   |dkr|\}}|\}}d||f || fS t ƒ  |||¡S )NrO   z*NUMTODSINTERVAL(TO_NUMBER(%s - %s), 'DAY'))rR   Úsubtract_temporals)r   rd   r  r  Úlhs_sqlÚ
lhs_paramsÚrhs_sqlÚ
rhs_paramsre   r   r    r  <  s
   z%DatabaseOperations.subtract_temporalsc                 C   s   |r| j jjt|ƒ S t|ƒS )z5Oracle restricts the number of parameters in a query.)rm   ÚfeaturesÚmax_query_paramsÚlen)r   r  Úobjsr   r   r    Úbulk_batch_sizeC  s   z"DatabaseOperations.bulk_batch_sizec                    s   | j jjr	tƒ jS dS )Nz"django.db.backends.oracle.compiler)rm   r  Úhas_fetch_offset_supportrR   Úcompiler_moduler   re   r   r    r%  I  s   
z"DatabaseOperations.compiler_moduler>   )F)DÚ__name__Ú
__module__Ú__qualname__Úinteger_field_rangesr   Úset_operatorsrà   Ú"cast_char_field_without_max_lengthÚcast_data_typesr!   r,   r4   ÚreÚcompiler6   r;   r=   r?   r@   rH   rJ   rS   rW   rX   rY   rZ   r[   r\   r]   Ústaticmethodra   r`   r|   r„   rˆ   r‰   r�   r‘   r™   rž   rŸ   r¡   r£   r¦   r¨   r¯   r°   r³   r´   r¹   r
   rº   rÙ   rÒ   rñ   rò   rõ   rö   rü   r  r	  rß   r“   r  r  r#  r%  Ú__classcell__r   r   re   r    r      s�    ûý
		


!

0"
		r   )rþ   r-  rt   Ú	functoolsr   Údjango.confr   Ú"django.db.backends.base.operationsr   Údjango.db.backends.utilsr   r   Údjango.db.utilsr   Údjango.utilsr   Údjango.utils.encodingr	   Údjango.utils.functionalr
   Úbaser   Úutilsr   r   r   r   r   r   r   r    Ú<module>   s    