diff options
| author | Christian Cleberg <[email protected]> | 2026-04-12 11:43:27 -0500 |
|---|---|---|
| committer | Christian Cleberg <[email protected]> | 2026-04-12 11:43:27 -0500 |
| commit | 4390684995c9933f213faec2cb20f39af6d506d2 (patch) | |
| tree | fac5252756c038e9fdb3444a1a4d68863e1683f3 /alembic/versions/20260412_0007_actor_priority_boost.py | |
| parent | daaeba8be8df4d0876312e540bdd420e16ff724c (diff) | |
| download | hutch-stats-4390684995c9933f213faec2cb20f39af6d506d2.tar.gz hutch-stats-4390684995c9933f213faec2cb20f39af6d506d2.tar.bz2 hutch-stats-4390684995c9933f213faec2cb20f39af6d506d2.zip | |
add temporary self-graph indexing priority
Diffstat (limited to 'alembic/versions/20260412_0007_actor_priority_boost.py')
| -rw-r--r-- | alembic/versions/20260412_0007_actor_priority_boost.py | 29 |
1 files changed, 29 insertions, 0 deletions
diff --git a/alembic/versions/20260412_0007_actor_priority_boost.py b/alembic/versions/20260412_0007_actor_priority_boost.py new file mode 100644 index 0000000..ecceca6 --- /dev/null +++ b/alembic/versions/20260412_0007_actor_priority_boost.py @@ -0,0 +1,29 @@ +"""temporary actor priority boost marker""" + +from __future__ import annotations + +from alembic import op +import sqlalchemy as sa +from sqlalchemy import inspect + + +revision = "20260412_0007" +down_revision = "20260411_0006" +branch_labels = None +depends_on = None + + +def _column_names(table_name: str) -> set[str]: + return {column["name"] for column in inspect(op.get_bind()).get_columns(table_name)} + + +def upgrade() -> None: + columns = _column_names("tracked_actors") + if "priority_boosted_at" not in columns: + op.add_column("tracked_actors", sa.Column("priority_boosted_at", sa.DateTime(timezone=True), nullable=True)) + + +def downgrade() -> None: + columns = _column_names("tracked_actors") + if "priority_boosted_at" in columns: + op.drop_column("tracked_actors", "priority_boosted_at") |
