2022-10-04 15:24:58 +00:00
|
|
|
from files.classes import *
|
2022-12-11 23:44:34 +00:00
|
|
|
from files.helpers.config.const import *
|
2023-07-14 03:32:52 +00:00
|
|
|
from files.helpers.config.boosted_sites import *
|
2022-11-15 09:19:08 +00:00
|
|
|
from files.helpers.get import *
|
2023-05-07 19:26:08 +00:00
|
|
|
from files.helpers.alerts import *
|
2023-10-05 10:09:58 +00:00
|
|
|
from files.helpers.can_see import *
|
2022-11-15 09:19:08 +00:00
|
|
|
from files.routes.wrappers import *
|
|
|
|
from files.__main__ import app, limiter
|
2023-01-25 02:51:48 +00:00
|
|
|
from files.routes.routehelpers import get_alt_graph
|
2023-09-14 23:30:02 +00:00
|
|
|
from sqlalchemy.exc import IntegrityError, OperationalError
|
2023-09-09 14:23:31 +00:00
|
|
|
from sqlalchemy.orm.exc import StaleDataError
|
2022-10-04 15:24:58 +00:00
|
|
|
|
2023-01-23 12:40:44 +00:00
|
|
|
from math import floor
|
2023-08-03 02:52:29 +00:00
|
|
|
from datetime import datetime
|
2023-01-23 12:40:44 +00:00
|
|
|
|
2022-10-11 18:18:49 +00:00
|
|
|
def vote_post_comment(target_id, new, v, cls, vote_cls):
|
2022-10-11 13:01:39 +00:00
|
|
|
if new == "-1" and DISABLE_DOWNVOTES: abort(403)
|
2022-11-26 04:52:47 +00:00
|
|
|
if new not in {"-1", "0", "1"}: abort(400)
|
2023-07-07 23:18:03 +00:00
|
|
|
|
|
|
|
if request.headers.get("Authorization"):
|
2023-07-17 14:49:07 +00:00
|
|
|
abort(403, "Bots aren't allowed to vote right now!")
|
2023-07-07 23:18:03 +00:00
|
|
|
|
2022-10-04 15:24:58 +00:00
|
|
|
new = int(new)
|
2022-10-11 18:18:49 +00:00
|
|
|
target = None
|
2023-06-07 23:26:32 +00:00
|
|
|
if cls == Post:
|
2022-10-11 18:18:49 +00:00
|
|
|
target = get_post(target_id)
|
|
|
|
elif cls == Comment:
|
|
|
|
target = get_comment(target_id)
|
2023-06-23 13:46:42 +00:00
|
|
|
if not target.parent_post and not target.wall_user_id: abort(404)
|
2022-10-11 18:18:49 +00:00
|
|
|
else:
|
|
|
|
abort(404)
|
2022-10-04 15:24:58 +00:00
|
|
|
|
2023-10-05 10:09:58 +00:00
|
|
|
if not can_see(v, target): abort(403)
|
2022-11-12 13:55:16 +00:00
|
|
|
|
2022-10-04 15:24:58 +00:00
|
|
|
coin_delta = 1
|
2022-10-11 18:18:49 +00:00
|
|
|
if v.id == target.author.id:
|
2022-10-04 15:24:58 +00:00
|
|
|
coin_delta = 0
|
|
|
|
|
2023-11-06 11:41:37 +00:00
|
|
|
if v.truescore < 10:
|
|
|
|
coin_delta = 0
|
|
|
|
|
2022-11-27 23:16:33 +00:00
|
|
|
alt = False
|
2023-01-25 02:51:48 +00:00
|
|
|
if target.author.id in [x.id for x in get_alt_graph(v.id)]:
|
2022-11-15 14:55:21 +00:00
|
|
|
coin_delta = -1
|
2022-11-27 23:16:33 +00:00
|
|
|
alt = True
|
2022-11-15 14:55:21 +00:00
|
|
|
|
2023-08-22 21:38:12 +00:00
|
|
|
g.db.flush()
|
2023-03-16 06:27:58 +00:00
|
|
|
existing = g.db.query(vote_cls).filter_by(user_id=v.id)
|
2022-10-11 18:18:49 +00:00
|
|
|
if vote_cls == Vote:
|
2023-06-07 23:26:32 +00:00
|
|
|
existing = existing.filter_by(post_id=target.id)
|
2022-10-11 18:18:49 +00:00
|
|
|
elif vote_cls == CommentVote:
|
|
|
|
existing = existing.filter_by(comment_id=target.id)
|
|
|
|
else:
|
|
|
|
abort(400)
|
2022-10-11 18:29:14 +00:00
|
|
|
existing = existing.one_or_none()
|
2023-01-01 11:36:20 +00:00
|
|
|
|
2023-02-22 17:27:33 +00:00
|
|
|
|
2024-02-06 22:04:43 +00:00
|
|
|
if cls == Post and target.effortpost:
|
|
|
|
if SITE_NAME == 'WPD':
|
|
|
|
coin_mult = 8
|
|
|
|
else:
|
|
|
|
coin_mult = 4
|
|
|
|
else:
|
|
|
|
coin_mult = 1
|
|
|
|
|
|
|
|
if SITE_NAME == 'WPD':
|
|
|
|
coin_mult *= 4
|
|
|
|
|
|
|
|
if IS_EVENT():
|
|
|
|
coin_mult *= 2
|
|
|
|
|
|
|
|
if IS_HOMOWEEN() and target.author.zombie > 0:
|
|
|
|
coin_mult += 1
|
2023-02-08 03:36:24 +00:00
|
|
|
|
2023-09-28 23:58:09 +00:00
|
|
|
|
2022-10-23 22:23:51 +00:00
|
|
|
coin_value = coin_delta * coin_mult
|
2022-10-04 15:24:58 +00:00
|
|
|
|
2023-08-02 08:49:47 +00:00
|
|
|
imlazy = 0
|
|
|
|
|
2023-10-10 19:34:38 +00:00
|
|
|
if existing and existing.vote_type == new: return ""
|
2022-10-04 15:24:58 +00:00
|
|
|
if existing:
|
|
|
|
if existing.vote_type == 0 and new != 0:
|
2023-08-02 08:49:47 +00:00
|
|
|
imlazy = 1
|
2022-10-04 15:24:58 +00:00
|
|
|
existing.vote_type = new
|
2022-10-23 22:23:51 +00:00
|
|
|
existing.coins = coin_value
|
2023-03-16 06:27:58 +00:00
|
|
|
g.db.add(existing)
|
2022-10-04 15:24:58 +00:00
|
|
|
elif existing.vote_type != 0 and new == 0:
|
2023-08-02 08:49:47 +00:00
|
|
|
imlazy = 2
|
|
|
|
existing_coins = existing.coins
|
2023-03-16 06:27:58 +00:00
|
|
|
g.db.delete(existing)
|
2022-10-04 15:24:58 +00:00
|
|
|
else:
|
|
|
|
existing.vote_type = new
|
2023-03-16 06:27:58 +00:00
|
|
|
g.db.add(existing)
|
2022-10-04 15:24:58 +00:00
|
|
|
elif new != 0:
|
2023-08-02 08:49:47 +00:00
|
|
|
imlazy = 3
|
2022-10-04 15:24:58 +00:00
|
|
|
|
2023-09-04 18:25:51 +00:00
|
|
|
real = not alt and (new == -1 or v.has_real_votes)
|
2022-10-11 18:18:49 +00:00
|
|
|
vote = None
|
|
|
|
if vote_cls == Vote:
|
|
|
|
vote = Vote(user_id=v.id,
|
|
|
|
vote_type=new,
|
2023-06-07 23:26:32 +00:00
|
|
|
post_id=target_id,
|
2022-10-23 22:23:51 +00:00
|
|
|
real=real,
|
|
|
|
coins=coin_value
|
2022-10-11 18:18:49 +00:00
|
|
|
)
|
2023-01-01 11:36:20 +00:00
|
|
|
elif vote_cls == CommentVote:
|
2022-10-11 18:18:49 +00:00
|
|
|
vote = CommentVote(user_id=v.id,
|
|
|
|
vote_type=new,
|
|
|
|
comment_id=target_id,
|
2022-10-23 22:23:51 +00:00
|
|
|
real=real,
|
|
|
|
coins=coin_value
|
2022-10-11 18:18:49 +00:00
|
|
|
)
|
2023-03-16 06:27:58 +00:00
|
|
|
g.db.add(vote)
|
2022-10-11 18:18:49 +00:00
|
|
|
|
|
|
|
# this is hacky but it works, we should probably do better later
|
|
|
|
def get_vote_count(dir, real_instead_of_dir):
|
2023-09-04 15:23:28 +00:00
|
|
|
try: g.db.flush()
|
2023-09-09 14:23:31 +00:00
|
|
|
except StaleDataError:
|
|
|
|
abort(500, "You already cancelled your vote on this!")
|
2023-09-04 15:23:28 +00:00
|
|
|
except IntegrityError as e:
|
|
|
|
if str(e).startswith('(psycopg2.errors.UniqueViolation) duplicate key value violates unique constraint "'):
|
|
|
|
abort(400, "You already voted on this!")
|
|
|
|
raise
|
2023-09-14 23:30:02 +00:00
|
|
|
except OperationalError as e:
|
|
|
|
if str(e).startswith('(psycopg2.errors.QueryCanceled) canceling statement due to statement timeout'):
|
|
|
|
abort(409, f"Statement timeout while trying to register your vote!")
|
|
|
|
raise
|
2023-09-04 15:23:28 +00:00
|
|
|
|
2023-03-16 06:27:58 +00:00
|
|
|
votes = g.db.query(vote_cls)
|
2022-10-11 18:18:49 +00:00
|
|
|
if real_instead_of_dir:
|
2022-12-01 15:04:08 +00:00
|
|
|
votes = votes.filter(vote_cls.real == True)
|
2022-10-12 05:27:27 +00:00
|
|
|
else:
|
2022-12-01 15:04:08 +00:00
|
|
|
votes = votes.filter(vote_cls.vote_type == dir)
|
2022-10-12 05:27:27 +00:00
|
|
|
|
|
|
|
if vote_cls == Vote:
|
2023-06-07 23:26:32 +00:00
|
|
|
votes = votes.filter(vote_cls.post_id == target.id)
|
2022-10-11 18:18:49 +00:00
|
|
|
elif vote_cls == CommentVote:
|
2022-12-01 15:04:08 +00:00
|
|
|
votes = votes.filter(vote_cls.comment_id == target.id)
|
2022-10-11 18:18:49 +00:00
|
|
|
else:
|
|
|
|
return 0
|
2023-07-18 11:38:33 +00:00
|
|
|
|
2023-08-20 16:03:36 +00:00
|
|
|
return votes.count()
|
2023-07-18 11:38:33 +00:00
|
|
|
|
2022-10-11 18:18:49 +00:00
|
|
|
target.upvotes = get_vote_count(1, False)
|
|
|
|
target.downvotes = get_vote_count(-1, False)
|
2022-11-07 00:47:12 +00:00
|
|
|
|
2022-12-25 00:54:53 +00:00
|
|
|
if SITE_NAME == 'rDrama':
|
2022-11-07 01:35:46 +00:00
|
|
|
target.realupvotes = get_vote_count(0, True) # first arg is ignored here
|
2023-12-24 23:29:15 +00:00
|
|
|
elif cls == Post and target.hole in {'sandshit', 'slavshit'}:
|
2024-01-13 05:03:03 +00:00
|
|
|
target.realupvotes = target.upvotes
|
2023-01-24 10:05:16 +00:00
|
|
|
else:
|
|
|
|
target.realupvotes = target.upvotes - target.downvotes
|
|
|
|
|
|
|
|
mul = 1
|
|
|
|
if target.is_approved == PROGSTACK_ID:
|
|
|
|
mul = PROGSTACK_MUL
|
2023-06-07 23:26:32 +00:00
|
|
|
elif cls == Post and (any(i in target.title.lower() for i in ENCOURAGED) or any(i in str(target.url).lower() for i in ENCOURAGED2)):
|
2023-05-07 19:19:48 +00:00
|
|
|
mul = PROGSTACK_MUL
|
2023-05-13 23:45:18 +00:00
|
|
|
send_notification(AEVANN_ID, target.permalink)
|
2023-09-09 14:24:12 +00:00
|
|
|
elif SITE == 'rdrama.net' and target.author_id == 29:
|
2024-02-23 17:29:43 +00:00
|
|
|
mul = 4
|
2024-02-23 17:36:00 +00:00
|
|
|
elif cls == Post and (target.effortpost or target.hole == 'countryclub'):
|
|
|
|
mul = 4
|
2023-09-28 23:58:09 +00:00
|
|
|
elif target.author.progressivestack or (IS_HOMOWEEN() and target.author.zombie < 0) or target.author.admin_level >= PERMS['IS_PERMA_PROGSTACKED']:
|
2023-01-25 10:57:50 +00:00
|
|
|
mul = 2
|
2024-02-23 17:31:27 +00:00
|
|
|
elif SITE == 'rdrama.net' and target.author.new_user and not target.author.alts:
|
|
|
|
mul = 2
|
2024-02-23 17:31:45 +00:00
|
|
|
elif SITE == 'rdrama.net' and 6 <= datetime.fromtimestamp(target.created_utc).hour <= 10:
|
|
|
|
mul = 2
|
2023-06-07 23:26:32 +00:00
|
|
|
elif SITE == 'rdrama.net' and cls == Post:
|
2023-10-28 18:05:24 +00:00
|
|
|
if target.domain.endswith('.win') \
|
|
|
|
or any(i in target.domain for i in ('forum','community','chan','lemmy','mastodon')) \
|
|
|
|
or (target.domain in BOOSTED_SITES and not target.url.startswith('/')):
|
2023-01-24 10:05:16 +00:00
|
|
|
mul = 2
|
2024-02-23 18:30:09 +00:00
|
|
|
elif target.hole in STEALTH_HOLES + {'highrollerclub'}:
|
2023-07-05 20:39:15 +00:00
|
|
|
mul = 2
|
2023-10-07 17:55:50 +00:00
|
|
|
elif target.hole in BOOSTED_HOLES:
|
2023-05-06 12:18:26 +00:00
|
|
|
mul = 1.25
|
2023-01-24 10:05:16 +00:00
|
|
|
|
2024-02-23 17:29:33 +00:00
|
|
|
if target.body_html and target.author.id != 8768:
|
|
|
|
x = target.body_html.count('" target="_blank" rel="nofollow noopener">')
|
|
|
|
x += target.body_html.count('" rel="nofollow noopener" target="_blank">')
|
|
|
|
target.realupvotes += min(x*2, 20)
|
|
|
|
mul += min(x/10, 1)
|
|
|
|
|
2023-01-24 10:05:16 +00:00
|
|
|
target.realupvotes = floor(target.realupvotes * mul)
|
2022-11-07 00:46:48 +00:00
|
|
|
|
2023-03-16 06:27:58 +00:00
|
|
|
g.db.add(target)
|
2023-08-02 08:49:47 +00:00
|
|
|
|
|
|
|
if imlazy == 1:
|
|
|
|
target.author.pay_account('coins', coin_value)
|
|
|
|
target.author.truescore += coin_delta
|
|
|
|
elif imlazy == 2:
|
|
|
|
target.author.charge_account('coins', existing.coins, should_check_balance=False)
|
|
|
|
target.author.truescore -= coin_delta
|
|
|
|
elif imlazy == 3:
|
|
|
|
target.author.pay_account('coins', coin_value)
|
|
|
|
target.author.truescore += coin_delta
|
|
|
|
|
2023-10-10 19:34:38 +00:00
|
|
|
return ""
|
2022-10-04 15:24:58 +00:00
|
|
|
|
2023-07-25 20:06:18 +00:00
|
|
|
@app.get("/votes/<link>")
|
|
|
|
@limiter.limit(DEFAULT_RATELIMIT, deduct_when=lambda response: response.status_code < 400)
|
|
|
|
@limiter.limit(DEFAULT_RATELIMIT, deduct_when=lambda response: response.status_code < 400, key_func=get_ID)
|
|
|
|
@auth_required
|
|
|
|
def vote_info_get(v, link):
|
|
|
|
try:
|
2023-10-29 12:05:47 +00:00
|
|
|
if "p_" in link: obj = get_post(int(link.split("p_")[1]), v=v)
|
|
|
|
elif "c_" in link: obj = get_comment(int(link.split("c_")[1]), v=v)
|
2023-07-25 20:06:18 +00:00
|
|
|
else: abort(400)
|
|
|
|
except: abort(400)
|
|
|
|
|
2023-10-29 12:05:47 +00:00
|
|
|
if obj.ghost and v.admin_level < PERMS['SEE_GHOST_VOTES']:
|
2023-07-25 20:06:18 +00:00
|
|
|
abort(403)
|
|
|
|
|
2024-01-12 07:20:32 +00:00
|
|
|
if obj.author.shadowbanned and not (v and v.admin_level >= PERMS['USER_SHADOWBAN']):
|
2023-07-25 20:06:18 +00:00
|
|
|
abort(500)
|
|
|
|
|
2023-10-29 12:05:47 +00:00
|
|
|
if isinstance(obj, Post):
|
2023-07-25 20:06:18 +00:00
|
|
|
query = g.db.query(Vote).join(Vote.user).filter(
|
2023-10-29 12:05:47 +00:00
|
|
|
Vote.post_id == obj.id,
|
2023-07-25 20:06:18 +00:00
|
|
|
).order_by(Vote.created_utc)
|
|
|
|
|
|
|
|
ups = query.filter(Vote.vote_type == 1).all()
|
|
|
|
downs = query.filter(Vote.vote_type == -1).all()
|
|
|
|
|
2023-10-29 12:05:47 +00:00
|
|
|
elif isinstance(obj, Comment):
|
2023-07-25 20:06:18 +00:00
|
|
|
query = g.db.query(CommentVote).join(CommentVote.user).filter(
|
2023-10-29 12:05:47 +00:00
|
|
|
CommentVote.comment_id == obj.id,
|
2023-07-25 20:06:18 +00:00
|
|
|
).order_by(CommentVote.created_utc)
|
|
|
|
|
|
|
|
ups = query.filter(CommentVote.vote_type == 1).all()
|
|
|
|
downs = query.filter(CommentVote.vote_type == -1).all()
|
|
|
|
|
|
|
|
else: abort(400)
|
|
|
|
|
|
|
|
return render_template("votes.html",
|
|
|
|
v=v,
|
2023-10-29 12:05:47 +00:00
|
|
|
obj=obj,
|
2023-07-25 20:06:18 +00:00
|
|
|
ups=ups,
|
|
|
|
downs=downs)
|
2022-12-01 15:04:08 +00:00
|
|
|
|
2022-12-29 10:39:10 +00:00
|
|
|
@app.post("/vote/post/<int:post_id>/<new>")
|
2023-02-27 05:33:45 +00:00
|
|
|
@limiter.limit('1/second', scope=rpath)
|
2023-04-02 06:52:26 +00:00
|
|
|
@limiter.limit('1/second', scope=rpath, key_func=get_ID)
|
2023-07-13 13:50:46 +00:00
|
|
|
@limiter.limit("60/minute;1000/hour;2000/day", deduct_when=lambda response: response.status_code < 400)
|
|
|
|
@limiter.limit("60/minute;1000/hour;2000/day", deduct_when=lambda response: response.status_code < 400, key_func=get_ID)
|
2023-09-14 16:49:46 +00:00
|
|
|
@auth_required
|
2022-10-11 18:18:49 +00:00
|
|
|
def vote_post(post_id, new, v):
|
2023-06-07 23:26:32 +00:00
|
|
|
return vote_post_comment(post_id, new, v, Post, Vote)
|
2022-10-11 18:18:49 +00:00
|
|
|
|
2022-12-29 10:39:10 +00:00
|
|
|
@app.post("/vote/comment/<int:comment_id>/<new>")
|
2023-02-27 05:33:45 +00:00
|
|
|
@limiter.limit('1/second', scope=rpath)
|
2023-04-02 06:52:26 +00:00
|
|
|
@limiter.limit('1/second', scope=rpath, key_func=get_ID)
|
2023-07-13 13:50:46 +00:00
|
|
|
@limiter.limit("60/minute;1000/hour;2000/day", deduct_when=lambda response: response.status_code < 400)
|
|
|
|
@limiter.limit("60/minute;1000/hour;2000/day", deduct_when=lambda response: response.status_code < 400, key_func=get_ID)
|
2023-09-14 16:49:46 +00:00
|
|
|
@auth_required
|
2022-10-04 15:24:58 +00:00
|
|
|
def vote_comment(comment_id, new, v):
|
2022-10-11 18:18:49 +00:00
|
|
|
return vote_post_comment(comment_id, new, v, Comment, CommentVote)
|