Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 8 additions & 4 deletions masterbase/lib.py
Original file line number Diff line number Diff line change
Expand Up @@ -682,12 +682,16 @@ def add_report(engine: Engine, session_id: str, target_steam_id: str, reason: st
created_at = datetime.now().astimezone(timezone.utc).isoformat()
txn.execute(
sa.text(
"""INSERT INTO reports (
session_id, target_steam_id, created_at, reason
"""
INSERT INTO reports (
session_id, target_steam_id, created_at, updated_at, reason
) VALUES (
:session_id, :target_steam_id, :created_at, :reason)
:session_id, :target_steam_id, :created_at, :created_at, :reason
)
ON CONFLICT (session_id, target_steam_id)
DO UPDATE SET reason = EXCLUDED.reason, created_at = NOW();
DO UPDATE SET
reason = EXCLUDED.reason,
updated_at = EXCLUDED.created_at;
"""
),
{"target_steam_id": target_steam_id, "session_id": session_id, "created_at": created_at, "reason": reason},
Expand Down
43 changes: 43 additions & 0 deletions migrations/versions/bc89a4c6d2b4_report_update_timings.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,43 @@
"""Report update timings

Revision ID: bc89a4c6d2b4
Revises: 82f4e558463f
Create Date: 2024-06-23 21:30:34.321859

"""

from typing import Sequence, Union

import sqlalchemy as sa
from alembic import op

# revision identifiers, used by Alembic.
revision: str = "bc89a4c6d2b4"
down_revision: Union[str, None] = "82f4e558463f"
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
op.execute(
sa.text("""
ALTER TABLE reports
ADD COLUMN updated_at timestamptz;

UPDATE reports
SET updated_at = created_at
WHERE updated_at IS NULL;
""")
)


def downgrade() -> None:
op.execute(
sa.text(
"""
ALTER TABLE reports
DROP COLUMN updated_at;
"""
)
)
pass