o @j]@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) constraintcompilerh/var/www/www-root/data/www/ai-bot.pdev.uz/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.DropTableOpcCDi}|jdurts td|j|d<|jj||jfi|dSNSQLAlchemy 1.4+ required if_exists)r=r NotImplementedErrorr$ drop_tableto_tablemigration_contextr r r4rrrr?Ps    r?ops.DropColumnOpcCs0||j}|jj|j|fd|ji|jdS)Nr) to_columnrAr$ drop_columnr)rr4)r r r5rrrrE]s  rEops.CreateIndexOpcCsH||j}i}|jdurtstd|j|d<|jj|fi|dSNr< if_not_exists)to_indexrArHr r>r$ create_index)r r idxr4rrrrJgs   rJops.DropIndexOpcCr:r;)r=r r>r$ drop_indexrIrArBrrrrMus    rMops.CreateTableOpr cCsHi}|jdurts td|j|d<||j}|jj|fi||SrG)rHr r>r@rAr$ create_table)r r r4r/rrrrOs   rOops.RenameTableOpcC|jj|j|j|jddSNr)r$ rename_tabler)new_table_namerr r rrrrS  rSops.CreateTableCommentOpcC||j}|j|dSr)r@rAr$create_table_commentr r r/rrrrY rYops.DropTableCommentOpcCrXr)r@rAr$drop_table_commentrZrrrr]r[r]ops.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_columnr1rrrr6indexesrJr%supports_commentsinline_commentsr"create_column_comment) r r r)r5rr4r8rindex with_commentr"rrrr_s*       r_ops.AddConstraintOpcCs|j||jdSr)r$r6 to_constraintrArUrrrcreate_constraints rhops.DropConstraintOpcCs(|j|jj|j|j|j|jddSrR)r$r2r.generic_constraintconstraint_namer)constraint_typerrUrrrr2sr2ops.BulkInsertOpcCrQ)N) multiinsert)r$ bulk_insertr/rowsrnrUrrrrorVroops.ExecuteSQLOpcCs|jjj|j|jddS)N)execution_options)rAr$executesqltextrrrUrrr execute_sqls ru)r rr rrN)r rr r9rN)r rr rCrN)r rr rFrN)r rr rLrN)r rr rNrr )r rr rPrN)r rr rWrN)r rr r\rN)r rr r^rN)r rr rfrN)r rr rirN)r rr rmrN)r rr rqrN)+typingr sqlalchemyrrrbaserutil.sqla_compatr r sqlalchemy.sql.schemar implementation_for AlterColumnOpr3 DropTableOpr? DropColumnOprE CreateIndexOprJ DropIndexOprM CreateTableOprO RenameTableOprSCreateTableCommentOprYDropTableCommentOpr] AddColumnOpr_AddConstraintOprhDropConstraintOpr2 BulkInsertOpro ExecuteSQLOprurrrrs         >