o
    :ZcDj                     @   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mZ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m Z  G dd deZ!dS )    N)	lru_cache)settings)BaseDatabaseOperations)strip_quotestruncate_name)ExistsExpressionWrapperRawSQL)Q)DatabaseError)timezone)force_bytes	force_str)cached_property   )Database)BulkInsertMapper	InsertVarOracle_datetimec                	       s&  e Zd ZdddddddddZi ejddiZdZd	Zd
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d!d" Z fd#d$Zd%d& Zd'd( Zd)d* Zd+d, Zd-d. Zd/d0 Zd1d2 Zed3d4 Zed5d6 Z d7d8 Z!d9d: Z"d;d< Z#d=d> Z$d?d@ Z%dAdB Z&dCdD Z'd}dFdGZ(dHdI Z)dJdK Z*dLdM Z+dNdO Z,dPdQ Z-dRdS Z.dTdU Z/dVdW Z0dXdY Z1dZd[ Z2e3d\d] Z4d~d_d`Z5dadb Z6dcdd Z7dedf Z8d~dgdhZ9didj Z:dkdl Z;dmdn Z< fdodpZ=dqdr Z>dsdt Z?dudv Z@ fdwdxZAdydz ZBd{d| ZC  ZDS )DatabaseOperations)lg]    g] )l9V l   9V )r   r   )iayi )SmallIntegerFieldIntegerFieldBigIntegerFieldPositiveSmallIntegerFieldPositiveIntegerFieldSmallAutoField	AutoFieldBigAutoField
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)z	NUMBER(5))r   r   r   	TextFieldc                 C      dS )NzSSELECT cache_key FROM %s ORDER BY cache_key OFFSET %%s ROWS FETCH FIRST 1 ROWS ONLY selfr#   r#   Z/var/www/chikooza/env/lib/python3.10/site-packages/django/db/backends/oracle/operations.pycache_key_culling_sqlF      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_sqlI   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_sqlX   s   z!DatabaseOperations.date_trunc_sqlz^[\w/:+-]+$c                 C   s8   d|v r|| dd  S d|v r|| dd  S |S )N+-)find)r%   tznamer#   r#   r&   _prepare_tzname_deltai   s
   z(DatabaseOperations._prepare_tzname_deltac                 C   sJ   t js|S | j|std| | jj|kr#d|| jj| |f S |S )NzInvalid time zone name: %sz8CAST((FROM_TZ(%s, '%s') AT TIME ZONE '%s') AS TIMESTAMP))r   USE_TZ
_tzname_rematch
ValueError
connectiontimezone_namer?   r%   r1   r>   r#   r#   r&   _convert_field_to_tzp   s   z'DatabaseOperations._convert_field_to_tzc                 C   s   |  ||}d| S )Nr9   rG   rF   r#   r#   r&   datetime_cast_date_sql   s   z)DatabaseOperations.datetime_cast_date_sqlc                 C   s   |  ||S NrH   rF   r#   r#   r&   datetime_cast_time_sql   s   z)DatabaseOperations.datetime_cast_time_sqlc                 C   s   |  ||}| ||S rJ   )rG   r2   )r%   r0   r1   r>   r#   r#   r&   datetime_extract_sql   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 )Nr3   r6   r+   r7   r*   r8   dayr9   hourTRUNC(%s, 'HH24')minuteTRUNC(%s, 'MI')CAST(%s AS DATE))rG   r.   )r%   r0   r1   r>   sqlr#   r#   r&   datetime_trunc_sql   s*   	z%DatabaseOperations.datetime_trunc_sqlc                 C   s<   |dkr
d| }|S |dkrd| }|S |dkrd| }|S )NrN   rO   rP   rQ   secondrR   r#   )r%   r0   r1   rS   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   r@   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&   r_      s2   
z$DatabaseOperations.get_db_convertersc                 C      t |tjr
| }|S rJ   )
isinstancer   LOBreadr%   valuern   rD   r#   r#   r&   rc         z*DatabaseOperations.convert_textfield_valuec                 C   s   t |tjrt| }|S rJ   )rt   r   ru   r   rv   rw   r#   r#   r&   rd      s   z,DatabaseOperations.convert_binaryfield_valuec                 C   s   |dv rt |}|S )N)r   r   )boolrw   r#   r#   r&   re      s   z-DatabaseOperations.convert_booleanfield_valuec                 C   s   |d urt || jj }|S rJ   )r   
make_awarerD   rw   r#   r#   r&   rf      s   z.DatabaseOperations.convert_datetimefield_valuec                 C   rs   rJ   )rt   r   	Timestampdaterw   r#   r#   r&   rg      ry   z*DatabaseOperations.convert_datefield_valuec                 C   rs   rJ   )rt   r   r|   timerw   r#   r#   r&   rh      ry   z*DatabaseOperations.convert_timefield_valuec                 C   s   |d ur	t |}|S rJ   )uuidUUIDrw   r#   r#   r&   ri      s   
z*DatabaseOperations.convert_uuidfield_valuec                 C      | d u rdS | S N r#   rx   rn   rD   r#   r#   r&   rm         z'DatabaseOperations.convert_empty_stringc                 C   r   )N    r#   r   r#   r#   r&   rl      r   z&DatabaseOperations.convert_empty_bytesc                 C   r"   )Nz DEFERRABLE INITIALLY DEFERREDr#   r$   r#   r#   r&   deferrable_sql   r(   z!DatabaseOperations.deferrable_sqlc                 C   s6   |j  }|d u s|g krtdt|tr|S |gS )NzThe 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getvaluer   rt   list)r%   cursorrx   r#   r#   r&   fetch_returned_insert_columns   s   
z0DatabaseOperations.fetch_returned_insert_columnsc                 C   s   |r	| dr	dS dS )Nru   zDBMS_LOB.SUBSTR(%s)%s)endswith)r%   db_typerp   r#   r#   r&   field_cast_sql  s   z!DatabaseOperations.field_cast_sqlc                 C   s   d S rJ   r#   r$   r#   r#   r&   no_limit_value  r(   z!DatabaseOperations.no_limit_valuec                 C   s@   |  ||\}}ddd |rd| nd |rd| nd fD S )N c                 s   s    | ]}|r|V  qd S rJ   r#   ).0rS   r#   r#   r&   	<genexpr>  s    z6DatabaseOperations.limit_offset_sql.<locals>.<genexpr>zOFFSET %d ROWSzFETCH FIRST %d ROWS ONLY)_get_limit_offset_paramsjoin)r%   low_mark	high_markfetchoffsetr#   r#   r&   limit_offset_sql  s
   

