o 8hO@sddlmZddlmZddlmZddlmZddl m Z ddl m Z er,dd l m Z eej  d:ddZeejd;ddZeej   dddZeej   d?d d!Zeej "  d@d#d$Zeej %  dAd&d'Zeej (  dBd)d*Z eej!dCd,d-Z"eej# .  dDd/d0Z$eej% 1  dEd2d3Z&eej' 4  dFd5d6Z(eej) 7  dGd8d9Z*dS)H) TYPE_CHECKINGschema)ops) Operations)_copy)sqla_2)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) constraintcompilera/var/www/html/figdemos/bartoux_crm/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_defaultZ modify_namemodify_nullablemodify_commentr# schema_objtablerColumn constraintsdrop_constraint alter_columnkwcolumnadd_constraint)r r rr!r rr(r)rrZnew_column_namerr"r#rtrrrrr2sn        r2ops.DropTableOpcC8i}|jdur |j|d<|jj||jfi|dSN if_exists)r:r$ drop_tableto_tablemigration_contextr r r3rrrr;Ps    r;ops.DropColumnOpcCs4||j}|jj|j|f|j|jd|jdS)N)rr:)Z to_columnr=r$ drop_columnr)rr:r3)r r r4rrrr@Zs  r@ops.CreateIndexOpcCs<||j}i}|jdur|j|d<|jj|fi|dSN if_not_exists)to_indexr=rCr$ create_index)r r idxr3rrrrEhs   rEops.DropIndexOpcCr8r9)r:r$ drop_indexrDr=r>rrrrHss    rHops.CreateTableOpr cCs<i}|jdur |j|d<||j}|jj|fi||SrB)rCr<r=r$ create_table)r r r3r.rrrrJs    rJops.RenameTableOpcC|jj|j|j|jddS)Nr)r$ rename_tabler)new_table_namerr r rrrrM  rMops.CreateTableCommentOpcC||j}|j|dSr)r<r=r$create_table_commentr r r.rrrrS rSops.DropTableCommentOpcCrRr)r<r=r$drop_table_commentrTrrrrWrUrWops.AddColumnOpc Cs|j}|j}|j}|j}|jdurt|}|jj|||d}|jj||f||j d||j D]}t |t j s>|j|q0|jD]}|j|qB|jjjoU|jjj } |j} | re| rg|j|dSdSdS)Nr)rrC)r)r4rr3r.r r-r$ add_columnrCr0rrrr5indexesrEr%supports_commentsinline_commentsr"create_column_comment) r r r)r4rr3r6rindex with_commentr"rrrrYs:       rYops.AddConstraintOpcCs|j||jdSr)r$r5 to_constraintr=rOrrrcreate_constraints rbops.DropConstraintOpcCsTi}|jdurts td|j|d<|jj|jj|j|j|j |j dfi|dS)NzSQLAlchemy 2.0 requiredr:r) r:r NotImplementedErrorr$r1r-Zgeneric_constraintconstraint_namer)constraint_typerr>rrrr1s   r1ops.BulkInsertOpcCrL)N) multiinsert)r$ bulk_insertr.rowsrhrOrrrrirPriops.ExecuteSQLOpcCs|jjj|j|jddS)N)execution_options)r=r$executesqltextrlrOrrr execute_sqls ro)r rr rrN)r rr r7rN)r rr r?rN)r rr rArN)r rr rGrN)r rr rIrr )r rr rKrN)r rr rQrN)r rr rVrN)r rr rXrN)r rr r`rN)r rr rcrN)r rr rgrN)r rr rkrN)+typingr sqlalchemyrrrbaserutil.sqla_compatr r sqlalchemy.sql.schemar Zimplementation_for AlterColumnOpr2 DropTableOpr; DropColumnOpr@ CreateIndexOprE DropIndexOprH CreateTableOprJZ RenameTableOprMCreateTableCommentOprSDropTableCommentOprW AddColumnOprYAddConstraintOprbDropConstraintOpr1Z BulkInsertOpriZ ExecuteSQLOprorrrrs         >           !