feat: add overseas financial news and indices support

stock-lab/app/db.py

@@ -17,6 +17,7 @@ def init_db():
             CREATE TABLE IF NOT EXISTS articles (
                 id INTEGER PRIMARY KEY AUTOINCREMENT,
                 hash TEXT UNIQUE NOT NULL,
+                category TEXT DEFAULT 'domestic',
                 title TEXT NOT NULL,
                 link TEXT,
                 summary TEXT,
@@ -26,6 +27,11 @@ def init_db():
             )
         """)
         conn.execute("CREATE INDEX IF NOT EXISTS idx_articles_crawled ON articles(crawled_at DESC)")
+        
+        # 컬럼 추가 (기존 테이블 마이그레이션)
+        cols = {r["name"] for r in conn.execute("PRAGMA table_info(articles)").fetchall()}
+        if "category" not in cols:
+            conn.execute("ALTER TABLE articles ADD COLUMN category TEXT DEFAULT 'domestic'")
 
 def save_articles(articles: List[Dict[str, str]]) -> int:
     count = 0
@@ -36,19 +42,26 @@ def save_articles(articles: List[Dict[str, str]]) -> int:
             h = hashlib.md5(unique_str.encode()).hexdigest()
             
             try:
+                cat = a.get("category", "domestic")
                 conn.execute("""
-                    INSERT INTO articles (hash, title, link, summary, press, pub_date, crawled_at)
-                    VALUES (?, ?, ?, ?, ?, ?, ?)
-                """, (h, a['title'], a['link'], a['summary'], a['press'], a['date'], a['crawled_at']))
+                    INSERT INTO articles (hash, category, title, link, summary, press, pub_date, crawled_at)
+                    VALUES (?, ?, ?, ?, ?, ?, ?, ?)
+                """, (h, cat, a['title'], a['link'], a['summary'], a['press'], a['date'], a['crawled_at']))
                 count += 1
             except sqlite3.IntegrityError:
                 pass # 이미 존재함
     return count
 
-def get_latest_articles(limit: int = 20) -> List[Dict[str, Any]]:
+def get_latest_articles(limit: int = 20, category: str = None) -> List[Dict[str, Any]]:
     with _conn() as conn:
-        rows = conn.execute(
-            "SELECT * FROM articles ORDER BY crawled_at DESC, id DESC LIMIT ?", 
-            (limit,)
-        ).fetchall()
+        if category:
+            rows = conn.execute(
+                "SELECT * FROM articles WHERE category = ? ORDER BY crawled_at DESC, id DESC LIMIT ?", 
+                (category, limit)
+            ).fetchall()
+        else:
+            rows = conn.execute(
+                "SELECT * FROM articles ORDER BY crawled_at DESC, id DESC LIMIT ?", 
+                (limit,)
+            ).fetchall()
         return [dict(r) for r in rows]