
    sgh9                     J    d dl mZ d dlmZ d dlmZ d dlmZ  G d de      Zy)    )BaseDatabaseSchemaEditor)IndexColumns)sql)strip_quotesc                        e Zd ZdZdZdZdZdZdZdZ	dZ
d	Zd
Zd# fd	ZdZdZd Z fdZd Zd Zd Zd Zd Zd Z fdZ	 d$ fd	Z fdZd$dZd$dZd% fd	Zddddddddddddd  fd!
Zd" Z xZ S )&DatabaseSchemaEditorzeUPDATE %(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                     |t         |   ||      S | j                  j                  j	                  t        |      |      }t         |   |d       S N)superexecute
connectionopscompose_sqlstr)selfr   params	__class__s      h/var/www/enzed_healthcare/enzed_env/lib/python3.12/site-packages/django/db/backends/postgresql/schema.pyr   zDatabaseSchemaEditor.execute*   sJ    >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    t        j                  || j                  j                        S r   )r   quoter   )r   values     r   quote_valuez DatabaseSchemaEditor.quote_value:   s    yy : :;;r   c                 r    t         |   ||      }| j                  ||      }||j                  |       |S r   )r   _field_indexes_sql_create_like_index_sqlappend)r   modelfieldoutputlike_index_statementr   s        r   r   z'DatabaseSchemaEditor._field_indexes_sql=   s?    +E59#::5%H+MM./r   c                     |j                   r|j                  | j                        S | j                  j                  j	                  |j                         |j                  | j                              S r   )is_relationrel_db_typer   
data_typesgetget_internal_typedb_typer   r    s     r   _field_data_typez%DatabaseSchemaEditor._field_data_typeD   sY    $$T__55))--##%MM$//*
 	
r   c              #      K   |j                   j                         dk(  r$| j                  |j                         E d {    y | j                  |j                          y 7 #w)N
ArrayField)
base_fieldr(   _field_base_data_typesr+   r*   s     r   r/   z+DatabaseSchemaEditor._field_base_data_typesL   sT     --/<?2253C3CDDD''(8(899 Es   <A$A"$A$c                    |j                  | j                        }||j                  s|j                  rd|v ryt	        |dd      }|s#|j
                  rt	        |j                  dd      }|r| j                  |      sy|j                  d      r| j                  ||gddg      S |j                  d	      r| j                  ||gdd
g      S y)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   z+DatabaseSchemaEditor._create_like_index_sqlS   s    
 --4??-;ENNell g~ %UNDAN!e&7&7!(););^T!Rd&F&F~&V!!),--!7"45	 .   ##F+--!7"12	 .   r   c                 0   |j                   ryd}|j                         }|j                         }|dk(  r=||k(  r8t        | j                  |            t        | j                  |            k7  r|S y| j	                  |      | j	                  |      k7  r|S y)N z USING %(column)s::%(type)sr-   )	generatedr(   listr/   r+   )r   	new_field	old_field	using_sqlnew_internal_typeold_internal_types         r   
_using_sqlzDatabaseSchemaEditor._using_sqlz   s    1	%779%779,1BFW1WD//	:;t++I6@  !   ""9-1F1Fy1QQr   c                     | j                   j                         5 }| j                   j                  j                  ||      D ]  }|d   |k(  s|d   c cd d d        S  	 d d d        y # 1 sw Y   y xY w)Ncolumnname)r   cursorintrospectionget_sequences)r   tablerO   rQ   sequences        r   _get_sequence_namez'DatabaseSchemaEditor._get_sequence_name   sw    __##% 	, OO99GGPUV ,H%/#F++	, 	,,	, 		, s   2A)A)A))A2c                     |j                   xs |j                  xrs |j                  d      xr |j                  d       xsL |j                  d      xr |j                  d       xs% |j                  d      xr |j                  d       S )Nr4   r:   citext)r<   r=   rA   )r   rI   old_typenew_types       r   (_is_changing_type_of_indexed_text_columnz=DatabaseSchemaEditor._is_changing_type_of_indexed_text_column   s    ""6i&6&6 
  +RH4G4G	4R0R U##F+OH4G4G4O0OU##H-Sh6I6I(6S2S	
r   c           	         |j                  | j                        }|d   }| j                  |||      rT| j                  |j                  j
                  |j                  gd      }	| j                  | j                  ||	             d| _	        | j                  ||      x}
r| xj                  |
z  c_	        |j                         }|j                         }t        |j                  j
                        }h d}||v }||v }|rn|slt        |j                        }| j                  | j                  |      |ddz  g f| j                  | j                  |      | j                  |      d	z  g fgfS |r|s| j                  | j                  | j                  |      | j                  t        |j                              d	z         t        |j                        }t         | E  ||||||      \  }}g }| j%                  ||      x}r#| j&                  d
