forked from rDrama/rDrama
1019 lines
35 KiB
Python
1019 lines
35 KiB
Python
import qrcode
|
|
import io
|
|
import time
|
|
import math
|
|
from files.classes.user import ViewerRelationship
|
|
from files.helpers.alerts import *
|
|
from files.helpers.sanitize import *
|
|
from files.helpers.const import *
|
|
from files.mail import *
|
|
from flask import *
|
|
from files.__main__ import app, limiter, db_session
|
|
from pusher_push_notifications import PushNotifications
|
|
from collections import Counter
|
|
|
|
site = environ.get("DOMAIN").strip()
|
|
|
|
if PUSHER_ID: beams_client = PushNotifications(instance_id=PUSHER_ID, secret_key=PUSHER_KEY)
|
|
|
|
db = db_session()
|
|
|
|
votes1 = db.query(Vote.user_id, func.count(Vote.user_id)).filter(Vote.vote_type==1).group_by(Vote.user_id).order_by(func.count(Vote.user_id).desc()).all()
|
|
votes2 = db.query(CommentVote.user_id, func.count(CommentVote.user_id)).filter(CommentVote.vote_type==1).group_by(CommentVote.user_id).order_by(func.count(CommentVote.user_id).desc()).all()
|
|
votes3 = Counter(dict(votes1)) + Counter(dict(votes2))
|
|
users14 = db.query(User).filter(User.id.in_(votes3.keys())).all()
|
|
users15 = []
|
|
for user in users14:
|
|
users15.append((user, votes3[user.id]-user.post_count-user.comment_count))
|
|
users15 = sorted(users15, key=lambda x: x[1], reverse=True)[:25]
|
|
|
|
if site == 'rdrama.net':
|
|
topmakers = {}
|
|
for k, val in marseys.items():
|
|
if val in topmakers: topmakers[val] += 1
|
|
else: topmakers[val] = 1
|
|
|
|
topmakers.pop('unknown','anton-d')
|
|
topmakers2 = db.query(User).filter(func.lower(User.username).in_(topmakers.keys())).all()
|
|
topmakers3 = []
|
|
for user in topmakers2:
|
|
topmakers3.append((user, topmakers[user.username.lower()]))
|
|
topmakers = sorted(topmakers3, key=lambda x: x[1], reverse=True)[:25]
|
|
|
|
db.close()
|
|
|
|
@app.get("/grassed")
|
|
@auth_required
|
|
def grassed(v):
|
|
users = g.db.query(User).filter(User.ban_reason.like('grass award used by @%')).all()
|
|
|
|
return render_template("grassed.html", v=v, users=users)
|
|
|
|
@app.get("/agendaposters")
|
|
@auth_required
|
|
def agendaposters(v):
|
|
users = [x for x in g.db.query(User).filter_by(agendaposter = True).order_by(User.username).all()]
|
|
return render_template("agendaposters.html", v=v, users=users)
|
|
|
|
|
|
@app.get("/@<username>/upvoters")
|
|
@auth_required
|
|
def upvoters(v, username):
|
|
id = get_user(username).id
|
|
|
|
votes = g.db.query(Vote.user_id, func.count(Vote.user_id)).join(Submission, Vote.submission_id==Submission.id).filter(Vote.vote_type==1, Submission.author_id==id).group_by(Vote.user_id).order_by(func.count(Vote.user_id).desc()).all()
|
|
|
|
votes2 = g.db.query(CommentVote.user_id, func.count(CommentVote.user_id)).join(Comment, CommentVote.comment_id==Comment.id).filter(CommentVote.vote_type==1, Comment.author_id==id).group_by(CommentVote.user_id).order_by(func.count(CommentVote.user_id).desc()).all()
|
|
|
|
votes = Counter(dict(votes)) + Counter(dict(votes2))
|
|
|
|
users = g.db.query(User).filter(User.id.in_(votes.keys())).all()
|
|
users2 = []
|
|
for user in users: users2.append((user, votes[user.id]))
|
|
|
|
users = sorted(users2, key=lambda x: x[1], reverse=True)[:25]
|
|
|
|
return render_template("voters.html", v=v, users=users, name='Up', name2=f'@{username} biggest simps')
|
|
|
|
@app.get("/@<username>/downvoters")
|
|
@auth_required
|
|
def downvoters(v, username):
|
|
id = get_user(username).id
|
|
|
|
votes = g.db.query(Vote.user_id, func.count(Vote.user_id)).join(Submission, Vote.submission_id==Submission.id).filter(Vote.vote_type==-1, Submission.author_id==id).group_by(Vote.user_id).order_by(func.count(Vote.user_id).desc()).all()
|
|
|
|
votes2 = g.db.query(CommentVote.user_id, func.count(CommentVote.user_id)).join(Comment, CommentVote.comment_id==Comment.id).filter(CommentVote.vote_type==-1, Comment.author_id==id).group_by(CommentVote.user_id).order_by(func.count(CommentVote.user_id).desc()).all()
|
|
|
|
votes = Counter(dict(votes)) + Counter(dict(votes2))
|
|
|
|
users = g.db.query(User).filter(User.id.in_(votes.keys())).all()
|
|
users2 = []
|
|
for user in users: users2.append((user, votes[user.id]))
|
|
|
|
users = sorted(users2, key=lambda x: x[1], reverse=True)[:25]
|
|
|
|
return render_template("voters.html", v=v, users=users, name='Down', name2=f'@{username} biggest haters')
|
|
|
|
@app.get("/@<username>/upvoting")
|
|
@auth_required
|
|
def upvoting(v, username):
|
|
id = get_user(username).id
|
|
|
|
votes = g.db.query(Submission.author_id, func.count(Submission.author_id)).join(Vote, Vote.submission_id==Submission.id).filter(Vote.vote_type==1, Vote.user_id==id).group_by(Submission.author_id).order_by(func.count(Submission.author_id).desc()).all()
|
|
|
|
# votes2 = g.db.query(Comment.author_id, func.count(Comment.author_id)).join(CommentVote, CommentVote.comment_id==Comment.id).filter(CommentVote.vote_type==1, CommentVote.user_id==id).group_by(Comment.author_id).order_by(func.count(Comment.author_id).desc()).all()
|
|
|
|
votes = Counter(dict(votes))
|
|
# + Counter(dict(votes2))
|
|
|
|
users = g.db.query(User).filter(User.id.in_(votes.keys())).all()
|
|
users2 = []
|
|
for user in users: users2.append((user, votes[user.id]))
|
|
|
|
users = sorted(users2, key=lambda x: x[1], reverse=True)[:25]
|
|
|
|
return render_template("voters.html", v=v, users=users, name='Up', name2=f'Who @{username} simps for')
|
|
|
|
@app.get("/@<username>/downvoting")
|
|
@auth_required
|
|
def downvoting(v, username):
|
|
id = get_user(username).id
|
|
|
|
votes = g.db.query(Submission.author_id, func.count(Submission.author_id)).join(Vote, Vote.submission_id==Submission.id).filter(Vote.vote_type==-1, Vote.user_id==id).group_by(Submission.author_id).order_by(func.count(Submission.author_id).desc()).all()
|
|
|
|
# votes2 = g.db.query(Comment.author_id, func.count(Comment.author_id)).join(CommentVote, CommentVote.comment_id==Comment.id).filter(CommentVote.vote_type==-1, CommentVote.user_id==id).group_by(Comment.author_id).order_by(func.count(Comment.author_id).desc()).all()
|
|
|
|
votes = Counter(dict(votes))
|
|
# + Counter(dict(votes2))
|
|
|
|
users = g.db.query(User).filter(User.id.in_(votes.keys())).all()
|
|
users2 = []
|
|
for user in users: users2.append((user, votes[user.id]))
|
|
|
|
users = sorted(users2, key=lambda x: x[1], reverse=True)[:25]
|
|
|
|
return render_template("voters.html", v=v, users=users, name='Down', name2=f'Who @{username} hates')
|
|
|
|
@app.post("/pay_rent")
|
|
@limiter.limit("1/second;30/minute;200/hour;1000/day")
|
|
@auth_required
|
|
def pay_rent(v):
|
|
if v.coins < 500: return {"error":"You must have more than 500 coins."}
|
|
v.coins -= 500
|
|
v.rent_utc = int(time.time())
|
|
g.db.add(v)
|
|
u = get_account(253)
|
|
u.coins += 500
|
|
g.db.add(u)
|
|
send_repeatable_notification(u.id, f"@{v.username} has paid rent!")
|
|
g.db.commit()
|
|
return {"message": "Rent paid!"}
|
|
|
|
|
|
@app.post("/steal")
|
|
@limiter.limit("1/second;30/minute;200/hour;1000/day")
|
|
@auth_required
|
|
def steal(v):
|
|
if int(time.time()) - v.created_utc < 604800:
|
|
return {"error":"You must have an account older than 1 week in order to attempt stealing."}
|
|
if v.coins < 5000:
|
|
return {"error":"You must have more than 5000 coins in order to attempt stealing."}
|
|
u = get_account(253)
|
|
if random.randint(1, 10) < 5:
|
|
v.coins += 700
|
|
v.steal_utc = int(time.time())
|
|
g.db.add(v)
|
|
u.coins -= 700
|
|
g.db.add(u)
|
|
send_repeatable_notification(u.id, f"Some [grubby little rentoid](/@{v.username}) has absconded with 700 of your hard-earned coins to fuel his Funko Pop addiction. Stop being so trusting.")
|
|
send_repeatable_notification(v.id, "You have successfully shorted your heroic landlord 700 coins in rent. You're slightly less materially poor, but somehow even moreso morally. Are you proud of yourself?")
|
|
g.db.commit()
|
|
return {"message": "Attempt successful!"}
|
|
|
|
else:
|
|
if random.random() < 0.15:
|
|
send_repeatable_notification(u.id, f"You caught [this sniveling little renthog](/@{v.username}) trying to rob you. After beating him within an inch of his life, you sold his Nintendo Switch for 500 coins and called the cops. He was sentenced to one (1) day in renthog prison.")
|
|
send_repeatable_notification(v.id, "The ever-vigilant landchad has caught you trying to steal his hard-earned rent money. The police take you away and laugh as you impotently stutter A-ACAB :sob: You are fined 500 coins and sentenced to one (1) day in renthog prison.")
|
|
v.ban(days=1, reason="Jailed thief")
|
|
v.fail_utc = int(time.time())
|
|
else:
|
|
send_repeatable_notification(u.id, f"You caught [this sniveling little renthog](/@{v.username}) trying to rob you. After beating him within an inch of his life, you showed mercy in exchange for a 500 dramacoin tip. This time.")
|
|
send_repeatable_notification(v.id, "The ever-vigilant landchad has caught you trying to steal his hard-earned rent money. You were able to convince him to spare your life with a 500 dramacoin tip. This time.")
|
|
v.fail2_utc = int(time.time())
|
|
v.coins -= 500
|
|
g.db.add(v)
|
|
u.coins += 500
|
|
g.db.add(u)
|
|
g.db.commit()
|
|
return {"message": "Attempt failed!"}
|
|
|
|
|
|
@app.get("/rentoids")
|
|
@auth_required
|
|
def rentoids(v):
|
|
users = g.db.query(User).filter(User.rent_utc > 0).all()
|
|
return render_template("rentoids.html", v=v, users=users)
|
|
|
|
|
|
@app.get("/thiefs")
|
|
@auth_required
|
|
def thiefs(v):
|
|
successful = g.db.query(User).filter(User.steal_utc > 0).all()
|
|
failed = g.db.query(User).filter(User.fail_utc > 0).all()
|
|
failed2 = g.db.query(User).filter(User.fail2_utc > 0).all()
|
|
return render_template("thiefs.html", v=v, successful=successful, failed=failed, failed2=failed2)
|
|
|
|
|
|
@app.post("/@<username>/suicide")
|
|
@limiter.limit("1/second;30/minute;200/hour;1000/day")
|
|
@auth_required
|
|
def suicide(v, username):
|
|
t = int(time.time())
|
|
if v.admin_level == 0 and t - v.suicide_utc < 86400: return {"message": "You're on 1-day cooldown!"}
|
|
user = get_user(username)
|
|
suicide = f"Hi there,\n\nA [concerned user](/id/{v.id}) reached out to us about you.\n\nWhen you're in the middle of something painful, it may feel like you don't have a lot of options. But whatever you're going through, you deserve help and there are people who are here for you.\n\nThere are resources available in your area that are free, confidential, and available 24/7:\n\n- Call, Text, or Chat with Canada's [Crisis Services Canada](https://www.crisisservicescanada.ca/en/)\n- Call, Email, or Visit the UK's [Samaritans](https://www.samaritans.org/)\n- Text CHAT to America's [Crisis Text Line](https://www.crisistextline.org/) at 741741.\nIf you don't see a resource in your area above, the moderators keep a comprehensive list of resources and hotlines for people organized by location. Find Someone Now\n\nIf you think you may be depressed or struggling in another way, don't ignore it or brush it aside. Take yourself and your feelings seriously, and reach out to someone.\n\nIt may not feel like it, but you have options. There are people available to listen to you, and ways to move forward.\n\nYour fellow users care about you and there are people who want to help."
|
|
send_repeatable_notification(user.id, suicide)
|
|
v.suicide_utc = t
|
|
g.db.add(v)
|
|
g.db.commit()
|
|
return {"message": "Help message sent!"}
|
|
|
|
|
|
@app.get("/@<username>/coins")
|
|
@auth_required
|
|
def get_coins(v, username):
|
|
user = get_user(username)
|
|
if user != None: return {"coins": user.coins}, 200
|
|
else: return {"error": "invalid_user"}, 404
|
|
|
|
@app.post("/@<username>/transfer_coins")
|
|
@limiter.limit("1/second;30/minute;200/hour;1000/day")
|
|
@is_not_permabanned
|
|
def transfer_coins(v, username):
|
|
receiver = g.db.query(User).filter_by(username=username).one_or_none()
|
|
|
|
if receiver is None: return {"error": "That user doesn't exist."}, 404
|
|
|
|
if receiver.id != v.id:
|
|
amount = request.values.get("amount", "").strip()
|
|
amount = int(amount) if amount.isdigit() else None
|
|
|
|
if amount is None or amount <= 0: return {"error": f"Invalid amount of {app.config['COINS_NAME']}."}, 400
|
|
if v.coins < amount: return {"error": f"You don't have enough {app.config['COINS_NAME']}"}, 400
|
|
if amount < 100: return {"error": f"You have to gift at least 100 {app.config['COINS_NAME']}."}, 400
|
|
|
|
if not v.patron and not receiver.patron and not v.alts_patron and not receiver.alts_patron:
|
|
tax = math.ceil(amount*0.03)
|
|
tax_receiver = g.db.query(User).filter_by(id=TAX_RECEIVER_ID).one_or_none()
|
|
tax_receiver.coins += tax
|
|
g.db.add(tax_receiver)
|
|
else: tax = 0
|
|
|
|
if TAX_RECEIVER_ID:
|
|
log_message = f"@{v.username} has transferred {amount} {app.config['COINS_NAME']} to @{receiver.username}"
|
|
send_repeatable_notification(TAX_RECEIVER_ID, log_message)
|
|
|
|
receiver.coins += amount-tax
|
|
v.coins -= amount
|
|
send_repeatable_notification(receiver.id, f":marseycapitalistmanlet: @{v.username} has gifted you {amount-tax} {app.config['COINS_NAME']}!")
|
|
g.db.add(receiver)
|
|
g.db.add(v)
|
|
|
|
g.db.commit()
|
|
return {"message": f"{amount-tax} {app.config['COINS_NAME']} transferred!"}, 200
|
|
|
|
return {"message": f"You can't transfer {app.config['COINS_NAME']} to yourself!"}, 400
|
|
|
|
|
|
@app.post("/@<username>/transfer_bux")
|
|
@limiter.limit("1/second;30/minute;200/hour;1000/day")
|
|
@is_not_permabanned
|
|
def transfer_bux(v, username):
|
|
receiver = g.db.query(User).filter_by(username=username).one_or_none()
|
|
|
|
if not receiver: return {"error": "That user doesn't exist."}, 404
|
|
|
|
if receiver.id != v.id:
|
|
amount = request.values.get("amount", "").strip()
|
|
amount = int(amount) if amount.isdigit() else None
|
|
|
|
if not amount or amount < 0: return {"error": "Invalid amount of marseybux."}, 400
|
|
if v.procoins < amount: return {"error": "You don't have enough marseybux"}, 400
|
|
if amount < 100: return {"error": "You have to gift at least 100 marseybux."}, 400
|
|
|
|
if TAX_RECEIVER_ID:
|
|
log_message = f"@{v.username} has transferred {amount} Marseybux to @{receiver.username}"
|
|
send_repeatable_notification(TAX_RECEIVER_ID, log_message)
|
|
|
|
receiver.procoins += amount
|
|
v.procoins -= amount
|
|
send_repeatable_notification(receiver.id, f":marseycapitalistmanlet: @{v.username} has gifted you {amount} marseybux!")
|
|
g.db.add(receiver)
|
|
g.db.add(v)
|
|
|
|
g.db.commit()
|
|
return {"message": f"{amount} marseybux transferred!"}, 200
|
|
|
|
return {"message": "You can't transfer marseybux to yourself!"}, 400
|
|
|
|
|
|
@app.get("/leaderboard")
|
|
@auth_required
|
|
def leaderboard(v):
|
|
users = g.db.query(User)
|
|
|
|
users1 = users.order_by(User.coins.desc()).all()
|
|
pos1 = users1.index(v)
|
|
users1 = users1[:25]
|
|
|
|
users2 = users.order_by(User.stored_subscriber_count.desc()).all()
|
|
pos2 = users2.index(v)
|
|
users2 = users2[:25]
|
|
|
|
users3 = users.order_by(User.post_count.desc()).all()
|
|
pos3 = users3.index(v)
|
|
users3 = users3[:25]
|
|
|
|
users4 = users.order_by(User.comment_count.desc()).all()
|
|
pos4 = users4.index(v)
|
|
users4 = users4[:25]
|
|
|
|
users5 = users.order_by(User.received_award_count.desc()).limit(25).all()
|
|
|
|
if request.host == 'pcmemes.net': users6 = users.order_by(User.basedcount.desc()).limit(10).all()
|
|
else: users6 = None
|
|
|
|
users7 = users.order_by(User.coins_spent.desc()).all()
|
|
pos7 = users7.index(v)
|
|
users7 = users7[:25]
|
|
|
|
votes1 = g.db.query(Submission.author_id, func.count(Submission.author_id)).join(Vote, Vote.submission_id==Submission.id).filter(Vote.vote_type==-1).group_by(Submission.author_id).order_by(func.count(Submission.author_id).desc()).all()
|
|
votes2 = g.db.query(Comment.author_id, func.count(Comment.author_id)).join(CommentVote, CommentVote.comment_id==Comment.id).filter(CommentVote.vote_type==-1).group_by(Comment.author_id).order_by(func.count(Comment.author_id).desc()).all()
|
|
votes3 = Counter(dict(votes1)) + Counter(dict(votes2))
|
|
users8 = g.db.query(User).filter(User.id.in_(votes3.keys())).all()
|
|
users9 = []
|
|
for user in users8: users9.append((user, votes3[user.id]))
|
|
users9 = sorted(users9, key=lambda x: x[1], reverse=True)[:25]
|
|
|
|
users10 = g.db.query(User).order_by(User.truecoins.desc()).all()
|
|
pos10 = users10.index(v)
|
|
users10 = users10[:25]
|
|
|
|
badges = g.db.query(Badge.user_id, func.count(Badge.user_id)).group_by(Badge.user_id).order_by(func.count(Badge.user_id).desc()).all()
|
|
badges = dict(badges)
|
|
users11 = g.db.query(User).filter(User.id.in_(badges.keys())).all()
|
|
users12 = []
|
|
for user in users11: users12.append((user, badges[user.id]))
|
|
users12 = sorted(users12, key=lambda x: x[1], reverse=True)[:25]
|
|
|
|
if request.host == 'rdrama.net': users13 = topmakers
|
|
else: users13 = None
|
|
|
|
return render_template("leaderboard.html", v=v, users1=users1, pos1=pos1, users2=users2, pos2=pos2, users3=users3, pos3=pos3, users4=users4, pos4=pos4, users5=users5, users6=users6, users7=users7, pos7=pos7, users9=users9, users10=users10, pos10=pos10, users12=users12, users13=users13, users15=users15)
|
|
|
|
|
|
@app.get("/@<username>/css")
|
|
@auth_required
|
|
def get_css(v, username):
|
|
user = get_user(username)
|
|
if user.css: css = user.css
|
|
else: css = ""
|
|
resp=make_response(css)
|
|
resp.headers.add("Content-Type", "text/css")
|
|
return resp
|
|
|
|
@app.get("/@<username>/profilecss")
|
|
@auth_required
|
|
def get_profilecss(v, username):
|
|
user = get_user(username)
|
|
if user.profilecss: profilecss = user.profilecss
|
|
else: profilecss = ""
|
|
resp=make_response(profilecss)
|
|
resp.headers.add("Content-Type", "text/css")
|
|
return resp
|
|
|
|
@app.get("/songs/<id>")
|
|
@auth_required
|
|
def songs(v, id):
|
|
try: id = int(id)
|
|
except: return "", 400
|
|
user = g.db.query(User).filter_by(id=id).one_or_none()
|
|
if user and user.song: return redirect(f"/static/song/{user.song}.mp3")
|
|
else: abort(404)
|
|
|
|
@app.get("/song/<song>")
|
|
@app.get("/static/song/<song>")
|
|
def song(song):
|
|
if request.path.startswith('/song/'): return redirect(request.full_path.replace('/song/', '/static/song/'))
|
|
resp = make_response(send_from_directory('/songs', song))
|
|
resp.headers.remove("Cache-Control")
|
|
resp.headers.add("Cache-Control", "public, max-age=2628000")
|
|
return resp
|
|
|
|
@app.post("/subscribe/<post_id>")
|
|
@limiter.limit("1/second;30/minute;200/hour;1000/day")
|
|
@auth_required
|
|
def subscribe(v, post_id):
|
|
new_sub = Subscription(user_id=v.id, submission_id=post_id)
|
|
g.db.add(new_sub)
|
|
g.db.commit()
|
|
return {"message": "Post subscribed!"}
|
|
|
|
@app.post("/unsubscribe/<post_id>")
|
|
@limiter.limit("1/second;30/minute;200/hour;1000/day")
|
|
@auth_required
|
|
def unsubscribe(v, post_id):
|
|
sub=g.db.query(Subscription).filter_by(user_id=v.id, submission_id=post_id).one_or_none()
|
|
if sub:
|
|
g.db.delete(sub)
|
|
g.db.commit()
|
|
return {"message": "Post unsubscribed!"}
|
|
|
|
@app.get("/report_bugs")
|
|
@auth_required
|
|
def reportbugs(v):
|
|
return redirect(f'/post/{BUG_THREAD}')
|
|
|
|
@app.post("/@<username>/message")
|
|
@limiter.limit("1/second;2/minute;10/hour;50/day")
|
|
@is_not_permabanned
|
|
def message2(v, username):
|
|
|
|
user = get_user(username, v=v)
|
|
if hasattr(user, 'is_blocking') and user.is_blocking: return {"error": "You're blocking this user."}, 403
|
|
|
|
if v.admin_level <= 1 and hasattr(user, 'is_blocked') and user.is_blocked:
|
|
return {"error": "This user is blocking you."}, 403
|
|
|
|
if v.shadowbanned: return {"message": "Message sent!"}
|
|
|
|
message = request.values.get("message", "").strip()[:1000].strip()
|
|
|
|
if not message: return {"error": "message is empty"}
|
|
|
|
if 'linkedin.com' in message: return {"error": "This domain 'linkedin.com' is banned."}, 403
|
|
|
|
message = re.sub('!\[\]\((.*?)\)', r'\1', message)
|
|
|
|
text_html = sanitize(message, noimages=True)
|
|
|
|
existing = g.db.query(Comment.id).filter(Comment.author_id == v.id,
|
|
Comment.sentto == user.id,
|
|
Comment.body_html == text_html,
|
|
).first()
|
|
|
|
if existing: return {"error": "Message already exists."}, 403
|
|
|
|
new_comment = Comment(author_id=v.id,
|
|
parent_submission=None,
|
|
level=1,
|
|
sentto=user.id,
|
|
body_html=text_html,
|
|
)
|
|
g.db.add(new_comment)
|
|
|
|
g.db.flush()
|
|
|
|
|
|
notif = Notification(comment_id=new_comment.id, user_id=user.id)
|
|
g.db.add(notif)
|
|
|
|
if PUSHER_ID:
|
|
if len(message) > 500: notifbody = message[:500] + '...'
|
|
else: notifbody = message
|
|
|
|
beams_client.publish_to_interests(
|
|
interests=[f'{request.host}{user.id}'],
|
|
publish_body={
|
|
'web': {
|
|
'notification': {
|
|
'title': f'New message from @{v.username}',
|
|
'body': notifbody,
|
|
'deep_link': f'https://{site}/notifications?messages=true',
|
|
'icon': f'https://{request.host}/assets/images/{SITE_NAME}/icon.webp',
|
|
}
|
|
},
|
|
'fcm': {
|
|
'notification': {
|
|
'title': f'New message from @{v.username}',
|
|
'body': notifbody,
|
|
},
|
|
'data': {
|
|
'url': 'notifications?messages=true',
|
|
}
|
|
}
|
|
},
|
|
)
|
|
|
|
g.db.commit()
|
|
|
|
return {"message": "Message sent!"}
|
|
|
|
|
|
@app.post("/reply")
|
|
@limiter.limit("1/second;6/minute;50/hour;200/day")
|
|
@auth_required
|
|
def messagereply(v):
|
|
|
|
message = request.values.get("body", "").strip()[:1000].strip()
|
|
|
|
if not message: return {"error": "message is empty"}
|
|
|
|
if 'linkedin.com' in message: return {"error": "this domain 'linkedin.com' is banned"}
|
|
|
|
message = re.sub('!\[\]\((.*?)\)', r'\1', message)
|
|
|
|
id = int(request.values.get("parent_id"))
|
|
parent = get_comment(id, v=v)
|
|
user_id = parent.author.id
|
|
|
|
if v.id == user_id: user_id = parent.sentto
|
|
|
|
text_html = sanitize(message, noimages=True)
|
|
|
|
new_comment = Comment(author_id=v.id,
|
|
parent_submission=None,
|
|
parent_comment_id=id,
|
|
top_comment_id=parent.top_comment_id if parent.top_comment_id else parent.id,
|
|
level=parent.level + 1,
|
|
sentto=user_id,
|
|
body_html=text_html,
|
|
)
|
|
g.db.add(new_comment)
|
|
g.db.flush()
|
|
|
|
if PUSHER_ID and user_id != v.id:
|
|
notif = Notification(comment_id=new_comment.id, user_id=user_id)
|
|
g.db.add(notif)
|
|
|
|
if len(message) > 500: notifbody = message[:500] + '...'
|
|
else: notifbody = message
|
|
|
|
beams_client.publish_to_interests(
|
|
interests=[f'{request.host}{user_id}'],
|
|
publish_body={
|
|
'web': {
|
|
'notification': {
|
|
'title': f'New message from @{v.username}',
|
|
'body': notifbody,
|
|
'deep_link': f'https://{site}/notifications?messages=true',
|
|
'icon': f'https://{request.host}/assets/images/{SITE_NAME}/icon.webp',
|
|
}
|
|
},
|
|
'fcm': {
|
|
'notification': {
|
|
'title': f'New message from @{v.username}',
|
|
'body': notifbody,
|
|
},
|
|
'data': {
|
|
'url': 'notifications?messages=true',
|
|
}
|
|
}
|
|
},
|
|
)
|
|
|
|
|
|
if new_comment.top_comment.sentto == 0:
|
|
admins = g.db.query(User).filter(User.admin_level > 2, User.id != v.id, User.id != user_id).all()
|
|
for admin in admins:
|
|
notif = Notification(comment_id=new_comment.id, user_id=admin.id)
|
|
g.db.add(notif)
|
|
g.db.commit()
|
|
|
|
return render_template("comments.html", v=v, comments=[new_comment])
|
|
|
|
@app.get("/2faqr/<secret>")
|
|
@auth_required
|
|
def mfa_qr(secret, v):
|
|
x = pyotp.TOTP(secret)
|
|
qr = qrcode.QRCode(
|
|
error_correction=qrcode.constants.ERROR_CORRECT_L
|
|
)
|
|
qr.add_data(x.provisioning_uri(v.username, issuer_name=app.config["SITE_NAME"]))
|
|
img = qr.make_image(fill_color="#000000", back_color="white")
|
|
|
|
mem = io.BytesIO()
|
|
|
|
img.save(mem, format="PNG")
|
|
mem.seek(0, 0)
|
|
return send_file(mem, mimetype="image/png", as_attachment=False)
|
|
|
|
|
|
@app.get("/is_available/<name>")
|
|
@auth_required
|
|
def api_is_available(name, v):
|
|
|
|
name=name.strip()
|
|
|
|
if len(name)<3 or len(name)>25:
|
|
return {name:False}
|
|
|
|
name2 = name.replace('_','\_')
|
|
|
|
x= g.db.query(User).filter(
|
|
or_(
|
|
User.username.ilike(name2),
|
|
User.original_username.ilike(name2)
|
|
)
|
|
).one_or_none()
|
|
|
|
if x:
|
|
return {name: False}
|
|
else:
|
|
return {name: True}
|
|
|
|
@app.get("/id/<id>")
|
|
@auth_required
|
|
def user_id(id, v):
|
|
try: id = int(id)
|
|
except: abort(404)
|
|
user = get_account(id)
|
|
return redirect(user.url)
|
|
|
|
@app.get("/u/<username>")
|
|
@auth_required
|
|
def redditor_moment_redirect(username, v):
|
|
return redirect(f"/@{username}")
|
|
|
|
@app.get("/@<username>/followers")
|
|
@auth_required
|
|
def followers(username, v):
|
|
u = get_user(username, v=v)
|
|
users = g.db.query(User).join(Follow, Follow.target_id == u.id).filter(Follow.user_id == User.id).order_by(Follow.id).all()
|
|
return render_template("followers.html", v=v, u=u, users=users)
|
|
|
|
@app.get("/@<username>/following")
|
|
@auth_required
|
|
def following(username, v):
|
|
u = get_user(username, v=v)
|
|
users = g.db.query(User).join(Follow, Follow.user_id == u.id).filter(Follow.target_id == User.id).order_by(Follow.id).all()
|
|
return render_template("following.html", v=v, u=u, users=users)
|
|
|
|
@app.get("/views")
|
|
@auth_required
|
|
def visitors(v):
|
|
if request.host == 'rdrama.net' and v.admin_level < 1 and not v.patron: return render_template("errors/patron.html", v=v)
|
|
viewers=sorted(v.viewers, key = lambda x: x.last_view_utc, reverse=True)
|
|
return render_template("viewers.html", v=v, viewers=viewers)
|
|
|
|
|
|
@app.get("/@<username>")
|
|
@app.get("/logged_out/@<username>")
|
|
@auth_desired
|
|
def u_username(username, v=None):
|
|
|
|
|
|
if not v and not request.path.startswith('/logged_out'): return redirect(f"/logged_out{request.full_path}")
|
|
|
|
if v and request.path.startswith('/logged_out'): v = None
|
|
|
|
|
|
|
|
u = get_user(username, v=v)
|
|
|
|
|
|
if username != u.username:
|
|
return redirect(request.full_path.replace(username, u.username))
|
|
|
|
if u.reserved:
|
|
if request.headers.get("Authorization") or request.headers.get("xhr"): return {"error": f"That username is reserved for: {u.reserved}"}
|
|
return render_template("userpage_reserved.html", u=u, v=v)
|
|
|
|
if v and u.id != v.id:
|
|
view = g.db.query(ViewerRelationship).filter(
|
|
and_(
|
|
ViewerRelationship.viewer_id == v.id,
|
|
ViewerRelationship.user_id == u.id
|
|
)
|
|
).first()
|
|
|
|
if view:
|
|
view.last_view_utc = g.timestamp
|
|
else:
|
|
view = ViewerRelationship(user_id = u.id,
|
|
viewer_id = v.id)
|
|
|
|
g.db.add(view)
|
|
g.db.commit()
|
|
|
|
|
|
if u.is_private and (not v or (v.id != u.id and v.admin_level < 2 and not v.eye)):
|
|
|
|
if v and u.id == LLM_ID:
|
|
if int(time.time()) - v.rent_utc > 600:
|
|
if request.headers.get("Authorization") or request.headers.get("xhr"): return {"error": "That userpage is private"}
|
|
return render_template("userpage_private.html", time=int(time.time()), u=u, v=v)
|
|
else:
|
|
if request.headers.get("Authorization") or request.headers.get("xhr"): return {"error": "That userpage is private"}
|
|
return render_template("userpage_private.html", time=int(time.time()), u=u, v=v)
|
|
|
|
|
|
if v and hasattr(u, 'is_blocking') and u.is_blocking:
|
|
if request.headers.get("Authorization") or request.headers.get("xhr"): return {"error": f"You are blocking @{u.username}."}
|
|
return render_template("userpage_blocking.html", u=u, v=v)
|
|
|
|
|
|
if v and v.admin_level < 2 and hasattr(u, 'is_blocked') and u.is_blocked:
|
|
if request.headers.get("Authorization") or request.headers.get("xhr"): return {"error": "This person is blocking you."}
|
|
return render_template("userpage_blocked.html", u=u, v=v)
|
|
|
|
|
|
sort = request.values.get("sort", "new")
|
|
t = request.values.get("t", "all")
|
|
page = int(request.values.get("page", "1"))
|
|
page = max(page, 1)
|
|
|
|
ids = u.userpagelisting(v=v, page=page, sort=sort, t=t)
|
|
|
|
next_exists = (len(ids) > 25)
|
|
ids = ids[:25]
|
|
|
|
if page == 1:
|
|
sticky = []
|
|
sticky = g.db.query(Submission).filter_by(is_pinned=True, author_id=u.id).all()
|
|
if sticky:
|
|
for p in sticky:
|
|
ids = [p.id] + ids
|
|
|
|
listing = get_posts(ids, v=v)
|
|
|
|
if u.unban_utc:
|
|
if request.headers.get("Authorization"): {"data": [x.json for x in listing]}
|
|
return render_template("userpage.html",
|
|
unban=u.unban_string,
|
|
u=u,
|
|
v=v,
|
|
listing=listing,
|
|
page=page,
|
|
sort=sort,
|
|
t=t,
|
|
next_exists=next_exists,
|
|
is_following=(v and u.has_follower(v)))
|
|
|
|
|
|
|
|
if request.headers.get("Authorization"): return {"data": [x.json for x in listing]}
|
|
return render_template("userpage.html",
|
|
u=u,
|
|
v=v,
|
|
listing=listing,
|
|
page=page,
|
|
sort=sort,
|
|
t=t,
|
|
next_exists=next_exists,
|
|
is_following=(v and u.has_follower(v)))
|
|
|
|
|
|
@app.get("/@<username>/comments")
|
|
@app.get("/logged_out/@<username>/comments")
|
|
@auth_desired
|
|
def u_username_comments(username, v=None):
|
|
|
|
|
|
if not v and not request.path.startswith('/logged_out'): return redirect(f"/logged_out{request.full_path}")
|
|
|
|
if v and request.path.startswith('/logged_out'): v = None
|
|
|
|
|
|
|
|
user = get_user(username, v=v)
|
|
|
|
|
|
if username != user.username: return redirect(f'/id/{user.id}/comments')
|
|
|
|
u = user
|
|
|
|
if u.reserved:
|
|
if request.headers.get("Authorization") or request.headers.get("xhr"): return {"error": f"That username is reserved for: {u.reserved}"}
|
|
return render_template("userpage_reserved.html",
|
|
u=u,
|
|
v=v)
|
|
|
|
|
|
if u.is_private and (not v or (v.id != u.id and v.admin_level < 2 and not v.eye)):
|
|
if v and u.id == LLM_ID:
|
|
if int(time.time()) - v.rent_utc > 600:
|
|
if request.headers.get("Authorization") or request.headers.get("xhr"): return {"error": "That userpage is private"}
|
|
return render_template("userpage_private.html", time=int(time.time()), u=u, v=v)
|
|
else:
|
|
if request.headers.get("Authorization") or request.headers.get("xhr"): return {"error": "That userpage is private"}
|
|
return render_template("userpage_private.html", time=int(time.time()), u=u, v=v)
|
|
|
|
if v and hasattr(u, 'is_blocking') and u.is_blocking:
|
|
if request.headers.get("Authorization") or request.headers.get("xhr"): return {"error": f"You are blocking @{u.username}."}
|
|
return render_template("userpage_blocking.html", u=u, v=v)
|
|
|
|
if v and v.admin_level < 2 and hasattr(u, 'is_blocked') and u.is_blocked:
|
|
if request.headers.get("Authorization") or request.headers.get("xhr"): return {"error": "This person is blocking you."}
|
|
return render_template("userpage_blocked.html", u=u, v=v)
|
|
|
|
|
|
page = max(int(request.values.get("page", "1")), 1)
|
|
sort=request.values.get("sort","new")
|
|
t=request.values.get("t","all")
|
|
|
|
|
|
comments = g.db.query(Comment.id).filter(Comment.author_id == u.id, Comment.parent_submission != None)
|
|
|
|
if (not v) or (v.id != u.id and v.admin_level == 0):
|
|
comments = comments.filter(Comment.deleted_utc == 0)
|
|
comments = comments.filter(Comment.is_banned == False)
|
|
|
|
now = int(time.time())
|
|
if t == 'hour':
|
|
cutoff = now - 3600
|
|
elif t == 'day':
|
|
cutoff = now - 86400
|
|
elif t == 'week':
|
|
cutoff = now - 604800
|
|
elif t == 'month':
|
|
cutoff = now - 2592000
|
|
elif t == 'year':
|
|
cutoff = now - 31536000
|
|
else:
|
|
cutoff = 0
|
|
comments = comments.filter(Comment.created_utc >= cutoff)
|
|
|
|
if sort == "new":
|
|
comments = comments.order_by(Comment.created_utc.desc())
|
|
elif sort == "old":
|
|
comments = comments.order_by(Comment.created_utc.asc())
|
|
elif sort == "controversial":
|
|
comments = comments.order_by(-1 * Comment.upvotes * Comment.downvotes * Comment.downvotes)
|
|
elif sort == "top":
|
|
comments = comments.order_by(Comment.downvotes - Comment.upvotes)
|
|
elif sort == "bottom":
|
|
comments = comments.order_by(Comment.upvotes - Comment.downvotes)
|
|
|
|
comments = comments.offset(25 * (page - 1)).limit(26).all()
|
|
ids = [x.id for x in comments]
|
|
|
|
next_exists = (len(ids) > 25)
|
|
ids = ids[:25]
|
|
|
|
listing = get_comments(ids, v=v)
|
|
|
|
is_following = (v and user.has_follower(v))
|
|
|
|
if request.headers.get("Authorization"): return {"data": [c.json for c in listing]}
|
|
return render_template("userpage_comments.html", u=user, v=v, listing=listing, page=page, sort=sort, t=t,next_exists=next_exists, is_following=is_following, standalone=True)
|
|
|
|
|
|
@app.get("/@<username>/info")
|
|
@auth_required
|
|
def u_username_info(username, v=None):
|
|
|
|
user=get_user(username, v=v)
|
|
|
|
if hasattr(user, 'is_blocking') and user.is_blocking:
|
|
return {"error": "You're blocking this user."}, 401
|
|
elif hasattr(user, 'is_blocked') and user.is_blocked:
|
|
return {"error": "This user is blocking you."}, 403
|
|
|
|
return user.json
|
|
|
|
|
|
@app.post("/follow/<username>")
|
|
@limiter.limit("1/second;30/minute;200/hour;1000/day")
|
|
@auth_required
|
|
def follow_user(username, v):
|
|
|
|
target = get_user(username)
|
|
|
|
if target.id==v.id: return {"error": "You can't follow yourself!"}, 400
|
|
|
|
if g.db.query(Follow).filter_by(user_id=v.id, target_id=target.id).one_or_none(): return {"message": "User followed!"}
|
|
|
|
new_follow = Follow(user_id=v.id, target_id=target.id)
|
|
g.db.add(new_follow)
|
|
|
|
g.db.flush()
|
|
target.stored_subscriber_count = g.db.query(Follow.id).filter_by(target_id=target.id).count()
|
|
g.db.add(target)
|
|
|
|
send_notification(target.id, f"@{v.username} has followed you!")
|
|
|
|
g.db.commit()
|
|
|
|
return {"message": "User followed!"}
|
|
|
|
@app.post("/unfollow/<username>")
|
|
@limiter.limit("1/second;30/minute;200/hour;1000/day")
|
|
@auth_required
|
|
def unfollow_user(username, v):
|
|
|
|
target = get_user(username)
|
|
|
|
if target.fish: return {"error": "You can't unfollow this user!"}
|
|
|
|
follow = g.db.query(Follow).filter_by(user_id=v.id, target_id=target.id).one_or_none()
|
|
|
|
if follow:
|
|
g.db.delete(follow)
|
|
|
|
g.db.flush()
|
|
target.stored_subscriber_count = g.db.query(Follow.id).filter_by(target_id=target.id).count()
|
|
g.db.add(target)
|
|
|
|
send_notification(target.id, f"@{v.username} has unfollowed you!")
|
|
|
|
g.db.commit()
|
|
|
|
return {"message": "User unfollowed!"}
|
|
|
|
@app.post("/remove_follow/<username>")
|
|
@limiter.limit("1/second;30/minute;200/hour;1000/day")
|
|
@auth_required
|
|
def remove_follow(username, v):
|
|
target = get_user(username)
|
|
|
|
follow = g.db.query(Follow).filter_by(user_id=target.id, target_id=v.id).one_or_none()
|
|
|
|
if not follow: return {"message": "Follower removed!"}
|
|
|
|
g.db.delete(follow)
|
|
|
|
g.db.flush()
|
|
v.stored_subscriber_count = g.db.query(Follow.id).filter_by(target_id=v.id).count()
|
|
g.db.add(v)
|
|
|
|
send_repeatable_notification(target.id, f"@{v.username} has removed your follow!")
|
|
|
|
g.db.commit()
|
|
|
|
return {"message": "Follower removed!"}
|
|
|
|
@app.get("/uid/<id>/pic")
|
|
@app.get("/uid/<id>/pic/profile")
|
|
@app.get("/logged_out/uid/<id>/pic")
|
|
@limiter.exempt
|
|
@auth_desired
|
|
def user_profile_uid(v, id):
|
|
if not v and not request.path.startswith('/logged_out'): return redirect(f"/logged_out{request.full_path}")
|
|
if v and request.path.startswith('/logged_out'): v = None
|
|
|
|
try: id = int(id)
|
|
except:
|
|
try: id = int(id, 36)
|
|
except: abort(404)
|
|
x=get_account(id)
|
|
return redirect(x.profile_url)
|
|
|
|
@app.get("/@<username>/pic")
|
|
@limiter.exempt
|
|
@auth_required
|
|
def user_profile_name(v, username):
|
|
x = get_user(username)
|
|
return redirect(x.profile_url)
|
|
|
|
@app.get("/@<username>/saved/posts")
|
|
@auth_required
|
|
def saved_posts(v, username):
|
|
|
|
page=int(request.values.get("page",1))
|
|
|
|
ids=v.saved_idlist(page=page)
|
|
|
|
next_exists=len(ids)>25
|
|
|
|
ids=ids[:25]
|
|
|
|
listing = get_posts(ids, v=v)
|
|
|
|
if request.headers.get("Authorization"): return {"data": [x.json for x in listing]}
|
|
return render_template("userpage.html",
|
|
u=v,
|
|
v=v,
|
|
listing=listing,
|
|
page=page,
|
|
next_exists=next_exists,
|
|
)
|
|
|
|
|
|
@app.get("/@<username>/saved/comments")
|
|
@auth_required
|
|
def saved_comments(v, username):
|
|
|
|
page=int(request.values.get("page",1))
|
|
|
|
firstrange = 25 * (page - 1)
|
|
secondrange = firstrange+26
|
|
|
|
ids=v.saved_comment_idlist()[firstrange:secondrange]
|
|
|
|
next_exists=len(ids) > 25
|
|
|
|
ids=ids[:25]
|
|
|
|
listing = get_comments(ids, v=v)
|
|
|
|
|
|
if request.headers.get("Authorization"): return {"data": [x.json for x in listing]}
|
|
return render_template("userpage_comments.html",
|
|
u=v,
|
|
v=v,
|
|
listing=listing,
|
|
page=page,
|
|
next_exists=next_exists,
|
|
standalone=True)
|
|
|
|
|
|
@app.post("/fp/<fp>")
|
|
@auth_required
|
|
def fp(v, fp):
|
|
if v.username != fp:
|
|
v.fp = fp
|
|
users = g.db.query(User).filter(User.fp == fp, User.id != v.id).all()
|
|
if v.email and v.is_activated:
|
|
users += g.db.query(User).filter(User.email == v.email, User.is_activated, User.id != v.id).all()
|
|
for u in users:
|
|
li = [v.id, u.id]
|
|
existing = g.db.query(Alt).filter(Alt.user1.in_(li), Alt.user2.in_(li)).first()
|
|
if existing: continue
|
|
new_alt = Alt(user1=v.id, user2=u.id)
|
|
g.db.add(new_alt)
|
|
g.db.flush()
|
|
print('\n\n' + v.username + ' + ' + u.username + '\n\n')
|
|
g.db.add(v)
|
|
g.db.commit()
|
|
return '', 204 |