o Df@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_14)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) constraintcompilerR/home/ubuntu/webapp/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!r rr(r)rrnew_column_namerr"r#rtrrrrr3sn        r3ops.DropTableOpcC|j||jdSr)r$ drop_tableto_tablemigration_contextr r rrrr;Ps r;ops.DropColumnOpcCs0||j}|jj|j|fd|ji|jdS)Nr) to_columnr=r$ drop_columnr)rr4)r r r5rrrrAWs  rAops.CreateIndexOpcCsH||j}i}|jdurtstd|j|d<|jj|fi|dS)NSQLAlchemy 1.4+ required if_not_exists)to_indexr=rDr NotImplementedErrorr$ create_index)r r idxr4rrrrGas   rGops.DropIndexOpcCsDi}|jdurts td|j|d<|jj||jfi|dS)NrC if_exists)rJr rFr$ drop_indexrEr=)r r r4rrrrKos    rKops.CreateTableOpr cCs||j}|j||Sr)r<r=r$ create_tabler r r/rrrrM~s  rMops.RenameTableOpcC|jj|j|j|jddSNr)r$ rename_tabler)new_table_namerr>rrrrR  rRops.CreateTableCommentOpcC||j}|j|dSr)r<r=r$create_table_commentrNrrrrW rWops.DropTableCommentOpcCrVr)r<r=r$drop_table_commentrNrrrrZrXrZops.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)r5rr4r/r r.r$ add_columnr1rrrr6indexesrGr%supports_commentsinline_commentsr"create_column_comment) r r r)r5rr4r8rindex with_commentr"rrrr\s*       r\ops.AddConstraintOpcCr:r)r$r6 to_constraintr=r>rrrcreate_constraints reops.DropConstraintOpcCs(|j|jj|j|j|j|jddSrQ)r$r2r.generic_constraintconstraint_namer)constraint_typerr>rrrr2sr2ops.BulkInsertOpcCrP)N) multiinsert)r$ bulk_insertr/rowsrkr>rrrrlrTrlops.ExecuteSQLOpcCs|jjj|j|jddS)N)execution_options)r=r$executesqltextror>rrr execute_sqls rr)r rr rrN)r rr r9rN)r rr r?rN)r rr rBrN)r rr rIrN)r rr rLrr )r rr rOrN)r rr rUrN)r rr rYrN)r rr r[rN)r rr rcrN)r rr rfrN)r rr rjrN)r rr rnrN)+typingr sqlalchemyrrrbaserutil.sqla_compatr r sqlalchemy.sql.schemar implementation_for AlterColumnOpr3 DropTableOpr; DropColumnOprA CreateIndexOprG DropIndexOprK CreateTableOprM RenameTableOprRCreateTableCommentOprWDropTableCommentOprZ AddColumnOpr\AddConstraintOpreDropConstraintOpr2 BulkInsertOprl ExecuteSQLOprrrrrrs         >