| j                  |      iz  g fg}||fS |rz|rx||k7  rst         | E  ||||||      \  }}t        |j                        }dddd}g }| j%                  ||      x}r'| j(                  | j                  |      ||   dz  g fg}||fS t         | E  ||||||      S )Nr1   typer5   r8   z2ALTER COLUMN %(column)s TYPE %(type)s%(collation)s>   	AutoFieldBigAutoFieldSmallAutoFieldrE   )rO   r]   	collation)rT   rO   rU   integerbigintsmallint)r_   r`   ra   )rU   r]   )db_parametersr   r[   _create_index_name_metadb_tablerO   r   _delete_index_sqlsql_alter_column_typerM   r(   r   
quote_namesql_add_identitysql_drop_indentityr   _alter_column_type_sqlrV   sql_delete_sequencesql_alter_sequence_type)r   r   rI   rH   rZ   old_collationnew_collationold_db_paramsrY   
index_namerJ   rK   rL   rT   auto_field_typesold_is_autonew_is_autorO   fragment_other_actionssequence_namedb_typesr   s                          r   ro   z+DatabaseSchemaEditor._alter_column_type_sql   sJ   
 "//4??/K (88HhW00$$y'7'7&8 1 J LL//zBC A 	" 	9==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y)X}mKHa
 M $ 7 7v FF}F 00&(F ! ]**[->BS-S'8y)X}mKHa ")"2"23F& (",H M $ 7 7v FF}F 44(,(F$,->$?
 	! ]**71y)X}m r   Fc	           
         t         |   ||||||||       |j                  s|j                  s|j                  s+|j                  s|j                  s| j	                  |||      r%| j                  ||      }	|	| j                  |	       |j                  ro|j                  sb|j                  sU| j                  |j                  j                  |j                  gd      }
| j                  | j                  ||
             y y y y )Nr5   r^   )r   _alter_fieldr<   r=   r[   r   r   rg   rh   ri   rO   rj   )r   r   rI   rH   rY   rZ   rt   new_db_paramsstrictr"   index_to_remover   s              r   r   z!DatabaseSchemaEditor._alter_field  s    			
 $$	(8(8i>P>P$$)9)9==x
 $(#>#>ui#P #/12 Y%7%79;K;K"55$$y'7'7&8 6 O LL//GH	 <L%7r   c                 `    |rt        ||| j                  ||      S t        |   ||||      S )N)col_suffixesr9   )r   rl   r   _index_columns)r   rT   columnsr   r9   r   s        r   r   z#DatabaseSchemaEditor._index_columns1  s>    )#  w%eWlINNr   c                 N    | j                  |j                  || |      d        y )Nconcurrently)r   )r   
create_sqlr   r   indexr   s       r   	add_indexzDatabaseSchemaEditor.add_index<  s*    UD|DT 	 	
r   c                 J    | j                  |j                  || |             y )Nr   )r   
remove_sqlr   s       r   remove_indexz!DatabaseSchemaEditor.remove_indexA  s     U%%eT%MNr   Nc                 Z    |r| j                   n| j                  }t        |   |||      S r   )sql_delete_index_concurrentlysql_delete_indexr   rj   )r   r   rP   r   r   r   s        r   rj   z&DatabaseSchemaEditor._delete_index_sqlD  s7      ..&& 	
 w(c::r   rE   )r7   rP   r8   usingdb_tablespacer   r   r9   	conditionr   includeexpressionsc                v    |xs |s| j                   n| j                  }t        |   |||||||||	|
||      S )N)r7   rP   r8   r   r   r   r   r9   r   r   r   )sql_create_indexsql_create_index_concurrentlyr   rB   )r   r   r7   rP   r8   r   r   r   r   r9   r   r   r   r   r   s                 r   rB   z&DatabaseSchemaEditor._create_index_sqlL  sf    "  
 !!33 	
 w('%# ) 
 	
r   c                     | j                   j                         5 }|j                  d|g       |j                         }|r|d   nd cd d d        S # 1 sw Y   y xY w)Nz
                SELECT collisdeterministic
                FROM pg_collation
                WHERE collname = %s
                r   )r   rQ   r   fetchone)r   rC   rQ   rows       r   r@   z0DatabaseSchemaEditor._is_collation_deterministicq  s[    __##% 
	+NN
    //#C 3q6d
	+ 
	+ 
	+s   ,AA)r	   )F)NF)!__name__
__module____qualname__sql_update_with_defaultrq   rp   r   r   r   r   sql_create_column_inline_fksql_delete_fksql_delete_procedurer   rm   rn   r   r   r+   r/   r   rM   rV   r[   ro   r   r   r   r   rj   rB   r@   __classcell__)r   s   @r   r   r      s    	*  RH	9 
	9 " 7$P!
	<  	9  J*	+ 
 	P <
:%N 
mp (IT	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u+3 u+r   