migrationt to websocket

app.py

@@ -5,11 +5,13 @@ from fastapi.staticfiles import StaticFiles
 from pymongo import MongoClient
 from datetime import datetime
 import uvicorn
+
 from urllib.parse import quote_plus
 from dotenv import load_dotenv
 import os
 import yaml
 
+
 with open("coin.yaml", "r") as file:
     data = yaml.safe_load(file)
 interest_coins = data["interest"]
@@ -17,7 +19,8 @@ interest_coins = data["interest"]
 load_dotenv()
 
 app = FastAPI()
-
+app.mount("/static", StaticFiles(directory="static"), name="static")
+data_points = []
 
 # Setup templates directory
 templates = Jinja2Templates(directory="templates")
@@ -45,6 +48,25 @@ async def view_coin_graph(request: Request, coin: str):
         return {"coin": "not found"}
 
 
+@app.get("/coin/{coin}/candle1min")
+async def view_coin_candle_graph(request: Request, coin: str):
+    if coin in interest_coins:
+        return templates.TemplateResponse(
+            request=request, name="candle_coin.html", context={"coin": coin}
+        )
+    else:
+        return {"coin": "not found"}
+
+
+@app.post("/update-data")
+async def update_data(data: dict):
+    data_points.append(data)
+    # Keep only last 300 points to ensure we have enough data to generate 100 candles
+    if len(data_points) > 300:
+        data_points.pop(0)
+    return {"status": "success"}
+
+
 @app.get("/data/{coin}")
 async def get_data(coin):
     collection = db[coin]
@@ -68,5 +90,117 @@ async def get_data(coin):
     return {"times": times, "prices": prices}
 
 
+@app.get("/data/{coin}/get-candle/all")
+async def get_candle_data_all(coin):
+    if coin in interest_coins:
+        target = coin + "USDT" + "_kline"
+        collection = db[target.lower()]
+        latest_doc = collection.find_one(sort=[("T", -1)])
+        latest_end_time = latest_doc["T"]
+        time_threshold = latest_end_time - (
+            3000
+        )  # 60 seconds before the latest endTime
+        pipeline = [
+            # Filter based on endTime
+            {"$match": {"T": {"$gte": time_threshold}}},
+            # Group by start time and end time
+            {
+                "$group": {
+                    "_id": {"startTime": "$t", "endTime": "$T"},
+                    "open": {"$first": "$o"},
+                    "high": {"$max": "$h"},
+                    "low": {"$min": "$l"},
+                    "close": {"$last": "$c"},
+                    "volume": {"$sum": "$v"},
+                    "weighted_price": {"$sum": {"$multiply": ["$c", "$v"]}},
+                    "count": {"$sum": 1},
+                }
+            },
+            # Sort by start time
+            {"$sort": {"_id.startTime": 1}},
+            # Reshape for output
+            {
+                "$project": {
+                    "_id": 0,
+                    "time": {"$multiply": ["$_id.startTime", 1000]},
+                    "open": 1,
+                    "high": 1,
+                    "low": 1,
+                    "close": 1,
+                    "volume": 1,
+                    "vwap": {
+                        "$cond": {
+                            "if": {"$eq": ["$volume", 0]},
+                            "then": 0,
+                            "else": {"$divide": ["$weighted_price", "$volume"]},
+                        }
+                    },
+                    "trades": "$count",
+                    "endTime": "$_id.endTime",
+                }
+            },
+        ]
+
+        candles = list(collection.aggregate(pipeline))
+        return candles
+    else:
+        return None
+
+
+@app.get("/data/{coin}/get-candle")
+async def get_candle_data(coin):
+    if coin in interest_coins:
+        target = coin + "USDT" + "_kline"
+        collection = db[target.lower()]
+        latest_doc = collection.find_one(sort=[("T", -1)])
+        latest_end_time = latest_doc["T"]
+        time_threshold = latest_end_time - (20)  # 60 seconds before the latest endTime
+        pipeline = [
+            # Filter based on endTime
+            {"$match": {"T": {"$gte": time_threshold}}},
+            # Group by start time and end time
+            {
+                "$group": {
+                    "_id": {"startTime": "$t", "endTime": "$T"},
+                    "open": {"$first": "$o"},
+                    "high": {"$max": "$h"},
+                    "low": {"$min": "$l"},
+                    "close": {"$last": "$c"},
+                    "volume": {"$sum": "$v"},
+                    "weighted_price": {"$sum": {"$multiply": ["$c", "$v"]}},
+                    "count": {"$sum": 1},
+                }
+            },
+            # Sort by start time
+            {"$sort": {"_id.startTime": 1}},
+            # Reshape for output
+            {
+                "$project": {
+                    "_id": 0,
+                    "time": {"$multiply": ["$_id.startTime", 1000]},
+                    "open": 1,
+                    "high": 1,
+                    "low": 1,
+                    "close": 1,
+                    "volume": 1,
+                    "vwap": {
+                        "$cond": {
+                            "if": {"$eq": ["$volume", 0]},
+                            "then": 0,
+                            "else": {"$divide": ["$weighted_price", "$volume"]},
+                        }
+                    },
+                    "trades": "$count",
+                    "endTime": "$_id.endTime",
+                }
+            },
+        ]
+
+        candles = list(collection.aggregate(pipeline))
+        return candles[0]
+    else:
+        return None
+
+
 if __name__ == "__main__":
     uvicorn.run("app:app", host="0.0.0.0", port=8000, reload=True)