[DOP-25282] Use timestamptz fields

Author: dolfinus

syncmaster/db/migrations/versions/2023-11-23_0001_create_user_table.py

@@ -28,8 +28,8 @@ def upgrade():
         sa.Column("middle_name", sa.String(length=256), nullable=True),
         sa.Column("is_superuser", sa.Boolean(), nullable=False),
         sa.Column("is_active", sa.Boolean(), nullable=False),
-        sa.Column("created_at", sa.DateTime(), server_default=sa.text("now()"), nullable=False),
-        sa.Column("updated_at", sa.DateTime(), server_default=sa.text("now()"), nullable=False),
+        sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.text("now()"), nullable=False),
+        sa.Column("updated_at", sa.DateTime(timezone=True), server_default=sa.text("now()"), nullable=False),
         sa.Column("is_deleted", sa.Boolean(), nullable=False),
         sa.PrimaryKeyConstraint("id", name=op.f("pk__user")),
         sa.UniqueConstraint("email", name=op.f("uq__user__email")),

syncmaster/db/migrations/versions/2023-11-23_0002_create_group_table.py

@@ -25,8 +25,8 @@ def upgrade():
         sa.Column("name", sa.String(length=256), nullable=False),
         sa.Column("description", sa.String(length=512), nullable=False),
         sa.Column("owner_id", sa.BigInteger(), nullable=False),