z#DatabaseOperations.limit_offset_sqlc                 C   sz   |j }t|ttfr!t|D ]\}}|d| t|dd}q|S t|tr;| D ]\}}|d| t|dd}q*|S )Nz:arg%dreplace)errorsz:%s)		statementrt   tupler   	enumerater   r   dictitems)r%   r   rS   paramsr   iparamkeyr#   r#   r&   last_executed_query  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%   r0   rp   r#   r#   r&   lookup_cast+  s   zDatabaseOperations.lookup_castc                 C   r"   )Ni  r#   r$   r#   r#   r&   max_in_list_size0  r(   z#DatabaseOperations.max_in_list_sizec                 C   r"   )N   r#   r$   r#   r#   r&   max_name_length3  r(   z"DatabaseOperations.max_name_lengthc                 C   r"   )NNULLr#   r$   r#   r#   r&   pk_default_value6  r(   z#DatabaseOperations.pk_default_valuec                 C      |S rJ   r#   )r%   xr#   r#   r&   prep_for_iexact_query9  r(   z(DatabaseOperations.prep_for_iexact_queryc                 C   s   |d u rdS |  S r   )rv   r%   rx   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   r   )r%   namer#   r#   r&   
quote_nameA  s   zDatabaseOperations.quote_namec                 C   r"   )NzDBMS_RANDOM.RANDOMr#   r$   r#   r#   r&   random_function_sqlN  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%   r0   match_optionr#   r#   r&   regex_lookupQ  s
   zDatabaseOperations.regex_lookupc                 C   sB   |sdS d|  |d jjj|  |d jf }|t|d ffS )N)r   r#   zRETURNING %s.%s INTO %%sr   )r   model_metadb_tablecolumnr   )r%   fieldsrS   r#   r#   r&   return_insert_columnsX  s   z(DatabaseOperations.return_insert_columnsc                 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)
                )rD   r   r   fetchall)r%   r   	recursiver   r#   r#   r&   __foreign_key_constraintsa  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-   r   tabler#   r#   r&   	<setcomp>  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>  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   rS   r#   r   r&   	sql_flush  s.   

