Update main.py

main.py

@@ -2,7 +2,7 @@ from fastapi.responses import HTMLResponse
 from fastapi.templating import Jinja2Templates
 from fastapi import FastAPI, Request, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
-from pycatchs.levels import eqt
+from pycatchs.stdb import get_equity_data, get_stocks_data
 
 app = FastAPI()
 
@@ -35,7 +35,15 @@ async def read_root(request: Request):
 @app.get('/data')
 async def get_data(ticker: str, date: str, qty: int):
     try:
-        response = await eqt(ticker, date, qty)
+        response = get_equity_data(ticker, date, qty)
+        return response
+    except:
+        return {"Timeout" : "Error"}
+
+@app.get('/stocks')
+async def get_stocks_data():
+    try:
+        response = get_stocks_data()
         return response
     except:
         return {"Timeout" : "Error"}