From 6e2520431a8108c5cc310d94b60aa47218c87787 Mon Sep 17 00:00:00 2001 From: A Farzat Date: Mon, 6 Oct 2025 13:14:34 +0300 Subject: Add new Flask routes to get subscriptions info --- wsgi/__init__.py | 18 +++++++++++++++--- 1 file changed, 15 insertions(+), 3 deletions(-) (limited to 'wsgi/__init__.py') 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/") 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/") +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()] -- cgit v1.2.3-70-g09d2