Rename database/db.txt to database/db.py

database/db.py

@@ -0,0 +1,163 @@
+from __future__ import annotations
+
+from pathlib import Path
+from typing import Iterable, Mapping, Any
+
+import duckdb
+import pandas as pd
+
+from config.settings import DUCKDB_PATH
+
+
+def get_connection() -> duckdb.DuckDBPyConnection:
+    Path("data").mkdir(parents=True, exist_ok=True)
+    conn = duckdb.connect(DUCKDB_PATH)
+    initialize_schema(conn)
+    return conn
+
+
+def initialize_schema(conn: duckdb.DuckDBPyConnection) -> None:
+    conn.execute(
+        """
+        CREATE TABLE IF NOT EXISTS bets (
+            bet_id BIGINT,
+            created_at TIMESTAMP,
+            sportsbook TEXT,
+            market TEXT,
+            selection TEXT,
+            odds INTEGER,
+            stake DOUBLE,
+            result TEXT,
+            profit DOUBLE,
+            game_id TEXT,
+            notes TEXT
+        );
+        """
+    )
+
+    conn.execute(
+        """
+        CREATE TABLE IF NOT EXISTS cached_schedule (
+            fetched_at TIMESTAMP,
+            game_id TEXT,
+            game_date TIMESTAMP,
+            status TEXT,
+            away_team TEXT,
+            home_team TEXT,
+            away_score INTEGER,
+            home_score INTEGER,
+            venue TEXT
+        );
+        """
+    )
+
+    conn.execute(
+        """
+        CREATE TABLE IF NOT EXISTS cached_odds (
+            fetched_at TIMESTAMP,
+            event_id TEXT,
+            commence_time TIMESTAMP,
+            home_team TEXT,
+            away_team TEXT,
+            sportsbook TEXT,
+            market_key TEXT,
+            outcome_name TEXT,
+            price INTEGER,
+            point DOUBLE
+        );
+        """
+    )
+
+    conn.execute(
+        """
+        CREATE TABLE IF NOT EXISTS cached_weather (
+            fetched_at TIMESTAMP,
+            venue_key TEXT,
+            location_name TEXT,
+            temperature_f DOUBLE,
+            humidity INTEGER,
+            wind_speed_mph DOUBLE,
+            wind_deg INTEGER,
+            description TEXT
+        );
+        """
+    )
+
+
+def upsert_dataframe(
+    conn: duckdb.DuckDBPyConnection,
+    table_name: str,
+    df: pd.DataFrame,
+    replace: bool = True,
+) -> None:
+    if df.empty:
+        return
+
+    temp_name = f"tmp_{table_name}"
+    conn.register(temp_name, df)
+    if replace:
+        conn.execute(f"DELETE FROM {table_name}")
+    conn.execute(f"INSERT INTO {table_name} SELECT * FROM {temp_name}")
+    conn.unregister(temp_name)
+
+
+def read_table(conn: duckdb.DuckDBPyConnection, table_name: str) -> pd.DataFrame:
+    return conn.execute(f"SELECT * FROM {table_name}").df()
+
+
+def insert_bet(
+    conn: duckdb.DuckDBPyConnection,
+    bet_id: int,
+    created_at: str,
+    sportsbook: str,
+    market: str,
+    selection: str,
+    odds: int,
+    stake: float,
+    result: str,
+    profit: float,
+    game_id: str,
+    notes: str,
+) -> None:
+    conn.execute(
+        """
+        INSERT INTO bets (
+            bet_id, created_at, sportsbook, market, selection, odds, stake,
+            result, profit, game_id, notes
+        ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+        """,
+        [
+            bet_id,
+            created_at,
+            sportsbook,
+            market,
+            selection,
+            odds,
+            stake,
+            result,
+            profit,
+            game_id,
+            notes,
+        ],
+    )
+
+
+def next_bet_id(conn: duckdb.DuckDBPyConnection) -> int:
+    value = conn.execute("SELECT COALESCE(MAX(bet_id), 0) + 1 FROM bets").fetchone()[0]
+    return int(value)
+
+
+def update_bet_result(
+    conn: duckdb.DuckDBPyConnection,
+    bet_id: int,
+    result: str,
+    profit: float,
+) -> None:
+    conn.execute(
+        """
+        UPDATE bets
+        SET result = ?, profit = ?
+        WHERE bet_id = ?
+        """,
+        [result, profit, bet_id],
+    )