Add new Flask routes to get subscriptions info

1 files changed, 15 insertions, 3 deletions

diff --git a/wsgi/__init__.py b/wsgi/__init__.py
index d3343b0..d029c12 100644
--- a/wsgi/__init__.py
+++ b/wsgi/__init__.py
@@ -5,11 +5,23 @@ from components.videos import VideoTuple
 from flask import Flask
 from flask_cors import CORS
 
+from .utils import vid_dicts_from_tuple_list, sub_info_from_dict
+
 app = Flask(__name__)
 CORS(app)
 
 @app.route("/vid-from-link/<id>")
 def videos_from_link(id: str) -> List[Dict[str, Any]]:
-    sub = subscriptions.find_one({"_id": id})
-    assert sub
-    return [VideoTuple._make(vid)._asdict() for vid in sub["videos"]]
+    sub_dict = subscriptions.find_one({"_id": id})
+    assert sub_dict
+    return vid_dicts_from_tuple_list(sub_dict["videos"])
+
+@app.route("/sub-info/<id>")
+def sub_dict(id: str) -> Dict[str, Any]:
+    sub_dict = subscriptions.find_one({"_id": id})
+    assert sub_dict
+    return sub_info_from_dict(sub_dict)
+
+@app.route("/subs-info")
+def subs_info() -> List[Dict[str, Any]]:
+    return [sub_info_from_dict(sub_dict) for sub_dict in subscriptions.find()]