main commit

venv/lib/python3.12/site-packages/alembic/operations/toimpl.py

@@ -1,6 +1,3 @@
-# mypy: allow-untyped-defs, allow-incomplete-defs, allow-untyped-calls
-# mypy: no-warn-return-any, allow-any-generics
-
 from typing import TYPE_CHECKING
 
 from sqlalchemy import schema as sa_schema
@@ -79,11 +76,8 @@ def alter_column(
 
 @Operations.implementation_for(ops.DropTableOp)
 def drop_table(operations: "Operations", operation: "ops.DropTableOp") -> None:
-    kw = {}
-    if operation.if_exists is not None:
-        kw["if_exists"] = operation.if_exists
     operations.impl.drop_table(
-        operation.to_table(operations.migration_context), **kw
+        operation.to_table(operations.migration_context)
     )
 
 
@@ -93,11 +87,7 @@ def drop_column(
 ) -> None:
     column = operation.to_column(operations.migration_context)
     operations.impl.drop_column(
-        operation.table_name,
-        column,
-        schema=operation.schema,
-        if_exists=operation.if_exists,
-        **operation.kw,
+        operation.table_name, column, schema=operation.schema, **operation.kw
     )
 
 
@@ -108,6 +98,9 @@ def create_index(
     idx = operation.to_index(operations.migration_context)
     kw = {}
     if operation.if_not_exists is not None:
+        if not sqla_2:
+            raise NotImplementedError("SQLAlchemy 2.0+ required")
+
         kw["if_not_exists"] = operation.if_not_exists
     operations.impl.create_index(idx, **kw)
 
@@ -116,6 +109,9 @@ def create_index(
 def drop_index(operations: "Operations", operation: "ops.DropIndexOp") -> None:
     kw = {}
     if operation.if_exists is not None:
+        if not sqla_2:
+            raise NotImplementedError("SQLAlchemy 2.0+ required")
+
         kw["if_exists"] = operation.if_exists
 
     operations.impl.drop_index(
@@ -128,11 +124,8 @@ def drop_index(operations: "Operations", operation: "ops.DropIndexOp") -> None:
 def create_table(
     operations: "Operations", operation: "ops.CreateTableOp"
 ) -> "Table":
-    kw = {}
-    if operation.if_not_exists is not None:
-        kw["if_not_exists"] = operation.if_not_exists
     table = operation.to_table(operations.migration_context)
-    operations.impl.create_table(table, **kw)
+    operations.impl.create_table(table)
     return table
 
 
@@ -172,13 +165,7 @@ def add_column(operations: "Operations", operation: "ops.AddColumnOp") -> None:
         column = _copy(column)
 
     t = operations.schema_obj.table(table_name, column, schema=schema)
-    operations.impl.add_column(
-        table_name,
-        column,
-        schema=schema,
-        if_not_exists=operation.if_not_exists,
-        **kw,
-    )
+    operations.impl.add_column(table_name, column, schema=schema, **kw)
 
     for constraint in t.constraints:
         if not isinstance(constraint, sa_schema.PrimaryKeyConstraint):
@@ -208,19 +195,13 @@ def create_constraint(
 def drop_constraint(
     operations: "Operations", operation: "ops.DropConstraintOp"
 ) -> None:
-    kw = {}
-    if operation.if_exists is not None:
-        if not sqla_2:
-            raise NotImplementedError("SQLAlchemy 2.0 required")
-        kw["if_exists"] = operation.if_exists
     operations.impl.drop_constraint(
         operations.schema_obj.generic_constraint(
             operation.constraint_name,
             operation.table_name,
             operation.constraint_type,
             schema=operation.schema,
-        ),
-        **kw,
+        )
     )