-        sa.Column("created_at", sa.DateTime(), server_default=sa.text("now()"), nullable=False),
-        sa.Column("updated_at", sa.DateTime(), server_default=sa.text("now()"), nullable=False),
+        sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.text("now()"), nullable=False),
+        sa.Column("updated_at", sa.DateTime(timezone=True), server_default=sa.text("now()"), nullable=False),
         sa.Column(
             "search_vector",
             postgresql.TSVECTOR(),

syncmaster/db/migrations/versions/2023-11-23_0003_create_queue_table.py

@@ -26,8 +26,8 @@ def upgrade():
         sa.Column("slug", sa.String(length=256), nullable=False),
         sa.Column("group_id", sa.BigInteger(), nullable=False),
         sa.Column("description", sa.String(length=512), nullable=False),
-        sa.Column("created_at", sa.DateTime(), server_default=sa.text("now()"), nullable=False),
-        sa.Column("updated_at", sa.DateTime(), server_default=sa.text("now()"), nullable=False),
+        sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.text("now()"), nullable=False),
+        sa.Column("updated_at", sa.DateTime(timezone=True), server_default=sa.text("now()"), nullable=False),
         sa.Column(
             "search_vector",
             postgresql.TSVECTOR(),

syncmaster/db/migrations/versions/2023-11-23_0004_create_connection_table.py

@@ -34,8 +34,8 @@ def upgrade():
         sa.Column("name", sa.String(length=128), nullable=False),
         sa.Column("description", sa.String(length=512), nullable=False),
         sa.Column("data", sa.JSON(), nullable=False),
-        sa.Column("created_at", sa.DateTime(), server_default=sa.text("now()"), nullable=False),
-        sa.Column("updated_at", sa.DateTime(), server_default=sa.text("now()"), nullable=False),
+        sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.text("now()"), nullable=False),
+        sa.Column("updated_at", sa.DateTime(timezone=True), server_default=sa.text("now()"), nullable=False),
         sa.Column("search_vector", postgresql.TSVECTOR(), sa.Computed(sql_expression, persisted=True), nullable=False),
         sa.ForeignKeyConstraint(
             ["group_id"],

syncmaster/db/migrations/versions/2023-11-23_0006_create_auth_data_table.py

@@ -22,8 +22,8 @@ def upgrade():
         "auth_data",
         sa.Column("connection_id", sa.BigInteger(), nullable=False),
         sa.Column("value", sa.String(), nullable=False),
-        sa.Column("created_at", sa.DateTime(), server_default=sa.text("now()"), nullable=False),
-        sa.Column("updated_at", sa.DateTime(), server_default=sa.text("now()"), nullable=False),
+        sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.text("now()"), nullable=False),
+        sa.Column("updated_at", sa.DateTime(timezone=True), server_default=sa.text("now()"), nullable=False),
         sa.ForeignKeyConstraint(
             ["connection_id"],
             ["connection.id"],

syncmaster/db/migrations/versions/2023-11-23_0007_create_transfer_table.py

@@ -49,8 +49,8 @@ def upgrade():
         sa.Column("is_scheduled", sa.Boolean(), nullable=False),
         sa.Column("schedule", sa.String(length=32), nullable=False),
         sa.Column("queue_id", sa.BigInteger(), nullable=False),
-        sa.Column("created_at", sa.DateTime(), server_default=sa.text("now()"), nullable=False),
-        sa.Column("updated_at", sa.DateTime(), server_default=sa.text("now()"), nullable=False),
+        sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.text("now()"), nullable=False),
+        sa.Column("updated_at", sa.DateTime(timezone=True), server_default=sa.text("now()"), nullable=False),
         sa.ForeignKeyConstraint(
             ["group_id"],
             ["group.id"],

syncmaster/db/migrations/versions/2023-11-23_0008_create_run_table.py

@@ -22,14 +22,14 @@ def upgrade():
         "run",
         sa.Column("id", sa.BigInteger(), nullable=False),
         sa.Column("transfer_id", sa.BigInteger(), nullable=False),
-        sa.Column("started_at", sa.DateTime(), nullable=True),
-        sa.Column("ended_at", sa.DateTime(), nullable=True),
+        sa.Column("started_at", sa.DateTime(timezone=True), nullable=True),
+        sa.Column("ended_at", sa.DateTime(timezone=True), nullable=True),
         sa.Column("status", sa.String(255), nullable=False),
         sa.Column("type", sa.String(64), nullable=False),
         sa.Column("log_url", sa.String(length=512), nullable=True),
         sa.Column("transfer_dump", sa.JSON(), nullable=False),
-        sa.Column("created_at", sa.DateTime(), server_default=sa.text("now()"), nullable=False),
-        sa.Column("updated_at", sa.DateTime(), server_default=sa.text("now()"), nullable=False),
+        sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.text("now()"), nullable=False),
+        sa.Column("updated_at", sa.DateTime(timezone=True), server_default=sa.text("now()"), nullable=False),
         sa.ForeignKeyConstraint(
             ["transfer_id"],
             ["transfer.id"],

syncmaster/db/migrations/versions/2023-11-23_0009_create_celery_tables.py

@@ -36,7 +36,7 @@ def upgrade():
         sa.Column("task_id", sa.String(length=155), nullable=True),
         sa.Column("status", sa.String(length=50), nullable=True),
         sa.Column("result", sa.PickleType(), nullable=True),
-        sa.Column("date_done", sa.DateTime(), nullable=True),
+        sa.Column("date_done", sa.DateTime(timezone=True), nullable=True),
         sa.Column("traceback", sa.Text(), nullable=True),
         sa.Column("name", sa.String(length=155), nullable=True),
         sa.Column("args", sa.LargeBinary(), nullable=True),
@@ -57,7 +57,7 @@ def upgrade():
         sa.Column("id", sa.Integer(), taskset_id_sequence, autoincrement=True, nullable=False),
         sa.Column("taskset_id", sa.String(length=155), nullable=True),
         sa.Column("result", sa.PickleType(), nullable=True),
-        sa.Column("date_done", sa.DateTime(), nullable=True),
+        sa.Column("date_done", sa.DateTime(timezone=True), nullable=True),
         sa.PrimaryKeyConstraint("id"),
         sa.UniqueConstraint("taskset_id"),
         sqlite_autoincrement=True,

syncmaster/db/mixins/timestamp.py

@@ -1,17 +1,17 @@
 # SPDX-FileCopyrightText: 2023-2024 MTS PJSC
 # SPDX-License-Identifier: Apache-2.0
-from datetime import datetime
+from datetime import datetime, timezone
+from functools import partial
 
 from sqlalchemy import DateTime, func
 from sqlalchemy.orm import Mapped, mapped_column
 
 
 class TimestampMixin:
-    created_at: Mapped[datetime] = mapped_column(DateTime, server_default=func.now(), nullable=False)
+    created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), server_default=func.now(), nullable=False)
     updated_at: Mapped[datetime] = mapped_column(
-        DateTime,
+        DateTime(timezone=True),
         server_default=func.now(),
-        server_onupdate=func.now(),
-        onupdate=datetime.now,
+        onupdate=partial(datetime.now, tz=timezone.utc),
         nullable=False,
     )