forked from rDrama/rDrama
203 lines
5.5 KiB
Python
Executable File
203 lines
5.5 KiB
Python
Executable File
from files.helpers.wrappers import *
|
|
from files.helpers.get import *
|
|
from files.classes import *
|
|
from flask import *
|
|
from files.__main__ import app, limiter
|
|
from sqlalchemy.orm import joinedload
|
|
|
|
|
|
@app.get("/votes")
|
|
@auth_desired
|
|
def admin_vote_info_get(v):
|
|
|
|
|
|
link = request.values.get("link")
|
|
if not link: return render_template("votes.html", v=v)
|
|
|
|
try:
|
|
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)
|
|
except: abort(400)
|
|
|
|
if isinstance(thing, Submission):
|
|
|
|
ups = g.db.query(Vote
|
|
).options(joinedload(Vote.user)
|
|
).filter_by(submission_id=thing.id, vote_type=1
|
|
).order_by(Vote.id).all()
|
|
|
|
downs = g.db.query(Vote
|
|
).options(joinedload(Vote.user)
|
|
).filter_by(submission_id=thing.id, vote_type=-1
|
|
).order_by(Vote.id).all()
|
|
|
|
elif isinstance(thing, Comment):
|
|
|
|
ups = g.db.query(CommentVote
|
|
).options(joinedload(CommentVote.user)
|
|
).filter_by(comment_id=thing.id, vote_type=1
|
|
).order_by(CommentVote.id).all()
|
|
|
|
downs = g.db.query(CommentVote
|
|
).options(joinedload(CommentVote.user)
|
|
).filter_by(comment_id=thing.id, vote_type=-1
|
|
).order_by(CommentVote.id).all()
|
|
|
|
else: abort(400)
|
|
|
|
return render_template("votes.html",
|
|
v=v,
|
|
thing=thing,
|
|
ups=ups,
|
|
downs=downs,)
|
|
|
|
|
|
|
|
@app.post("/vote/post/<post_id>/<new>")
|
|
@auth_required
|
|
@validate_formkey
|
|
def api_vote_post(post_id, new, v):
|
|
|
|
if new not in ["-1", "0", "1"]: abort(400)
|
|
|
|
if request.headers.get("X-User-Type","") == "Bot": abort(403)
|
|
|
|
new = int(new)
|
|
|
|
post = get_post(post_id)
|
|
|
|
existing = g.db.query(Vote).options(lazyload('*')).filter_by(user_id=v.id, submission_id=post.id).first()
|
|
|
|
if existing and existing.vote_type == new: return "", 204
|
|
|
|
if existing:
|
|
if existing.vote_type == 0 and new != 0:
|
|
post.author.coins += 1
|
|
post.author.truecoins += 1
|
|
g.db.add(post.author)
|
|
existing.vote_type = new
|
|
g.db.add(existing)
|
|
elif existing.vote_type != 0 and new == 0:
|
|
post.author.coins -= 1
|
|
post.author.truecoins -= 1
|
|
g.db.add(post.author)
|
|
g.db.delete(existing)
|
|
else:
|
|
existing.vote_type = new
|
|
g.db.add(existing)
|
|
elif new != 0:
|
|
post.author.coins += 1
|
|
post.author.truecoins += 1
|
|
g.db.add(post.author)
|
|
vote = Vote(user_id=v.id,
|
|
vote_type=new,
|
|
submission_id=post_id,
|
|
app_id=v.client.application.id if v.client else None
|
|
)
|
|
g.db.add(vote)
|
|
|
|
if post.stickied and post.stickied.startswith("t:") and int(time.time()) > int(post.stickied[2:]):
|
|
post.stickied = None
|
|
g.db.add(post)
|
|
|
|
try:
|
|
g.db.flush()
|
|
post.upvotes = g.db.query(Vote.id).options(lazyload('*')).filter_by(submission_id=post.id, vote_type=1).count()
|
|
post.downvotes = g.db.query(Vote.id).options(lazyload('*')).filter_by(submission_id=post.id, vote_type=-1).count()
|
|
g.db.add(post)
|
|
g.db.commit()
|
|
except: g.db.rollback()
|
|
return "", 204
|
|
|
|
@app.post("/vote/comment/<comment_id>/<new>")
|
|
@auth_required
|
|
@validate_formkey
|
|
def api_vote_comment(comment_id, new, v):
|
|
|
|
if new not in ["-1", "0", "1"]: abort(400)
|
|
|
|
if request.headers.get("X-User-Type","") == "Bot": abort(403)
|
|
|
|
new = int(new)
|
|
|
|
try: comment_id = int(comment_id)
|
|
except:
|
|
try: comment_id = int(comment_id, 36)
|
|
except: abort(404)
|
|
|
|
comment = get_comment(comment_id)
|
|
|
|
existing = g.db.query(CommentVote).options(lazyload('*')).filter_by(user_id=v.id, comment_id=comment.id).first()
|
|
|
|
if existing and existing.vote_type == new: return "", 204
|
|
|
|
if existing:
|
|
if existing.vote_type == 0 and new != 0:
|
|
comment.author.coins += 1
|
|
comment.author.truecoins += 1
|
|
g.db.add(comment.author)
|
|
existing.vote_type = new
|
|
g.db.add(existing)
|
|
elif existing.vote_type != 0 and new == 0:
|
|
comment.author.coins -= 1
|
|
comment.author.truecoins -= 1
|
|
g.db.add(comment.author)
|
|
g.db.delete(existing)
|
|
else:
|
|
existing.vote_type = new
|
|
g.db.add(existing)
|
|
elif new != 0:
|
|
comment.author.coins += 1
|
|
comment.author.truecoins += 1
|
|
g.db.add(comment.author)
|
|
vote = CommentVote(user_id=v.id,
|
|
vote_type=new,
|
|
comment_id=comment_id,
|
|
app_id=v.client.application.id if v.client else None
|
|
)
|
|
|
|
g.db.add(vote)
|
|
|
|
try:
|
|
g.db.flush()
|
|
comment.upvotes = g.db.query(CommentVote.id).options(lazyload('*')).filter_by(comment_id=comment.id, vote_type=1).count()
|
|
comment.downvotes = g.db.query(CommentVote.id).options(lazyload('*')).filter_by(comment_id=comment.id, vote_type=-1).count()
|
|
g.db.add(comment)
|
|
g.db.commit()
|
|
except: g.db.rollback()
|
|
return "", 204
|
|
|
|
|
|
@app.post("/vote/poll/<comment_id>")
|
|
@auth_required
|
|
def api_vote_poll(comment_id, v):
|
|
|
|
vote = request.values.get("vote")
|
|
if vote == "true": new = 1
|
|
elif vote == "false": new = 0
|
|
else: abort(400)
|
|
|
|
comment_id = int(comment_id)
|
|
comment = get_comment(comment_id)
|
|
|
|
existing = g.db.query(CommentVote).options(lazyload('*')).filter_by(user_id=v.id, comment_id=comment.id).first()
|
|
|
|
if existing and existing.vote_type == new: return "", 204
|
|
|
|
if existing:
|
|
if new == 1:
|
|
existing.vote_type = new
|
|
g.db.add(existing)
|
|
else: g.db.delete(existing)
|
|
elif new == 1:
|
|
vote = CommentVote(user_id=v.id, vote_type=new, comment_id=comment.id)
|
|
g.db.add(vote)
|
|
|
|
try:
|
|
g.db.flush()
|
|
comment.upvotes = g.db.query(CommentVote.id).options(lazyload('*')).filter_by(comment_id=comment.id, vote_type=1).count()
|
|
g.db.add(comment)
|
|
g.db.commit()
|
|
except: g.db.rollback()
|
|
return "", 204 |