From 606d25294cd4441d5acde38819c36d05ab08ca77 Mon Sep 17 00:00:00 2001
From: MuslemRahimi <moslem_rahimi@hotmail.de>
Date: Sat, 27 Jul 2024 18:08:06 +0200
Subject: [PATCH] add reddit tracker endpoint

---
 app/main.py | 37 +++++++++++++++++++++++++++++++++++++
 1 file changed, 37 insertions(+)

diff --git a/app/main.py b/app/main.py
index d7cb309..fe9d84d 100755
--- a/app/main.py
+++ b/app/main.py
@@ -3266,6 +3266,43 @@ async def get_cramer_tracker(api_key: str = Security(get_api_key)):
     redis_client.set(cache_key, compressed_data)
     redis_client.expire(cache_key, 3600*3600)
 
+    return StreamingResponse(
+        io.BytesIO(compressed_data),
+        media_type="application/json",
+        headers={"Content-Encoding": "gzip"}
+    )
+
+@app.get("/reddit-tracker")
+async def get_reddit_tracker(api_key: str = Security(get_api_key)):
+    cache_key = f"reddit-tracker"
+    cached_result = redis_client.get(cache_key)
+    if cached_result:
+        return StreamingResponse(
+            io.BytesIO(cached_result),
+            media_type="application/json",
+            headers={"Content-Encoding": "gzip"}
+        )
+
+    try:
+        with open(f"json/reddit-tracker/wallstreetbets/data.json", 'rb') as file:
+            latest_post = orjson.loads(file.read())[0:25]
+    except:
+        latest_post = []
+
+    try:
+        with open(f"json/reddit-tracker/wallstreetbets/stats.json", 'rb') as file:
+            stats = orjson.loads(file.read())
+    except:
+        stats = []
+
+    res = {'posts': latest_post, 'stats': stats}
+
+    data = orjson.dumps(res)
+    compressed_data = gzip.compress(data)
+
+    redis_client.set(cache_key, compressed_data)
+    redis_client.expire(cache_key, 60*15)
+
     return StreamingResponse(
         io.BytesIO(compressed_data),
         media_type="application/json",