From b04bbef5b963a4a183c20d80cf8bf753150ca357 Mon Sep 17 00:00:00 2001
From: Aevann1 <Aevann1@protonmail.com>
Date: Mon, 2 Aug 2021 16:36:58 +0200
Subject: [PATCH] dfdf

---
 drama/routes/votes.py | 15 +++------------
 1 file changed, 3 insertions(+), 12 deletions(-)

diff --git a/drama/routes/votes.py b/drama/routes/votes.py
index e36a51ec6..aed31e04e 100644
--- a/drama/routes/votes.py
+++ b/drama/routes/votes.py
@@ -13,19 +13,10 @@ def admin_vote_info_get(v):
 	link = request.args.get("link")
 	if not link: return render_template("votes.html", v=v)
 
-	ids = re.search("://[^/]+\w+/post/(\w+)/[^/]+(/(\w+))?", link)
-
-	try:
-		post_id = ids.group(1)
-		comment_id = ids.group(3)
-	except: abort(400)
-
-	if comment_id:
-		thing = get_comment(int(comment_id), v=v)
-
-	else:
-		thing = get_post(int(post_id), v=v)
 
+	if "t2_" in link: thing = get_post(int(link.split("t2_")[1]), v=v)
+	elif "t3_" in link: thing = get_comment(int(link.split("t3_")[1]), v=v)
+	else: abort(400)
 
 	if isinstance(thing, Submission):