import os import json from typing import Optional from fastapi import FastAPI from fastapi.responses import JSONResponse from fastapi.middleware.cors import CORSMiddleware import requests from apscheduler.schedulers.background import BackgroundScheduler from pydantic import BaseModel from .db import ( init_db, save_articles, get_latest_articles, add_portfolio_item, get_all_portfolio, get_portfolio_item, update_portfolio_item, delete_portfolio_item, upsert_broker_cash, get_all_broker_cash, get_broker_cash, delete_broker_cash, ) from .scraper import fetch_market_news, fetch_major_indices, fetch_overseas_news from .price_fetcher import get_current_prices app = FastAPI() # CORS 설정 (프론트엔드 접근 허용) app.add_middleware( CORSMiddleware, allow_origins=["*"], # 운영 시에는 구체적인 도메인으로 제한하는 것이 좋음 allow_credentials=True, allow_methods=["*"], allow_headers=["*"], ) scheduler = BackgroundScheduler(timezone=os.getenv("TZ", "Asia/Seoul")) # Windows AI Server URL (NAS .env에서 설정) WINDOWS_AI_SERVER_URL = os.getenv("WINDOWS_AI_SERVER_URL", "http://192.168.0.5:8000") @app.on_event("startup") def on_startup(): init_db() # 매일 아침 8시 뉴스 스크랩 (NAS 자체 수행) scheduler.add_job(run_scraping_job, "cron", hour="8", minute="0") # 앱 시작 시에도 한 번 실행 (데이터 없으면) if not get_latest_articles(1): run_scraping_job() scheduler.start() def run_scraping_job(): print("[StockLab] Starting news scraping...") # 1. 국내 articles_kr = fetch_market_news() count_kr = save_articles(articles_kr) # 2. 해외 (임시 차단) # articles_world = fetch_overseas_news() # count_world = save_articles(articles_world) count_world = 0 print(f"[StockLab] Saved {count_kr} domestic, {count_world} overseas articles.") @app.get("/health") def health(): return {"ok": True} @app.get("/api/stock/news") def get_news(limit: int = 20, category: str = None): """최신 주식 뉴스 조회 (category: 'domestic' | 'overseas')""" return get_latest_articles(limit, category) @app.get("/api/stock/indices") def get_indices(): """주요 지표(KOSPI 등) 실시간 크롤링 조회""" return fetch_major_indices() @app.post("/api/stock/scrap") def trigger_scrap(): """수동 스크랩 트리거""" run_scraping_job() return {"ok": True} # --- Trading API (Windows Proxy) --- @app.get("/api/trade/balance") def get_balance(): """계좌 잔고 조회 (Windows AI Server Proxy)""" print(f"[Proxy] Requesting Balance from {WINDOWS_AI_SERVER_URL}...") try: resp = requests.get(f"{WINDOWS_AI_SERVER_URL}/trade/balance", timeout=5) if resp.status_code != 200: print(f"[ProxyError] Balance Error: {resp.status_code} {resp.text}") return JSONResponse(status_code=resp.status_code, content=resp.json()) print("[Proxy] Balance Success") return resp.json() except Exception as e: print(f"[ProxyError] Connection Failed: {e}") return JSONResponse( status_code=500, content={"error": "Connection Failed", "detail": str(e), "target": WINDOWS_AI_SERVER_URL} ) class OrderRequest(BaseModel): ticker: str # 종목 코드 (예: "005930") action: str # "BUY" or "SELL" quantity: int # 주문 수량 price: int = 0 # 0이면 시장가 reason: Optional[str] = "Manual Order" # 주문 사유 (AI 기록용) @app.post("/api/trade/order") def order_stock(req: OrderRequest): """주식 매수/매도 주문 (Windows AI Server Proxy)""" print(f"[Proxy] Order Request: {req.dict()} to {WINDOWS_AI_SERVER_URL}...") try: resp = requests.post(f"{WINDOWS_AI_SERVER_URL}/trade/order", json=req.dict(), timeout=10) if resp.status_code != 200: print(f"[ProxyError] Order Error: {resp.status_code} {resp.text}") return JSONResponse(status_code=resp.status_code, content=resp.json()) return resp.json() except Exception as e: print(f"[ProxyError] Order Connection Failed: {e}") return JSONResponse( status_code=500, content={"error": "Connection Failed", "detail": str(e), "target": WINDOWS_AI_SERVER_URL} ) @app.get("/api/version") def version(): return {"version": os.getenv("APP_VERSION", "dev")} # --- Portfolio API --- class PortfolioItemRequest(BaseModel): broker: str ticker: str name: str quantity: int avg_price: int class PortfolioUpdateRequest(BaseModel): broker: Optional[str] = None ticker: Optional[str] = None name: Optional[str] = None quantity: Optional[int] = None avg_price: Optional[int] = None @app.get("/api/portfolio") def get_portfolio(): """전체 포트폴리오 조회 (현재가 + 손익 + 예수금 포함)""" items = get_all_portfolio() cash_rows = get_all_broker_cash() total_cash = sum(r["cash"] for r in cash_rows) if not items: return { "holdings": [], "cash": cash_rows, "summary": { "total_buy": 0, "total_eval": 0, "total_profit": 0, "total_profit_rate": 0.0, "total_cash": total_cash, "total_assets": total_cash, }, } tickers = list({item["ticker"] for item in items}) prices = get_current_prices(tickers) holdings = [] total_buy = 0 total_eval = 0 for item in items: current_price = prices.get(item["ticker"]) buy_amount = item["avg_price"] * item["quantity"] eval_amount = current_price * item["quantity"] if current_price is not None else None profit_amount = (eval_amount - buy_amount) if eval_amount is not None else None profit_rate = round((profit_amount / buy_amount) * 100, 2) if (profit_amount is not None and buy_amount) else None holdings.append({ "id": item["id"], "broker": item["broker"], "ticker": item["ticker"], "name": item["name"], "quantity": item["quantity"], "avg_price": item["avg_price"], "current_price": current_price, "eval_amount": eval_amount, "profit_amount": profit_amount, "profit_rate": profit_rate, }) total_buy += buy_amount if eval_amount is not None: total_eval += eval_amount total_profit = total_eval - total_buy total_profit_rate = round((total_profit / total_buy) * 100, 2) if total_buy else 0.0 return { "holdings": holdings, "cash": cash_rows, "summary": { "total_buy": total_buy, "total_eval": total_eval, "total_profit": total_profit, "total_profit_rate": total_profit_rate, "total_cash": total_cash, "total_assets": total_eval + total_cash, }, } @app.post("/api/portfolio", status_code=201) def create_portfolio_item(req: PortfolioItemRequest): """포트폴리오 종목 추가""" item_id = add_portfolio_item(req.broker, req.ticker, req.name, req.quantity, req.avg_price) return {"id": item_id, "ok": True} @app.put("/api/portfolio/{item_id}") def update_portfolio(item_id: int, req: PortfolioUpdateRequest): """포트폴리오 종목 수정""" if get_portfolio_item(item_id) is None: return JSONResponse(status_code=404, content={"error": "Item not found"}) update_portfolio_item(item_id, **req.dict()) return {"ok": True} @app.delete("/api/portfolio/{item_id}") def delete_portfolio(item_id: int): """포트폴리오 종목 삭제""" if not delete_portfolio_item(item_id): return JSONResponse(status_code=404, content={"error": "Item not found"}) return {"ok": True} # --- Broker Cash API --- class BrokerCashRequest(BaseModel): broker: str cash: int @app.get("/api/portfolio/cash") def list_broker_cash(): """증권사별 예수금 전체 조회""" return get_all_broker_cash() @app.put("/api/portfolio/cash") def set_broker_cash(req: BrokerCashRequest): """증권사 예수금 등록 또는 수정 (upsert)""" upsert_broker_cash(req.broker, req.cash) return {"ok": True, "broker": req.broker, "cash": req.cash} @app.delete("/api/portfolio/cash/{broker}") def remove_broker_cash(broker: str): """증권사 예수금 삭제""" if not delete_broker_cash(broker): return JSONResponse(status_code=404, content={"error": "Broker not found"}) return {"ok": True}