
    h8                     J    S SK Jr  S SKJr  S SKJr  S SKJr   " S S\5      rg)    )BaseDatabaseSchemaEditor)IndexColumns)sql)strip_quotesc                   
  ^  \ rS rSrSrSrSrSrSrSr	Sr
S	rS
rSrS$U 4S jjrSrSrS rU 4S jrS rS rS rS rS rU 4S jr S%U 4S jjrU 4S jrS%S jrS%S jrS&U 4S jjrSSSSSSSSSSSSS .U 4S! jjrS" rS#r U =r!$ )'DatabaseSchemaEditor   zeUPDATE %(table)s SET %(column)s = %(default)s WHERE %(column)s IS NULL; SET CONSTRAINTS ALL IMMEDIATEz1ALTER SEQUENCE IF EXISTS %(sequence)s AS %(type)sz,DROP SEQUENCE IF EXISTS %(sequence)s CASCADEzZCREATE INDEX %(name)s ON %(table)s%(using)s (%(columns)s)%(include)s%(extra)s%(condition)szgCREATE INDEX CONCURRENTLY %(name)s ON %(table)s%(using)s (%(columns)s)%(include)s%(extra)s%(condition)szDROP INDEX IF EXISTS %(name)sz*DROP INDEX CONCURRENTLY IF EXISTS %(name)szyCONSTRAINT %(name)s REFERENCES %(to_table)s(%(to_column)s)%(deferrable)s; SET CONSTRAINTS %(namespace)s%(name)s IMMEDIATEzRSET CONSTRAINTS %(name)s IMMEDIATE; ALTER TABLE %(table)s DROP CONSTRAINT %(name)sz,DROP FUNCTION %(procedure)s(%(param_types)s) c                    > Uc  [         TU ]  X5      $ U R                  R                  R	                  [        U5      U5      n[         TU ]  US 5      $ N)superexecute
connectionopscompose_sqlstr)selfr   params	__class__s      V/var/www/html/env/lib/python3.13/site-packages/django/db/backends/postgresql/schema.pyr   DatabaseSchemaEditor.execute*   sH    >7?3//oo!!--c#h?wsD))    zRALTER TABLE %(table)s ALTER COLUMN %(column)s ADD GENERATED BY DEFAULT AS IDENTITYzEALTER TABLE %(table)s ALTER COLUMN %(column)s DROP IDENTITY IF EXISTSc                 V    [         R                  " XR                  R                  5      $ r   )r   quoter   )r   values     r   quote_value DatabaseSchemaEditor.quote_value:   s    yy : :;;r   c                 p   > [         TU ]  X5      nU R                  X5      nUb  UR                  U5        U$ r   )r   _field_indexes_sql_create_like_index_sqlappend)r   modelfieldoutputlike_index_statementr   s        r   r   'DatabaseSchemaEditor._field_indexes_sql=   s;    +E9#::5H+MM./r   c                     UR                   (       a  UR                  U R                  5      $ U R                  R                  R	                  UR                  5       UR                  U R                  5      5      $ r   )is_relationrel_db_typer   
data_typesgetget_internal_typedb_typer   r#   s     r   _field_data_type%DatabaseSchemaEditor._field_data_typeD   sY    $$T__55))--##%MM$//*
 	
