use TIMESTAMP from db_types

apache · Nov 25, 2024 · 5d9395d · 5d9395d
1 parent 7496cd4
commit 5d9395d
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/airflow/migrations/versions/0032_3_0_0_drop_execution_date_unique.py b/airflow/migrations/versions/0032_3_0_0_drop_execution_date_unique.py
@@ -33,7 +33,7 @@
 
 from alembic import op
 
-from airflow.utils.sqlalchemy import UtcDateTime
+from airflow.migrations.db_types import TIMESTAMP
 
 # revision identifiers, used by Alembic.
 revision = "1cdc775ca98f"
@@ -48,7 +48,7 @@ def upgrade():
         batch_op.alter_column(
             "execution_date",
             new_column_name="logical_date",
-            existing_type=UtcDateTime,
+            existing_type=TIMESTAMP(timezone=True),
             existing_nullable=False,
         )
     with op.batch_alter_table("dag_run", schema=None) as batch_op:
@@ -60,7 +60,7 @@ def downgrade():
         batch_op.alter_column(
             "logical_date",
             new_column_name="execution_date",
-            existing_type=UtcDateTime,
+            existing_type=TIMESTAMP(timezone=True),
             existing_nullable=False,
         )
     with op.batch_alter_table("dag_run", schema=None) as batch_op:

diff --git a/docs/apache-airflow/img/airflow_erd.sha256 b/docs/apache-airflow/img/airflow_erd.sha256
@@ -1 +1 @@
-65f26e091cabd5c67f45abb682d1ef4fe63cd0d701b5729e5cf95d9cf2599c6a
+9f404e900b4d166d170d878fc00cad8e4397a4288167eebef1d77281a881e92d
diff --git a/docs/apache-airflow/img/airflow_erd.svg b/docs/apache-airflow/img/airflow_erd.svg
Original file line number	Diff line number	Diff line change
		@@ -1 +1 @@
		65f26e091cabd5c67f45abb682d1ef4fe63cd0d701b5729e5cf95d9cf2599c6a
		9f404e900b4d166d170d878fc00cad8e4397a4288167eebef1d77281a881e92d