1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
|
from datetime import datetime, UTC
from typing import Any, Dict, List, Tuple
from flask import Flask, request
from flask_cors import CORS
from pymongo.errors import DuplicateKeyError
from components.database import subscriptions
from components.subscriptions.main import Subscription
from components.videos import VideoTuple
from components.extractor.extract_sub_info import get_sub_info_from_yt_url
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) -> Tuple[List[Dict[str, Any]], int]:
sub_dict = subscriptions.find_one({"_id": id})
if sub_dict:
return vid_dicts_from_tuple_list(sub_dict["videos"]), 200
return [{'error': "Subscription %s not found"%id }], 404
@app.route("/sub-info/<id>")
def sub_dict(id: str) -> Tuple[Dict[str, Any], int]:
sub_dict = subscriptions.find_one({"_id": id})
if sub_dict:
return sub_info_from_dict(sub_dict), 200
return {'error': "Subscription %s not found"%id }, 404
@app.route("/subs-info")
def subs_info() -> List[Dict[str, Any]]:
return [sub_info_from_dict(sub_dict) for sub_dict in subscriptions.find()]
@app.post("/add-sub/")
def add_sub() -> Tuple[Dict[str, Any], int]:
try:
sub_info = get_sub_info_from_yt_url(request.form["url"])
time_between_fetches = int(request.form["time_between_fetches"])
except:
return {'error': 'Invalid data'}, 400
sub = Subscription(
_id=sub_info["id"],
link=sub_info["link"],
title=sub_info["title"],
time_between_fetches=time_between_fetches,
)
try:
sub.insert()
return sub_info_from_dict(sub.asdict()), 201
except DuplicateKeyError:
return {'error': "Subscription %s already exists"%sub_info["id"] }, 409
@app.post("/set-time-between-fetches/")
def set_time_between_fetches() -> Dict[str, Any]:
time_between_fetches = int(request.form["time_between_fetches"])
result = subscriptions.update_one(
{"_id": request.form["_id"]},
{"$set": {"time_between_fetches": time_between_fetches}}
)
if not result.modified_count:
raise Exception("Subscription %s not found" % request.form["_id"])
return {
"_id": request.form["_id"],
"time_between_fetches": time_between_fetches,
}
@app.delete("/delete-sub/<id>")
def delete_sub(id: str) -> Dict[str, Any]:
result = subscriptions.delete_one({"_id": id})
if not result.deleted_count:
raise Exception("Subscription %s not found" % id)
return { "_id": id, }
@app.post("/set-viewed/")
def set_viewed() -> Dict[str, Any]:
viewed_time_str = request.form.get("viewed_time")
if viewed_time_str:
viewed_time = datetime.fromisoformat(viewed_time_str)
else:
viewed_time = datetime.now(tz=UTC)
result = subscriptions.update_one(
{"_id": request.form["_id"]},
{"$set": {"last_viewed": viewed_time}}
)
if not result.modified_count:
raise Exception("Subscription %s not found" % request.form["_id"])
return {
"_id": request.form["_id"],
"last_viewed": viewed_time,
}
|
