yj  ddlZddlmZddlmZdefdZddedejdzde fdZ dded edejdzde fd Z dded edejdzde fd Z dded edejdzde fdZ ddedededejdzde f dZded eddfdZdS)N)op)settingsreturnc$tjjSN)rDBSCHEMA8/DATA/AppData/hermes/projects/honcho/migrations/utils.py get_schemar s ; r table_name inspectorc|%tjtj}t }|||S)z(Check if a table exists in the database.Nschema)sainspectrget_bindr has_table)rrrs r table_existsr s?Jr{}}-- \\F   z&  9 99r column_namec|%tjtj}t }d|||D}||vS)z$Check if a column exists in a table.Ncg|] }|d S)namer ).0cols r z!column_exists..s)F r r)rrrrr get_columns)rrrrexisting_columnss r column_existsr!shJr{}}-- \\F(44Z4OO * **r fk_namec|%tjtj}t }|||}t fd|DS)z)Check if a foreign key exists in a table.Nrc3JK|]}|dkVdSrN)get)rfkr"s r zfk_exists..(s2@@Rrvvf~~(@@@@@@r )rrrrr get_foreign_keysany)rr"rr foreign_keyss ` r fk_existsr, sdJr{}}-- \\F--j-HHL @@@@<@@@ @ @@r index_namec|%tjtj}t }|||}t fd|DS)z$Check if an index exists in a table.Nrc30K|]}|dkVdSr%r )ridxr-s r r(zindex_exists..3s,<>G <<<.Ks-SSz&!_4SSSSSSr ) rrrrr get_check_constraintsget_unique_constraintsget_pk_constraintr) ValueErrorr*)rr4r5rr constraintsr<s ` r constraint_existsrB6sJr{}}-- \\F w55j5PP   66z&6QQ   00F0KK &!_44   00F0KK ;T;;<<< SSSS{SSS S SSr c t}tj}|dd}tjtj}|||}t fd|Dd}|td|dd|d std|dd dS|j }|j } | |} | |} | |} | } t||d s4| tjd | d| d | d| d | tjd| d| d| dtj|d|| tjd| d| d| ddS)a Make a column non-nullable using a non-blocking approach to minimize lock duration. WARNING: Only use this if you can guarantee that: 1. No NULL values currently exist in the column 2. The application code is already writing non-NULL values to this column or 3. The column has never accepted NULLs in practice This uses a 4-step process to avoid long exclusive locks: 1. Add CHECK constraint with NOT VALID (instant, no scan) 2. Validate the constraint (scans but allows concurrent read/writes to the table) 3. Set column NOT NULL (fast since we've validated the constraint) 4. Drop the redundant CHECK constraint Args: table_name: The name of the table column_name: The name of the column to make non-nullable _ _not_nullrc34K|]}|dk|VdSr%r )rrrs r r(z0make_column_non_nullable_safe..hs1MM#f+2L2L2L2L2L2LMMr NzColumn .z does not existnullablez% is already non-nullable, skipping...r7z ALTER TABLE z ADD CONSTRAINT z CHECK (z8 IS NOT NULL) NOT VALID z ALTER TABLE z! VALIDATE CONSTRAINT z F)rHrz ALTER TABLE z DROP CONSTRAINT z )r rrrrrnextr@printdialectidentifier_preparerquoterBexecutetext alter_column)rrrconnr4rcolumns column_inforKpreparer quoted_schema quoted_tablequoted_constraint quoted_columns ` r make_column_non_nullable_saferYNs&\\F ;==D#<>GMMMMwMMMtTTKL:LL LLLMMM z " W WW[WWWXXX lG*HNN6**M>>*--L 77NN;//M Z' B B   G*-9 1&    LL  &  )5  !2     O  LL  $  '3  ,     r r) sqlalchemyralembicr src.configrstrr Inspectorboolrr!r,r3rBrYr r r r`sC::S:R\D-@:D::::IM + + +"% +24,2E +  + + + +EIAAA!A.0lT.AA AAAAHL===!$=131D= ====&* TTTT T|d" T  TTTT0UcUUUUUUUUr