diff --git a/migrations/versions/bbec86de37c4_adds_geo_monitoring.py b/migrations/versions/bbec86de37c4_adds_geo_monitoring.py index 2dc031e..f16b5ba 100644 --- a/migrations/versions/bbec86de37c4_adds_geo_monitoring.py +++ b/migrations/versions/bbec86de37c4_adds_geo_monitoring.py @@ -3,7 +3,7 @@ from datetime import datetime from alembic import op import sqlalchemy as sa -from sqlalchemy import table, column +from sqlalchemy import column, func, table from sqlalchemy.orm import Session revision = 'bbec86de37c4' @@ -63,8 +63,8 @@ def upgrade(): country_code=country['Code'], description=country['Name'], risk_level_override=None, # Assuming risk level override is initially None - added=datetime.utcnow(), - updated=datetime.utcnow() + added=func.now(), + updated=func.now() ) ) @@ -94,6 +94,8 @@ def upgrade(): # Iterate over each row and create a corresponding entry in the Deprecation table for id_, deprecated, reason in results: if deprecated is not None: # Only migrate if there's a deprecation date + if isinstance(deprecated, str): + deprecated = datetime.strptime(deprecated, "%Y-%m-%d %H:%M:%S.%f") op.execute( deprecation_table.insert().values( resource_type=table_name.title(), # The class name is used, not the table name