From 22714994d2b7cfa238c8b2d54a3639cd6417e9b6 Mon Sep 17 00:00:00 2001 From: Pinapelz Date: Fri, 21 Nov 2025 22:30:47 -0800 Subject: scaffold implementation for remote sqlite db --- generate.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) (limited to 'generate.py') diff --git a/generate.py b/generate.py index 4fd5735..2531f1c 100644 --- a/generate.py +++ b/generate.py @@ -10,7 +10,7 @@ import hashlib import os from dotenv import load_dotenv from datetime import datetime, timedelta -from database import Database +from common import create_database_connection from feed import build_rss_from_news_feed from notifications import check_can_send_notifs, broadcast_to_topic @@ -27,10 +27,9 @@ def compute_json_hash(data): json.dumps(data, sort_keys=True).encode("utf-8") ).hexdigest() - def save_news_to_db(news_feed: list): log_output("Writing news to local save database. This is purely for archival reasons") - database = Database() + database = create_database_connection() for entry in news_feed: key = compute_json_hash(entry) database.add_news_entry(key, entry) @@ -68,7 +67,7 @@ def attempt_broadcast_notifications(news_data: list, title: str, topic: str, ima if not check_can_send_notifs: print("[WARNING] Skipping notifications as env vars are not properly configured. See template") else: - database = Database() + database = create_database_connection() cutoff = datetime.now() - timedelta(days=limit) for entry in news_data: if datetime.fromtimestamp(entry["timestamp"]) < cutoff: -- cgit v1.2.3