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):