fix: update data

routes/data.py

@@ -18,7 +18,8 @@ router = APIRouter()
     status_code=status.HTTP_200_OK
 )
 async def update_daily_price():
-    return Indicator.update_daily_price()
+    asyncio.create_task(Indicator.update_daily_price())
+    return {"message": "Updating"}
 
 
 @router.get(
@@ -36,7 +37,8 @@ async def update_entire_price():
     status_code=status.HTTP_200_OK
 )
 async def update_daily_rsi():
-    return Indicator.update_daily_rsi()
+    asyncio.create_task(Indicator.update_daily_rsi())
+    return {"message": "Updating"}
 
 
 @router.get(

services/indicator.py

@@ -83,7 +83,10 @@ class Indicator:
                     collection.find_one(sort=[("_id", DESCENDING)])
                 ret = None
                 if newest_record["time"] != last_date:
-                    lst_symbols = list(newest_record["value"].keys())
+                    rsi_collection = database.get_collection("rsi")
+                    tmp_record = \
+                        rsi_collection.find_one(sort=[("_id", DESCENDING)])
+                    lst_symbols = list(tmp_record.keys())[2:]
                     record = {}
                     record["time"] = last_date
                     values = []
@@ -244,7 +247,7 @@ class Indicator:
                 uri = os.environ.get("MONGODB_URI")
                 client = MongoClient(uri)
                 database = client.get_database("data")
-                collection = database.get_collection("price")
+                collection = database.get_collection("rsi")
                 tmp_df = longterm_ohlc_data("FPT",
                                             start_date,
                                             end_date,
@@ -258,13 +261,15 @@ class Indicator:
                 newest_record = \
                     collection.find_one(sort=[("_id", DESCENDING)])
                 if newest_record["time"] != last_date:
-                    lst_symbols = list(newest_record["value"].keys())
+                    lst_symbols = list(newest_record.keys())[2:]
                     record = {}
                     record["time"] = last_date
-                    str_symbols = ','.join(lst_symbols)
-                    data = requests.get('https://apipubaws.tcbs.com.vn/stock-insight/v1/stock/second-tc-price?tickers={}'.format(str_symbols)).json()
-                    for i in data["data"]:
-                        record[i["t"]] = i["rsi"]
+                    for batch_idx in range(10):
+                        str_symbols = ','.join(
+                            lst_symbols[batch_idx*10:(batch_idx+1)*10])
+                        data = requests.get('https://apipubaws.tcbs.com.vn/stock-insight/v1/stock/second-tc-price?tickers={}'.format(str_symbols)).json()
+                        for i in data["data"]:
+                            record[i["t"]] = i["rsi"]
                     collection.insert_one(record)
                     ret = {"message": "Updated data."}
                 else: