o Dft@sddlmZddlZddlZddlmZddlmZddlmZddlmZddlm Z ddlm Z dd lm Z dd lm Z dd lm Z dd lmZdd lmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddl m!Z!ddl"m#Z#ddl"m$Z$ddl"m%Z%ddl"m&Z&ddl"m'Z'ddl"m(Z(dd l"m)Z)dd!l"m*Z*dd"l"m+Z+dd#l,m-Z-dd$l,m.Z.d%d&l/m0Z0d%d'l1m2Z2d%d(l3m4Z4d%d)l3m5Z5d%d*l6m7Z7d%d+l6m8Z8d%d,l0m9Z9d%d-l:m;Z;e rsdd.lmZ>dd1l?m@Z@dd2lAmBZBdd3lCmDZDdd4lEmFZFdd5lEmGZGdd6lmHZHdd7lmIZIdd8lmJZJdd9lKmLZLdd:lKmMZMdd;lNmOZOddlSmTZTd%d?lUmVZVeWeXZYGd@dAdAe.ZZGdBdCdCe%Z[e;e+dDdodKdLZ\e;e[dDdpdMdNZ]e;e&dDdqdPdQZ^e;e*dDdrdSdTZ_e8`dUe7`dUdVe4jabdWGdXdYdYe4jaZce2jdeecdsd]d^Zfe2jgeedtdcddZhdudedfZidvdidjZjdwdmdnZkdS)x) annotationsN)Any)cast)Dict)List)Optional)Sequence)Tuple) TYPE_CHECKING)Union)Column)literal_column)Numeric)text)types)BIGINT)ExcludeConstraint)INTEGER) CreateIndex) ColumnClause) TextClause)FunctionElement)NULLTYPE) alter_column) alter_table) AlterColumn) ColumnComment)format_column_name)format_table_name) format_type)IdentityColumnDefault) RenameTable)ComparisonResult) DefaultImpl)util)render)ops) schemaobj)BatchOperations) Operations) sqla_compat)compiles)Literal)Index)UniqueConstraint)ARRAY) PGDDLCompiler)HSTORE)JSON)JSONB) ClauseElement) ColumnElement) quoted_name)MetaData)Table) TypeEngine)_ServerDefault)AutogenContext)_f_name)MigrationContextcs$eZdZdZdZejddhfZdcd d Zd dZddZ          dddefd%d& Z d'd(Z d)d*Z e d+e d,e d-e d.e d/fZdfd1d2Zdgd6d7Zdhd;d<Zdid@dAZdjdEdFZdkdIdJZ  dldmfdOdP ZdndUdVZdodXdYZdpd[d\Zdqd^d_ZdrdadbZZS)sPostgresqlImpl postgresqlTFLOATzDOUBLE PRECISIONindexr/kwrreturnNonecKsV|jddpd}|D]}||jjvr|jt|tjq |t |fi|dS)Npostgresql_include) kwargsgettablec append_columnr sqltypesNullType_execr)selfrCrDrGcolrHrHO/home/ubuntu/webapp/venv/lib/python3.10/site-packages/alembic/ddl/postgresql.py create_indexUs  zPostgresqlImpl.create_indexcCs0|jD]}|jdur|j|jvr||qdSN) constraintsnamenamed_constraintsdrop_constraint)rQ batch_implrK constraintrHrHrSprep_table_for_batchas    z#PostgresqlImpl.prep_table_for_batchc Cs|jr ||jjur dS|}||k}|rdSd|||jfvr | S|jj}t|tr@t|jtss  zAPostgresqlImpl.correct_for_autogen_constraints..)remover,sqla_2_skip_functional_indexes)rQconn_unique_constraints conn_indexesmetadata_unique_constraintsmetadata_indexesdoubled_constraintsixrHrHrScorrect_for_autogen_constraintss z.PostgresqlImpl.correct_for_autogen_constraintsz( asc nulls last)$z(?wsz=CreateExcludeConstraintOp.from_constraint..zColumnElement[bool] | None)rAr|rC deferrable initiallyr) r,_table_for_constraintrW _render_exprsrrAr|rGrHr)clsr[constraint_tablerHrHrSfrom_constraintos  z)CreateExcludeConstraintOp.from_constraintmigration_contextOptional[MigrationContext]cCsx|jdur|jSt|}|j|j|jd}t|j|j|j d|j }|j D] \}}}| t |tq'|||S)N)r|)rWrA)rCr) SchemaObjectsrKrtr|rr?r<rArDrJrMr rappend_constraint)rQrN schema_objtexclrrWoperrHrHrS to_constraints&    z'CreateExcludeConstraintOp.to_constraint operationsr+rdrrDOptional[Table]cOs||||fi|}||S)aIssue an alter to create an EXCLUDE constraint using the current migration context. .. note:: This method is Postgresql specific, and additionally requires at least SQLAlchemy 1.0. e.g.:: from alembic import op op.create_exclude_constraint( "user_excl", "user", ("period", "&&"), ("group", "="), where=("group != 'some group'"), ) Note that the expressions work the same way as that of the ``ExcludeConstraint`` object itself; if plain strings are passed, quoting rules must be applied manually. :param name: Name of the constraint. :param table_name: String name of the source table. :param elements: exclude conditions. :param where: SQL expression or SQL string with optional WHERE clause. :param deferrable: optional bool. If set, emit DEFERRABLE or NOT DEFERRABLE when issuing DDL for this constraint. :param initially: optional string. If set, emit INITIALLY when issuing DDL for this constraint. :param schema: Optional schema name to operate within. )invoke)rKrWr<rtr?rDrErHrHrSr7s+ z3CreateExcludeConstraintOp.create_exclude_constraintr*cOs.|jj|d<|||jj|fi|}||S)a3Issue a "create exclude constraint" instruction using the current batch migration context. .. note:: This method is Postgresql specific, and additionally requires at least SQLAlchemy 1.0. .. seealso:: :meth:`.Operations.create_exclude_constraint` r|)r.r|rtrY)rKrWr<r?rDrErHrHrSr8s  z9CreateExcludeConstraintOp.batch_create_exclude_constraint)NNN)r<r=rtr>r?r@rArBr|ryrCrDrErF)r[rrEr:rU)rNrOrEr) rWr+r<rdrtrdr?rrDrrErX) rWr*r<rdr?rrDrrErX) rrr__doc__constraint_typer classmethodrMrVr7r8rHrHrHrSr:Ps     -r:rr=rEcCst||ddS)NT)alter)_exclude_constraintrV)rrErHrHrS_add_exclude_constraintsr_r[rnamespace_metadatar9cCs&td||}|dur |St||dS)Nr;F)r'_user_defined_renderr^)r[rr`renderedrHrHrS!_render_inline_exclude_constraints  rccCs|j}|dur |ddS)Nz*from sqlalchemy.dialects import postgresqlz postgresql.)importsadd)rrdrHrHrS_postgresql_autogenerate_prefixs rfr]rcs*gj}jrdtjfjrdtjfjr,dtjf|s@|r@jjr@dt jjf|sQj rQdt j ffdd}|rt t j g}|ss|t t jj g7}| |dtd |d S|}d td |d S) NrGrHrr|rWcsNfddjD}jdur|dtj|ddD|S)Ncs$g|]\}}}dt||fqS)z(%s, %r))_render_potential_column)rsqltextrWopstring)rrHrSrFs zC_exclude_constraint..do_expr_where_opts..zwhere=%scSsg|] \}}d||fqS)z%s=%rrH)rkvrHrHrSrF$s)rJrArr'_render_potential_exprr)argsrr[optsrHrSdo_expr_where_optss   z/_exclude_constraint..do_expr_where_optsz-%(prefix)screate_exclude_constraint(%(args)s)z, )prefixrmz%%(prefix)sExcludeConstraint(%(args)s)) _has_batchrGrrdrHrrKr|r'_identrW_render_gen_namereprr_alembic_autogenerate_prefixjoinrf)r[rr] has_batchrprmrHrnrSr^s<  r^valueGUnion[ColumnClause[Any], Column[Any], TextClause, FunctionElement[Any]]cCsJt|tr|jr d}nd}|t||jdStj||t|ttfdS)Nz"%(prefix)sliteral_column(%(name)r)z%(prefix)scolumn(%(name)r))rqrW) wrap_in_text) rcr is_literalr'_sqlalchemy_autogenerate_prefixrWrlrr)ryrtemplaterHrHrSrg:s  rg)rr"rr2rErd)rrrr2rErd)rrrr2rErd)rr!rr2)rr=rEr:rErd)r[rrr=r`r9rErd)rr=rErd)r[rrr=r]rrErd)ryrzrr=rErd)l __future__rloggingrftypingrrrrrrr r r sqlalchemyr r rrrrNsqlalchemy.dialects.postgresqlrrrsqlalchemy.schemarsqlalchemy.sql.elementsrrsqlalchemy.sql.functionsrsqlalchemy.typesrbaserrrrrrr r!r"r.r#r$r]r& autogenerater'rWr(r)operations.baser*r+r,util.sqla_compatr-r.r/r0$sqlalchemy.dialects.postgresql.arrayr1#sqlalchemy.dialects.postgresql.baser2%sqlalchemy.dialects.postgresql.hstorer3#sqlalchemy.dialects.postgresql.jsonr4r5r6r7r8sqlalchemy.sql.schemar9r:sqlalchemy.sql.type_apir;r<autogenerate.apir=autogenerate.renderr>runtime.migrationr? getLoggerrrr@rr!r"r*r-register_operationAddConstraintOpregister_add_constraintr: renderers dispatch_forr__constraint_renderersrcrfr^rgrHrHrHrSs                                                             ,    "      ?