zDatabaseOperations.sql_flushc           	      C   sh   g }|D ]-}|  |d }| |d }| |d pd}| j|||t|t|d }|| q|S )Nr   r   idno_autofield_sequence_namer   r   r   column_name)_get_no_autofield_sequence_namer   _sequence_reset_sqlr   rb   )	r%   r   r   rS   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   r   local_fieldsrt   r   r   r   r   r   rb   r   many_to_manyremote_fieldthroughm2m_db_table)r%   r   
model_listr   outputr   r   fr   r   r   r#   r#   r&   sequence_reset_sql  sB   
	

z%DatabaseOperations.sequence_reset_sqlc                 C   r"   r   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   r@   
make_naiverD   rC   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  rt   strdatetimestrptimer   r  rC   r   rN   rP   rU   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))r   r^   combine_expression)r%   	connectorsub_expressionsr  r  rq   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   8  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 )r   getattrra   r   typesgetrb   r   )	r%   r   placeholder_rowsr   r!  selectr   placeholderrp   r#   r#   r&   bulk_insert_sqlI  s   z"DatabaseOperations.bulk_insert_sqlc           	         sF   |dkr|\}}|\}}g ||R }d||f |fS t  |||S )Nr[   z*NUMTODSINTERVAL(TO_NUMBER(%s - %s), 'DAY'))r^   subtract_temporals)	r%   rp   r  r  lhs_sql
lhs_paramsrhs_sql
rhs_paramsr   rq   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.)rD   featuresmax_query_paramslen)r%   r   objsr#   r#   r&   bulk_batch_sizef  s   z"DatabaseOperations.bulk_batch_sizec                 C   s@   t |trdS t |trt |jtrdS t |tr|jrdS dS )z}
        Oracle supports only EXISTS(...) or filters in the WHERE clause, others
        must be compared with True.
        TF)rt   r   r   rn   r
   r	   conditional)r%   rn   r#   r#   r&   0conditional_expression_supported_in_where_clausel  s   
zCDatabaseOperations.conditional_expression_supported_in_where_clauserJ   )F)E__name__
__module____qualname__integer_field_rangesr   set_operatorsr   "cast_char_field_without_max_lengthcast_data_typesr'   r2   r:   recompilerA   r?   rG   rI   rK   rL   rT   rV   r_   rc   rd   re   rf   rg   rh   ri   staticmethodrm   rl   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*  r3  r5  __classcell__r#   r#   rq   r&   r      s    

	


	!

0"
		r   )"r  r=  r   	functoolsr   django.confr   "django.db.backends.base.operationsr   django.db.backends.utilsr   r   django.db.models.expressionsr   r   r	   django.db.models.query_utilsr
   django.db.utilsr   django.utilsr   django.utils.encodingr   r   django.utils.functionalr   baser   utilsr   r   r   r   r#   r#   r#   r&   <module>   s     