r   c              #      #    UR                   R                  5       S:X  a$  U R                  UR                   5       S h  vN   g U R                  UR                   5      v   g  N#7f)N
ArrayField)
base_fieldr,   _field_base_data_typesr/   r.   s     r   r4   +DatabaseSchemaEditor._field_base_data_typesL   sS     --/<?2253C3CDDD''(8(899 Es   =A%A# $A%c                    UR                  U R                  S9nUb  UR                  (       d  UR                  (       a  SU;   a  g[	        USS5      nU(       d(  UR
                  (       a  [	        UR                  SS5      nU(       a  U R                  U5      (       d  gUR                  S5      (       a  U R                  UU/SS/S9$ UR                  S	5      (       a  U R                  UU/SS
/S9$ g)z
Return the statement to create an index with varchar operator pattern
when the column type is 'varchar' or 'text', otherwise return None.
r   N[db_collationvarchar_likevarchar_pattern_ops)fieldssuffix	opclassestexttext_pattern_ops)
r-   r   db_indexuniquegetattrr(   target_field_is_collation_deterministic
startswith_create_index_sql)r   r"   r#   r-   collation_names        r   r    +DatabaseSchemaEditor._create_like_index_sqlS   s    
 --4??-;ENNell g~ %UNDAN!e&7&7!(););^T!Rd&F&F~&V&V!!),,--!7"45	 .   ##F++--!7"12	 .   r   c                    SnUR                  5       nUR                  5       nUS:X  a?  XE:X  a:  [        U R                  U5      5      [        U R                  U5      5      :w  a  U$  gU R                  U5      U R                  U5      :w  a  U$ g)Nz USING %(column)s::%(type)sr2    )r,   listr4   r/   )r   	new_field	old_field	using_sqlnew_internal_typeold_internal_types         r   
_using_sqlDatabaseSchemaEditor._using_sqlz   s    1	%779%779,1B1WD//	:;t++I6@  !   ""9-1F1Fy1QQr   c                     U R                   R                  5        nU R                   R                  R                  X15       H  nUS   U:X  d  M  US   s  sS S S 5        $    S S S 5        g ! , (       d  f       g = f)Ncolumnname)r   cursorintrospectionget_sequences)r   tablerV   rX   sequences        r   _get_sequence_name'DatabaseSchemaEditor._get_sequence_name   si    __##% OO99GGVH%/#F++ &%V & 	 &% s   0A*A*A**
A8c           	        > UR                  U R                  S9nUS   nUR                  (       d  UR                  (       a  UR	                  S5      (       a  UR	                  S5      (       aX  UR	                  S5      (       a  UR	                  S5      (       a,  UR	                  S5      (       af  UR	                  S5      (       dP  U R                  UR                  R                  UR                  /SS9n	U R                  U R                  X5      5        SU l        U R                  X25      =n
(       a  U =R                  U
-  sl        UR                  5       nUR                  5       n[        UR                  R                  5      n1 S	knX;   nX;   nU(       as  U(       dl  [        UR                  5      nU R                  U R                  U5      US
S.-  / 4U R                   U R                  U5      U R                  U5      S.-  / 4/4$ U(       a  U(       d  U R                  U R"                  U R                  U5      U R                  [        UR                  5      5      S.-  5        [        UR                  5      n[$        TU ]M  XX4XV5      u  nn/ nU R)                  UU5      =n(       a#  U R*                  SU R                  U5      0-  / 4/nUU4$ U(       a  U(       az  X:w  au  [$        TU ]M  XX4XV5      u  nn[        UR                  5      nSSSS.n/ nU R)                  UU5      =n(       a'  U R,                  U R                  U5      UU   S.-  / 4/nUU4$ [$        TU ]M  XX4XV5      $ )Nr7   typer:   r@   citextr;   r>   z2ALTER COLUMN %(column)s TYPE %(type)s%(collation)s>   	AutoFieldBigAutoFieldSmallAutoFieldrL   )rV   r`   	collation)r[   rV   r\   integerbigintsmallint)rc   rd   re   )r\   r`   )db_parametersr   rB   rC   rG   _create_index_name_metadb_tablerV   r   _delete_index_sqlsql_alter_column_typerS   r,   r   
quote_namesql_add_identitysql_drop_indentityr   _alter_column_type_sqlr]   sql_delete_sequencesql_alter_sequence_type)r   r"   rO   rN   new_typeold_collationnew_collationold_db_paramsold_type
index_namerP   rQ   rR   r[   auto_field_typesold_is_autonew_is_autorV   fragment_other_actionssequence_namedb_typesr   s                          r   rs   +DatabaseSchemaEditor._alter_column_type_sql   s   
 "//4??/K ()"2"2  ++H4G4G	4R4R##F++H4G4G4O4O##H--h6I6I(6S6S00$$y'7'7&8 1 J LL//BC A 	" 	==9=&&)3&%779%779U[[112

 (;';{!)"2"23F .."&//&"9 (%'  --%)__U%;&*oof&=
 	 *  LL''!__U3"ool9;K;K.LM ")"2"23F'8)}KHa
 M $ 7 7v FF}F 00&(F ! ]**[->-S'8)}KHa ")"2"23F& (",H M $ 7 7v FF}F 44(,(F$,->$?
 	! ]**71)} r   Fc	           
      2  > [         TU ]  UUUUUUUU5        UR                  (       d"  UR                  (       d  UR                  (       d"  UR                  (       d6  UR                  (       a%  U R	                  X5      n	U	b  U R                  U	5        UR                  (       au  UR                  (       dc  UR                  (       dQ  U R                  UR                  R                  UR                  /SS9n
U R                  U R                  X5      5        g g g g )Nr;   rb   )r   _alter_fieldrB   rC   r    r   rk   rl   rm   rV   rn   )r   r"   rO   rN   rz   rv   ry   new_db_paramsstrictr%   index_to_remover   s              r   r   !DatabaseSchemaEditor._alter_field  s     			
 ##y'7'7Y=O=O  Y%5%5#'#>#>u#P #/12 Y%7%79;K;K"55$$y'7'7&8 6 O LL//GH	 <L%7r   c                 `   > U(       a  [        UUU R                  UUS9$ [        TU ]  XX45      $ )N)col_suffixesr?   )r   rp   r   _index_columns)r   r[   columnsr   r?   r   s        r   r   #DatabaseSchemaEditor._index_columns&  s:    )#  w%elNNr   c                 @    U R                  UR                  XUS9S S9  g )Nconcurrently)r   )r   
create_sqlr   r"   indexr   s       r   	add_indexDatabaseSchemaEditor.add_index1  s(    U|DT 	 	
r   c                 B    U R                  UR                  XUS95        g )Nr   )r   
remove_sqlr   s       r   remove_index!DatabaseSchemaEditor.remove_index6  s    U%%e%MNr   Nc                 b   > U(       a  U R                   OU R                  n[        TU ]  XU5      $ r   )sql_delete_index_concurrentlysql_delete_indexr   rn   )r   r"   rW   r   r   r   s        r   rn   &DatabaseSchemaEditor._delete_index_sql9  s5      ..&& 	
 w(c::r   rL   )r=   rW   r>   usingdb_tablespacer   r   r?   	conditionr   includeexpressionsc                   > U=(       d    U(       d  U R                   OU R                  n[        TU ]  UUUUUUUUU	U
UUS9$ )N)r=   rW   r>   r   r   r   r   r?   r   r   r   )sql_create_indexsql_create_index_concurrentlyr   rH   )r   r"   r=   rW   r>   r   r   r   r   r?   r   r   r   r   r   s                 r   rH   &DatabaseSchemaEditor._create_index_sqlA  sf    "  
 !!33 	
 w('%# ) 
 	
r   c                     U R                   R                  5        nUR                  SU/5        UR                  5       nU(       a  US   OS sS S S 5        $ ! , (       d  f       g = f)Nz
                SELECT collisdeterministic
                FROM pg_collation
                WHERE collname = %s
                r   )r   rX   r   fetchone)r   rI   rX   rows       r   rF   0DatabaseSchemaEditor._is_collation_deterministicf  sR    __##%NN
    //#C 3q6d &%%s   1A
A$)ro   )r
   )F)NF)"__name__
__module____qualname____firstlineno__sql_update_with_defaultru   rt   r   r   r   r   sql_create_column_inline_fksql_delete_fksql_delete_procedurer   rq   rr   r   r   r/   r4   r    rS   r]   rs   r   r   r   r   rn   rH   rF   __static_attributes____classcell__)r   s   @r   r   r      s    	*  RH	9 
	9 " 7$P!
	<  	9  J*	+ 
 	P <
:%Nqx "IH	O

O; #
 #
J+ +r   r   N)	django.db.backends.base.schemar   !django.db.backends.ddl_referencesr   )django.db.backends.postgresql.psycopg_anyr   django.db.backends.utilsr   r   r
   r   r   <module>r      s     C : 9 1j+3 j+r   