dylan/gander
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

FEAT: Added vods to search bar (needs to be integrated with search bar

Browse Source
This commit is contained in:
white
2025-03-06 17:17:32 +00:00
parent 00b491ac02
commit ead61bbabd
1 changed files with 18 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
web_server/blueprints/search_bar.py
View File

@@ -56,7 +56,7 @@ def search_results():
users = db.fetchall("SELECT user_id, username, is_live FROM users") users = db.fetchall("SELECT user_id, username, is_live FROM users")
for u in users: for u in users:
key = u.get("username") key = u.get("username")
score = rank_results(query, key) score = rank_results(query.lower(), key.lower())
u["score"] = score u["score"] = score
if score < 2: if score < 2:
res_dict.append(u) res_dict.append(u)
@@ -74,15 +74,29 @@ def search_results():
for s in streams: for s in streams:
key = s.get("username") key = s.get("username")
score = rank_results(query, key) score = rank_results(query.lower(), key.lower())
s["score"] = score s["score"] = score
if score < 2: if score < 2:
res_dict.append(s) res_dict.append(s)
streams = sorted(res_dict, key=lambda d: d["score"]) streams = sorted(res_dict, key=lambda d: d["score"])
streams = streams[:4] streams = streams[:4]
# 3 VODs
res_dict = []
vods = db.fetchall("""SELECT v.vod_id, v.title, u.user_id, u.username
FROM vods as v JOIN users as u
ON v.user_id = u.user_id""")
for v in vods:
key = v.get("title")
score = rank_results(query.lower(), key.lower())
v["score"] = score
if score < 2:
res_dict.append(v)
vods = sorted(res_dict, key=lambda d: d["score"])
vods = vods[:4]
db.close_connection() db.close_connection()
print(query, streams, users, categories, flush=True) print(query, streams, users, categories, vods, flush=True)
return jsonify({"streams": streams, "categories": categories, "users": users}) return jsonify({"streams": streams, "categories": categories, "users": users, "vods": vods})