diff --git a/alembic/versions/2025_08_11_0931-9a56916ea7d8_remove_agencies_ds_last_updated_at.py b/alembic/versions/2025_08_11_0931-9a56916ea7d8_remove_agencies_ds_last_updated_at.py new file mode 100644 index 00000000..a14cf32b --- /dev/null +++ b/alembic/versions/2025_08_11_0931-9a56916ea7d8_remove_agencies_ds_last_updated_at.py @@ -0,0 +1,31 @@ +"""Remove agencies.ds_last_updated_at + +Revision ID: 9a56916ea7d8 +Revises: c14d669d7c0d +Create Date: 2025-08-11 09:31:18.268319 + +""" +from typing import Sequence, Union + +from alembic import op +import sqlalchemy as sa + + +# revision identifiers, used by Alembic. +revision: str = '9a56916ea7d8' +down_revision: Union[str, None] = 'c14d669d7c0d' +branch_labels: Union[str, Sequence[str], None] = None +depends_on: Union[str, Sequence[str], None] = None + +COLUMN_NAME = "ds_last_updated_at" +TABLE_NAME = "agencies" + +def upgrade() -> None: + op.drop_column(TABLE_NAME, COLUMN_NAME) + + +def downgrade() -> None: + op.add_column( + table_name=TABLE_NAME, + column=sa.Column(COLUMN_NAME, sa.DateTime(), nullable=False), + ) diff --git a/src/db/models/instantiations/agency/pydantic/upsert.py b/src/db/models/instantiations/agency/pydantic/upsert.py index 1deeb6b5..c9d81336 100644 --- a/src/db/models/instantiations/agency/pydantic/upsert.py +++ b/src/db/models/instantiations/agency/pydantic/upsert.py @@ -20,4 +20,3 @@ def sa_model(cls) -> type[Base]: state: str | None county: str | None locality: str | None - ds_last_updated_at: datetime diff --git a/src/db/models/instantiations/agency/sqlalchemy.py b/src/db/models/instantiations/agency/sqlalchemy.py index 8310eeac..556bde88 100644 --- a/src/db/models/instantiations/agency/sqlalchemy.py +++ b/src/db/models/instantiations/agency/sqlalchemy.py @@ -23,11 +23,6 @@ class Agency( state = Column(String, nullable=True) county = Column(String, nullable=True) locality = Column(String, nullable=True) - ds_last_updated_at = Column( - DateTime, - nullable=True, - comment="The last time the agency was updated in the data sources database." - ) # Relationships automated_suggestions = relationship("AutomatedUrlAgencySuggestion", back_populates="agency") diff --git a/tests/automated/integration/tasks/scheduled/impl/sync/agency/existence_checker.py b/tests/automated/integration/tasks/scheduled/impl/sync/agency/existence_checker.py index e99f6112..44da9b6f 100644 --- a/tests/automated/integration/tasks/scheduled/impl/sync/agency/existence_checker.py +++ b/tests/automated/integration/tasks/scheduled/impl/sync/agency/existence_checker.py @@ -25,4 +25,3 @@ def check( assert info.state_name == agency.state assert info.county_name == agency.county assert info.locality_name == agency.locality - assert info.updated_at == agency.ds_last_updated_at \ No newline at end of file diff --git a/tests/automated/integration/tasks/scheduled/impl/sync/agency/test_no_new_results.py b/tests/automated/integration/tasks/scheduled/impl/sync/agency/test_no_new_results.py index 8c7b9abd..9fdd88bb 100644 --- a/tests/automated/integration/tasks/scheduled/impl/sync/agency/test_no_new_results.py +++ b/tests/automated/integration/tasks/scheduled/impl/sync/agency/test_no_new_results.py @@ -50,5 +50,4 @@ async def test_agency_sync_task_no_new_results( # Neither should be updated with new values checker = AgencyChecker() for agency in agencies: - with pytest.raises(AssertionError): - checker.check(agency) \ No newline at end of file + checker.check(agency) \ No newline at end of file