jasima/majuna
3
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: sqlite fix for migration

Browse source
This commit is contained in:
Iain Learmonth 2024-11-09 14:29:44 +00:00
parent ae1ac4ef54
commit 3be3a45606
1 changed files with 5 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

8
migrations/versions/bbec86de37c4_adds_geo_monitoring.py
View file

@ -3,7 +3,7 @@ from datetime import datetime
from alembic import op from alembic import op
import sqlalchemy as sa import sqlalchemy as sa
from sqlalchemy import table, column from sqlalchemy import column, func, table
from sqlalchemy.orm import Session from sqlalchemy.orm import Session
revision = 'bbec86de37c4' revision = 'bbec86de37c4'
@ -63,8 +63,8 @@ def upgrade():
country_code=country['Code'], country_code=country['Code'],
description=country['Name'], description=country['Name'],
risk_level_override=None, # Assuming risk level override is initially None risk_level_override=None, # Assuming risk level override is initially None
added=datetime.utcnow(), added=func.now(),
updated=datetime.utcnow() updated=func.now()
) )
) )
@ -94,6 +94,8 @@ def upgrade():
# Iterate over each row and create a corresponding entry in the Deprecation table # Iterate over each row and create a corresponding entry in the Deprecation table
for id_, deprecated, reason in results: for id_, deprecated, reason in results:
if deprecated is not None: # Only migrate if there's a deprecation date 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( op.execute(
deprecation_table.insert().values( deprecation_table.insert().values(
resource_type=table_name.title(), # The class name is used, not the table name resource_type=table_name.title(), # The class name is used, not the table name