p08-seo-tweaks

backend/database.py

@@ -1,6 +1,6 @@
 from collections.abc import Generator
 
-from sqlalchemy import create_engine
+from sqlalchemy import create_engine, text
 from sqlalchemy.orm import DeclarativeBase, Session, sessionmaker
 
 DATABASE_URL = "sqlite:///./data/clawfort.db"
@@ -25,3 +25,27 @@ def init_db() -> None:
     from backend.models import NewsItem, NewsTranslation  # noqa: F401
 
     Base.metadata.create_all(bind=engine)
+
+    migration_sql = {
+        "news_items": {
+            "tldr_points": "ALTER TABLE news_items ADD COLUMN tldr_points TEXT",
+            "summary_body": "ALTER TABLE news_items ADD COLUMN summary_body TEXT",
+            "source_citation": "ALTER TABLE news_items ADD COLUMN source_citation TEXT",
+            "summary_image_url": "ALTER TABLE news_items ADD COLUMN summary_image_url VARCHAR(2000)",
+            "summary_image_credit": "ALTER TABLE news_items ADD COLUMN summary_image_credit VARCHAR(500)",
+        },
+        "news_translations": {
+            "tldr_points": "ALTER TABLE news_translations ADD COLUMN tldr_points TEXT",
+            "summary_body": "ALTER TABLE news_translations ADD COLUMN summary_body TEXT",
+            "source_citation": "ALTER TABLE news_translations ADD COLUMN source_citation TEXT",
+        },
+    }
+
+    with engine.begin() as conn:
+        for table, columns in migration_sql.items():
+            existing_cols = {
+                row[1] for row in conn.execute(text(f"PRAGMA table_info({table})")).fetchall()
+            }
+            for col_name, ddl in columns.items():
+                if col_name not in existing_cols:
+                    conn.execute(text(ddl))