diff options
author | CaselIT <cfederico87@gmail.com> | 2022-09-11 22:38:46 +0200 |
---|---|---|
committer | CaselIT <cfederico87@gmail.com> | 2022-09-12 21:00:52 +0200 |
commit | 0e83fddb6c110bf1658564c248ffad9163a365fa (patch) | |
tree | 9b61207cd621fb526cda1daa55eaf94bd50aeb25 /alembic/autogenerate | |
parent | 747ec301529cf2a08d56d3596aedbf54a93b8742 (diff) | |
download | alembic-0e83fddb6c110bf1658564c248ffad9163a365fa.tar.gz |
Improve typing
Change-Id: I9fc86c4a92e1b76d19c9e891ff08ce8a46ad4e35
Diffstat (limited to 'alembic/autogenerate')
-rw-r--r-- | alembic/autogenerate/api.py | 6 | ||||
-rw-r--r-- | alembic/autogenerate/compare.py | 7 | ||||
-rw-r--r-- | alembic/autogenerate/render.py | 2 |
3 files changed, 8 insertions, 7 deletions
diff --git a/alembic/autogenerate/api.py b/alembic/autogenerate/api.py index 5f1c7f3..cbd64e1 100644 --- a/alembic/autogenerate/api.py +++ b/alembic/autogenerate/api.py @@ -523,12 +523,12 @@ class RevisionContext: def run_autogenerate( self, rev: tuple, migration_context: "MigrationContext" - ): + ) -> None: self._run_environment(rev, migration_context, True) def run_no_autogenerate( self, rev: tuple, migration_context: "MigrationContext" - ): + ) -> None: self._run_environment(rev, migration_context, False) def _run_environment( @@ -536,7 +536,7 @@ class RevisionContext: rev: tuple, migration_context: "MigrationContext", autogenerate: bool, - ): + ) -> None: if autogenerate: if self.command_args["sql"]: raise util.CommandError( diff --git a/alembic/autogenerate/compare.py b/alembic/autogenerate/compare.py index 5b69815..c32ab4d 100644 --- a/alembic/autogenerate/compare.py +++ b/alembic/autogenerate/compare.py @@ -616,8 +616,8 @@ def _compare_indexes_and_uniques( # we know are either added implicitly by the DB or that the DB # can't accurately report on autogen_context.migration_context.impl.correct_for_autogen_constraints( - conn_uniques, - conn_indexes, + conn_uniques, # type: ignore[arg-type] + conn_indexes, # type: ignore[arg-type] metadata_unique_constraints, metadata_indexes, ) @@ -1274,7 +1274,8 @@ def _compare_foreign_keys( ) conn_fks = set( - _make_foreign_key(const, conn_table) for const in conn_fks_list + _make_foreign_key(const, conn_table) # type: ignore[arg-type] + for const in conn_fks_list ) # give the dialect a chance to correct the FKs to match more diff --git a/alembic/autogenerate/render.py b/alembic/autogenerate/render.py index 9c992b4..1ac6753 100644 --- a/alembic/autogenerate/render.py +++ b/alembic/autogenerate/render.py @@ -989,7 +989,7 @@ def _fk_colspec( if table_fullname in namespace_metadata.tables: col = namespace_metadata.tables[table_fullname].c.get(colname) if col is not None: - colname = _ident(col.name) + colname = _ident(col.name) # type: ignore[assignment] colspec = "%s.%s" % (table_fullname, colname) |