mirror of
https://github.com/shawnkim1997/All-in-one-Financial-Analysis.git
synced 2026-08-14 10:58:05 +00:00
shawnkim1997andClaude Opus 4.6
b2acda81ee
Complete migration from Streamlit to Next.js 14 App Router + FastAPI backend. Frontend (Next.js 14): - 10 pages: Overview, Research, Valuation, Technical, Markets, Earnings, News, Portfolio, Filings, Settings - Terminal Noir dark theme with custom Tailwind config - TradingView Lightweight Charts for candlestick/volume - Valuation: DCF, Sensitivity Matrix, Monte Carlo, Tornado, Reverse DCF - Financial Statements table with YoY growth badges and margin rows - SEC EDGAR inline filing viewer with section tabs - News split-view with iframe article embedding - Technical Analysis with RSI, MACD, Bollinger, Fibonacci, Moving Averages - Earnings beat/miss visualization - AI Copilot chat panel with Gemini integration Backend (FastAPI): - 13 routers: market_data, financials, valuation, technical, earnings, insider, edgar, news, portfolio, analysis, chat, estimates, fx - Services: DCF engine, Monte Carlo simulation, sensitivity analysis, risk metrics, SEC parser, technical indicators - yfinance + yahooquery data sources with fallback pattern - SQLite caching layer Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
116 lines
3.4 KiB
Python
116 lines
3.4 KiB
Python
"""
|
|
PostgreSQL async connection manager for ATLAS Terminal.
|
|
Uses asyncpg for high-performance async PostgreSQL operations.
|
|
Configurable via DATABASE_URL environment variable.
|
|
"""
|
|
import os
|
|
import json
|
|
import logging
|
|
from typing import Optional, Any
|
|
from datetime import datetime, timezone
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
# Try to import asyncpg
|
|
try:
|
|
import asyncpg
|
|
HAS_ASYNCPG = True
|
|
except ImportError:
|
|
HAS_ASYNCPG = False
|
|
asyncpg = None
|
|
|
|
_pool: Optional[Any] = None
|
|
|
|
|
|
async def get_pg_pool() -> Optional[Any]:
|
|
"""Get or create the PostgreSQL connection pool."""
|
|
global _pool
|
|
if not HAS_ASYNCPG:
|
|
logger.warning("asyncpg not installed. Run: pip install asyncpg")
|
|
return None
|
|
if _pool is not None:
|
|
return _pool
|
|
|
|
database_url = os.getenv("DATABASE_URL", "")
|
|
if not database_url:
|
|
logger.info("No DATABASE_URL set, PostgreSQL disabled.")
|
|
return None
|
|
|
|
try:
|
|
_pool = await asyncpg.create_pool(
|
|
database_url,
|
|
min_size=2,
|
|
max_size=10,
|
|
command_timeout=30,
|
|
)
|
|
logger.info("PostgreSQL connection pool created.")
|
|
return _pool
|
|
except Exception as e:
|
|
logger.error(f"Failed to create PostgreSQL pool: {e}")
|
|
return None
|
|
|
|
|
|
async def init_pg_tables() -> None:
|
|
"""Create tables if they don't exist in PostgreSQL."""
|
|
pool = await get_pg_pool()
|
|
if not pool:
|
|
return
|
|
|
|
async with pool.acquire() as conn:
|
|
await conn.execute("""
|
|
CREATE TABLE IF NOT EXISTS portfolio_positions (
|
|
id SERIAL PRIMARY KEY,
|
|
ticker VARCHAR(20) NOT NULL,
|
|
name VARCHAR(200),
|
|
shares DOUBLE PRECISION NOT NULL DEFAULT 0,
|
|
avg_cost DOUBLE PRECISION NOT NULL DEFAULT 0,
|
|
currency VARCHAR(10) DEFAULT 'USD',
|
|
broker VARCHAR(100),
|
|
created_at TIMESTAMPTZ DEFAULT NOW(),
|
|
updated_at TIMESTAMPTZ DEFAULT NOW()
|
|
);
|
|
""")
|
|
await conn.execute("""
|
|
CREATE TABLE IF NOT EXISTS watchlist (
|
|
id SERIAL PRIMARY KEY,
|
|
ticker VARCHAR(20) NOT NULL UNIQUE,
|
|
added_at TIMESTAMPTZ DEFAULT NOW()
|
|
);
|
|
""")
|
|
await conn.execute("""
|
|
CREATE TABLE IF NOT EXISTS dashboards (
|
|
id SERIAL PRIMARY KEY,
|
|
name VARCHAR(200) NOT NULL,
|
|
layout_json JSONB DEFAULT '{}',
|
|
created_at TIMESTAMPTZ DEFAULT NOW(),
|
|
updated_at TIMESTAMPTZ DEFAULT NOW()
|
|
);
|
|
""")
|
|
await conn.execute("""
|
|
CREATE TABLE IF NOT EXISTS settings (
|
|
key VARCHAR(100) PRIMARY KEY,
|
|
value TEXT
|
|
);
|
|
""")
|
|
await conn.execute("""
|
|
CREATE TABLE IF NOT EXISTS cache (
|
|
key VARCHAR(500) PRIMARY KEY,
|
|
value JSONB,
|
|
expires_at TIMESTAMPTZ
|
|
);
|
|
""")
|
|
# Index for cache expiry cleanup
|
|
await conn.execute("""
|
|
CREATE INDEX IF NOT EXISTS idx_cache_expires ON cache(expires_at);
|
|
""")
|
|
logger.info("PostgreSQL tables initialized.")
|
|
|
|
|
|
async def close_pg_pool() -> None:
|
|
"""Close the PostgreSQL connection pool."""
|
|
global _pool
|
|
if _pool:
|
|
await _pool.close()
|
|
_pool = None
|
|
logger.info("PostgreSQL pool closed.")
|