o h@sddlmZddlmZddlmZddlmZddl m Z er&ddl m Z e ej d9ddZe ejd:ddZe ej  d;ddZe ej  ddd Ze ej ! d?d"d#Ze ej $ d@d%d&Ze ej ' dAd(d)Ze ej dBd+d,Z!e ej" - dCd.d/Z#e ej$ 0 dDd1d2Z%e ej& 3 dEd4d5Z'e ej( 6 dFd7d8Z)dS)G) TYPE_CHECKINGschema)ops) Operations)_copy)Table operationsr operationops.AlterColumnOpreturnNc s |jj|jjd|j}|j}|j}|j}|j}|j}|j }|j } |j } |j } |j } |j} fdd}|rU|rU|jj|t|||d}|jD] }||rT|j|qH|jj||f| | | |||||| | d |j|r|jj||j|||d}|jD]}||r|j|qdSdS)Ncs t|tj o|j p|SN) isinstance sa_schemaPrimaryKeyConstraint _create_rule) constraintcompilerm/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/alembic/operations/toimpl.py_count_constraint%sz'alter_column.._count_constraintr) nullableserver_defaultnametype_r existing_typeexisting_server_defaultexisting_nullablecommentexisting_comment)impldialectstatement_compilerrr r modify_type column_name table_namermodify_server_default modify_namemodify_nullablemodify_commentr" schema_objtablerColumn constraintsdrop_constraint alter_columnkwcolumnadd_constraint)r r rr rrr'r(rrnew_column_namerr!r"rtrrrrr2sn        r2ops.DropTableOpcC8i}|jdur |j|d<|jj||jfi|dSN if_exists)r;r# drop_tableto_tablemigration_contextr r r3rrrr<Os    r<ops.DropColumnOpcCs0||j}|jj|j|fd|ji|jdS)Nr) to_columnr>r# drop_columnr(rr3)r r r4rrrrBYs  rBops.CreateIndexOpcCs<||j}i}|jdur|j|d<|jj|fi|dSN if_not_exists)to_indexr>rEr# create_index)r r idxr3rrrrGcs   rGops.DropIndexOpcCr9r:)r;r# drop_indexrFr>r?rrrrJns    rJops.CreateTableOpr cCs<i}|jdur |j|d<||j}|jj|fi||SrD)rEr=r>r# create_table)r r r3r.rrrrLzs    rLops.RenameTableOpcC|jj|j|j|jddSNr)r# rename_tabler(new_table_namerr r rrrrP  rPops.CreateTableCommentOpcC||j}|j|dSr)r=r>r#create_table_commentr r r.rrrrV rVops.DropTableCommentOpcCrUr)r=r>r#drop_table_commentrWrrrrZrXrZops.AddColumnOpc Cs|j}|j}|j}|j}|jdurt|}|jj|||d}|jj||fd|i||j D]}t |t j s<|j |q.|jD]}|j|q@|jjjoS|jjj } |j} | rc| re|j|dSdSdS)Nrr)r(r4rr3r.r r-r# add_columnr0rrrr5indexesrGr$supports_commentsinline_commentsr!create_column_comment) r r r(r4rr3r7rindex with_commentr!rrrr\s*       r\ops.AddConstraintOpcCs|j||jdSr)r#r5 to_constraintr>rRrrrcreate_constraints reops.DropConstraintOpcCs(|j|jj|j|j|j|jddSrO)r#r1r-generic_constraintconstraint_namer(constraint_typerrRrrrr1sr1ops.BulkInsertOpcCrN)N) multiinsert)r# bulk_insertr.rowsrkrRrrrrlrSrlops.ExecuteSQLOpcCs|jjj|j|jddS)N)execution_options)r>r#executesqltextrorRrrr execute_sqls rr)r rr r rN)r rr r8rN)r rr r@rN)r rr rCrN)r rr rIrN)r rr rKrr )r rr rMrN)r rr rTrN)r rr rYrN)r rr r[rN)r rr rcrN)r rr rfrN)r rr rjrN)r rr rnrN)*typingr sqlalchemyrrrbaserutil.sqla_compatr sqlalchemy.sql.schemar implementation_for AlterColumnOpr2 DropTableOpr< DropColumnOprB CreateIndexOprG DropIndexOprJ CreateTableOprL RenameTableOprPCreateTableCommentOprVDropTableCommentOprZ AddColumnOpr\AddConstraintOpreDropConstraintOpr1 BulkInsertOprl ExecuteSQLOprrrrrrs        >