75 lines
2.8 KiB
Python
75 lines
2.8 KiB
Python
|
from fastapi import APIRouter, Depends, HTTPException, Query
|
||
|
|
from pathlib import Path
|
||
|
|
import logging
|
||
|
|
from quote_db import QuoteDatabase
|
||
|
|
from quote_service import QuoteService
|
||
|
|
from quote_contracts import (
|
||
|
|
FetchQuoteRequest, FetchQuoteResponse,
|
||
|
|
ListSymbolsRequest, ListSymbolsResponse,
|
||
|
|
ListDatesRequest, ListDatesResponse,
|
||
|
|
ListSessionRequest, ListSessionResponse,
|
||
|
|
IngestRequest, IngestResponse,
|
||
|
|
FetchBrokersResponse, FetchDataResponse,
|
||
|
|
BrokersSymbolsResponse, IngestAllRequest, IngestAllResponse
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
router = APIRouter()
|
||
|
|
|
||
|
|
def get_db():
|
||
|
|
db = QuoteDatabase("quotes.db")
|
||
|
|
try:
|
||
|
|
yield db
|
||
|
|
finally:
|
||
|
|
db.conn.close()
|
||
|
|
|
||
|
|
def get_service(db: QuoteDatabase = Depends(get_db)):
|
||
|
|
return QuoteService(db=db)
|
||
|
|
|
||
|
|
@router.post("/ingestall", response_model = IngestAllResponse)
|
||
|
|
def ingest_archives(request: IngestAllRequest, service: QuoteService = Depends(get_service)):
|
||
|
|
return service.ingest_archives_from_folder(request)
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
@router.post("/ingest", response_model=IngestResponse)
|
||
|
|
def ingest_quotes(request: IngestRequest, service: QuoteService = Depends(get_service)):
|
||
|
|
return service.ingest_archive(request)
|
||
|
|
|
||
|
|
@router.get("/api/data", response_model=FetchDataResponse)
|
||
|
|
async def get_data(
|
||
|
|
broker_a: str = Query(...),
|
||
|
|
symbol_a: str = Query(...),
|
||
|
|
broker_b: str = Query(...),
|
||
|
|
symbol_b: str = Query(...),
|
||
|
|
time_range_hours: str = Query('all', description="Time range: 'all' or hours (1, 6, 24)"),
|
||
|
|
limit: int = Query(1000, description="Maximum number of records"),
|
||
|
|
service: QuoteService = Depends(get_service)
|
||
|
|
):
|
||
|
|
data = service.get_data(broker_a, symbol_a, broker_b, symbol_b, limit, time_range_hours)
|
||
|
|
return {"data": data}
|
||
|
|
|
||
|
|
@router.get("/brokers", response_model=FetchBrokersResponse)
|
||
|
|
def get_all_brokers(service: QuoteService = Depends(get_service)):
|
||
|
|
return service.get_all_brokers()
|
||
|
|
|
||
|
|
@router.post("/symbols", response_model=ListSymbolsResponse)
|
||
|
|
def get_symbols(request: ListSymbolsRequest, service: QuoteService = Depends(get_service)):
|
||
|
|
return service.get_symbols(request)
|
||
|
|
|
||
|
|
@router.post("/dates", response_model=ListDatesResponse)
|
||
|
|
def get_dates(request: ListDatesRequest, service: QuoteService = Depends(get_service)):
|
||
|
|
return service.get_dates(request)
|
||
|
|
|
||
|
|
@router.post("/sessions", response_model=ListSessionResponse)
|
||
|
|
def get_sessions(request: ListSessionRequest, service: QuoteService = Depends(get_service)):
|
||
|
|
return service.get_sessions(request)
|
||
|
|
|
||
|
|
@router.post("/quotes", response_model=FetchQuoteResponse)
|
||
|
|
def get_quotes(request: FetchQuoteRequest, service: QuoteService = Depends(get_service)):
|
||
|
|
return service.get_quotes(request)
|
||
|
|
|
||
|
|
@router.get("/brokers&symbols", response_model=BrokersSymbolsResponse)
|
||
|
|
def get_brokers_and_symbols(service: QuoteService = Depends(get_service)):
|
||
|
|
return service.get_brokers_and_symbols()
|