fix: sqlite fix for migration

This commit is contained in:
Iain Learmonth 2024-11-09 14:29:44 +00:00
parent ae1ac4ef54
commit 3be3a45606

View file

@ -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