Integrated Turso database

.devcontainer/devcontainer.json

@@ -9,6 +9,11 @@
     // Setting it to 'false' ensures your Dockerfile's CMD command is respected.
     "overrideCommand": false,
 
+    "runArgs": [
+        "--env-file",
+        ".env"
+    ],
+
     // Any ports to forward from the container to your local machine
     "forwardPorts": [7860],
 

.gitignore

@@ -1,4 +1,5 @@
 */__pycache__
 venv
 .venv
-*.db
+*.db
+.env

app/database.py

@@ -2,6 +2,7 @@ from sqlalchemy import create_engine, UnicodeText, DateTime, ForeignKey
 from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column, Session, joinedload, relationship
 from sqlalchemy.sql import func
 from contextlib import contextmanager
+import os
 import datetime
 
 
@@ -34,8 +35,22 @@ class Source(Base):
     article_id: Mapped[int] = mapped_column(ForeignKey('article.id'))
 
 
+# get environment variables for database
+TURSO_DATABASE_URL = os.getenv('TURSO_DATABASE_URL')
+TURSO_AUTH_TOKEN = os.getenv('TURSO_AUTH_TOKEN')
+
 # create an engine
-engine = create_engine('sqlite:///news.db', echo=True)
+if not TURSO_DATABASE_URL or not TURSO_AUTH_TOKEN:
+    print('Using local SQLite database')
+    engine = create_engine('sqlite:///news.db', echo=True)
+else:
+    engine = create_engine(
+        f'sqlite+{TURSO_DATABASE_URL}?secure=true',
+        connect_args={
+            'auth_token': TURSO_AUTH_TOKEN
+        },
+        echo=True,
+    )
 
 # create tables if needed
 Base.metadata.create_all(engine)

app/main.py

@@ -5,6 +5,22 @@ from apscheduler.schedulers.background import BackgroundScheduler
 from .database import get_session, retrieve_articles
 from .update_news import update_news
 import threading
+import os
+
+
+UPDATE_INTERVAL_VAR = 'UPDATE_INTERVAL'
+
+DEFAULT_UPDATE_INTERVAL = 180  # in minutes
+
+# set update interval from environment variable if available
+try:
+    update_interval_str = os.getenv(UPDATE_INTERVAL_VAR)
+    if update_interval_str is not None:
+        update_interval = int(update_interval_str)
+    else:
+        update_interval = DEFAULT_UPDATE_INTERVAL
+except (ValueError, TypeError):
+    update_interval = DEFAULT_UPDATE_INTERVAL
 
 
 # for updating the news feed periodically
@@ -17,7 +33,7 @@ is_updating = threading.Lock()
 @asynccontextmanager
 async def lifespan(app: FastAPI):
     # update news periodically
-    scheduler.add_job(update_news, 'interval', minutes=30)
+    scheduler.add_job(update_news, 'interval', minutes=update_interval)
     scheduler.start()
     yield
     # stop the scheduler when closing the server