30 lines
1.0 KiB
Python
30 lines
1.0 KiB
Python
|
from datetime import timezone
|
||
|
|
from zoneinfo import ZoneInfo
|
||
|
|
from app import create_app, db
|
||
|
|
from src.models import Article, Issue
|
||
|
|
|
||
|
|
def migrate():
|
||
|
|
app = create_app(start_scheduler=False)
|
||
|
|
local_tz = ZoneInfo("America/New_York")
|
||
|
|
|
||
|
|
with app.app_context():
|
||
|
|
# Migrate Articles
|
||
|
|
for article in Article.query.all():
|
||
|
|
# Assume existing naive datetime is UTC
|
||
|
|
utc_dt = article.pub_date.replace(tzinfo=timezone.utc)
|
||
|
|
article.pub_date = utc_dt.astimezone(local_tz).replace(tzinfo=None)
|
||
|
|
|
||
|
|
utc_fetched = article.fetched_at.replace(tzinfo=timezone.utc)
|
||
|
|
article.fetched_at = utc_fetched.astimezone(local_tz).replace(tzinfo=None)
|
||
|
|
|
||
|
|
# Migrate Issues
|
||
|
|
for issue in Issue.query.all():
|
||
|
|
utc_created = issue.created_at.replace(tzinfo=timezone.utc)
|
||
|
|
issue.created_at = utc_created.astimezone(local_tz).replace(tzinfo=None)
|
||
|
|
|
||
|
|
db.session.commit()
|
||
|
|
print("Migration complete.")
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
migrate()
|