1254 lines
30 KiB
Python
1254 lines
30 KiB
Python
import time
|
|
import calendar
|
|
from sqlalchemy.orm import lazyload
|
|
import imagehash
|
|
from os import remove
|
|
from os.path import exists
|
|
from PIL import Image as IMAGE
|
|
|
|
from files.helpers.wrappers import *
|
|
from files.helpers.alerts import *
|
|
from files.helpers.sanitize import *
|
|
from files.helpers.markdown import *
|
|
from files.helpers.security import *
|
|
from files.helpers.get import *
|
|
from files.helpers.images import *
|
|
from files.helpers.const import *
|
|
from files.classes import *
|
|
from flask import *
|
|
import matplotlib.pyplot as plt
|
|
from files.__main__ import app, cache, limiter
|
|
from .front import frontlist
|
|
from files.helpers.discord import add_role
|
|
|
|
IMGUR_KEY = environ.get("IMGUR_KEY", "").strip()
|
|
|
|
@app.post("/@<username>/make_admin")
|
|
@admin_level_required(6)
|
|
def make_admin(v, username):
|
|
user = get_user(username)
|
|
if not user: abort(404)
|
|
user.admin_level = 6
|
|
g.db.add(user)
|
|
return {"message": "User has been made admin!"}
|
|
|
|
|
|
@app.post("/@<username>/make_fake_admin")
|
|
@admin_level_required(6)
|
|
def make_fake_admin(v, username):
|
|
user = get_user(username)
|
|
if not user: abort(404)
|
|
user.admin_level = 1
|
|
g.db.add(user)
|
|
return {"message": "User has been made fake admin!"}
|
|
|
|
|
|
@app.post("/@<username>/remove_admin")
|
|
@admin_level_required(6)
|
|
def remove_admin(v, username):
|
|
user = get_user(username)
|
|
if not user: abort(404)
|
|
user.admin_level = 0
|
|
g.db.add(user)
|
|
return {"message": "Admin removed!"}
|
|
|
|
|
|
@app.get('/admin/rules')
|
|
@admin_level_required(6)
|
|
def get_rules(v):
|
|
|
|
try:
|
|
with open('./rules.md', 'r') as f:
|
|
rules = f.read()
|
|
except Exception:
|
|
rules = None
|
|
|
|
return render_template('admin/rules.html', v=v, rules=rules)
|
|
|
|
|
|
@app.post('/admin/rules')
|
|
@admin_level_required(6)
|
|
@validate_formkey
|
|
def post_rules(v):
|
|
|
|
text = request.form.get('rules', '')
|
|
|
|
with open('./rules.md', 'w+') as f:
|
|
f.write(text)
|
|
|
|
with open('./rules.md', 'r') as f:
|
|
rules = f.read()
|
|
|
|
return render_template('admin/rules.html', v=v, rules=rules)
|
|
|
|
|
|
@app.get("/admin/shadowbanned")
|
|
@auth_required
|
|
def shadowbanned(v):
|
|
if not (v and v.admin_level == 6): abort(404)
|
|
users = [x for x in g.db.query(User).filter_by(shadowbanned = True).all()]
|
|
return render_template("banned.html", v=v, users=users)
|
|
|
|
|
|
@app.get("/admin/agendaposters")
|
|
@auth_required
|
|
def agendaposters(v):
|
|
if not (v and v.admin_level == 6): abort(404)
|
|
users = [x for x in g.db.query(User).filter_by(agendaposter = True).all()]
|
|
return render_template("banned.html", v=v, users=users)
|
|
|
|
|
|
@app.get("/admin/image_posts")
|
|
@admin_level_required(3)
|
|
def image_posts_listing(v):
|
|
|
|
try: page = int(request.args.get('page', 1))
|
|
except: page = 1
|
|
|
|
posts = g.db.query(Submission).order_by(Submission.id.desc())
|
|
|
|
firstrange = 25 * (page - 1)
|
|
secondrange = firstrange+26
|
|
posts = [x.id for x in posts if x.is_image][firstrange:secondrange]
|
|
next_exists = (len(posts) == 26)
|
|
posts = get_posts(posts[:25], v=v)
|
|
|
|
return render_template("admin/image_posts.html", v=v, listing=posts, next_exists=next_exists, page=page, sort="new")
|
|
|
|
|
|
@app.get("/admin/flagged/posts")
|
|
@admin_level_required(3)
|
|
def flagged_posts(v):
|
|
|
|
page = max(1, int(request.args.get("page", 1)))
|
|
|
|
posts = g.db.query(Submission).filter_by(
|
|
is_approved=0,
|
|
is_banned=False
|
|
).join(Submission.flags).order_by(Submission.id.desc()).offset(25 * (page - 1)).limit(26)
|
|
|
|
listing = [p.id for p in posts]
|
|
next_exists = (len(listing) == 26)
|
|
listing = listing[:25]
|
|
|
|
listing = get_posts(listing, v=v)
|
|
|
|
return render_template("admin/flagged_posts.html",
|
|
next_exists=next_exists, listing=listing, page=page, v=v)
|
|
|
|
|
|
@app.get("/admin/flagged/comments")
|
|
@admin_level_required(3)
|
|
def flagged_comments(v):
|
|
|
|
page = max(1, int(request.args.get("page", 1)))
|
|
|
|
posts = g.db.query(Comment
|
|
).filter_by(
|
|
is_approved=0,
|
|
is_banned=False
|
|
).join(Comment.flags).order_by(Comment.id.desc()).offset(25 * (page - 1)).limit(26).all()
|
|
|
|
listing = [p.id for p in posts]
|
|
next_exists = (len(listing) == 26)
|
|
listing = listing[:25]
|
|
|
|
listing = get_comments(listing, v=v)
|
|
|
|
return render_template("admin/flagged_comments.html",
|
|
next_exists=next_exists,
|
|
listing=listing,
|
|
page=page,
|
|
v=v,
|
|
standalone=True)
|
|
|
|
@app.get("/admin")
|
|
@admin_level_required(3)
|
|
def admin_home(v):
|
|
with open('./disablesignups', 'r') as f:
|
|
x = f.read()
|
|
return render_template("admin/admin_home.html", v=v, x=x)
|
|
|
|
@app.post("/admin/monthly")
|
|
@limiter.limit("1/day")
|
|
@admin_level_required(6)
|
|
def monthly(v):
|
|
thing = g.db.query(AwardRelationship).order_by(AwardRelationship.id.desc()).first().id
|
|
_awards = []
|
|
for u in g.db.query(User).filter(User.patron > 0).all():
|
|
grant_awards = {}
|
|
|
|
if u.patron == 1:
|
|
grant_awards["shit"] = 1
|
|
grant_awards["stars"] = 1
|
|
elif u.patron == 2:
|
|
grant_awards["shit"] = 3
|
|
grant_awards["stars"] = 3
|
|
elif u.patron == 3:
|
|
grant_awards["shit"] = 5
|
|
grant_awards["stars"] = 5
|
|
grant_awards["ban"] = 1
|
|
elif u.patron == 4:
|
|
grant_awards["shit"] = 10
|
|
grant_awards["stars"] = 10
|
|
grant_awards["ban"] = 3
|
|
elif u.patron == 5 or u.patron == 8:
|
|
grant_awards["shit"] = 20
|
|
grant_awards["stars"] = 20
|
|
grant_awards["ban"] = 6
|
|
|
|
|
|
for name in grant_awards:
|
|
for count in range(grant_awards[name]):
|
|
|
|
thing += 1
|
|
|
|
_awards.append(AwardRelationship(
|
|
id=thing,
|
|
user_id=u.id,
|
|
kind=name
|
|
))
|
|
|
|
text = "You were given the following awards:\n\n"
|
|
|
|
for key, value in grant_awards.items():
|
|
text += f" - **{value}** {AWARDS[key]['title']} {'Awards' if value != 1 else 'Award'}\n"
|
|
|
|
send_notification(NOTIFICATIONS_ACCOUNT, u, text)
|
|
|
|
g.db.bulk_save_objects(_awards)
|
|
|
|
|
|
return {"message": "Monthly awards granted"}
|
|
|
|
|
|
@app.post("/admin/disablesignups")
|
|
@admin_level_required(6)
|
|
@validate_formkey
|
|
def disablesignups(v):
|
|
with open('./disablesignups', 'r+') as f:
|
|
if f.read() == "yes":
|
|
f.write("no")
|
|
return {"message": "Signups enabed!"}
|
|
else:
|
|
f.write("yes")
|
|
return {"message": "Signups disabled!"}
|
|
|
|
@app.get("/admin/badge_grant")
|
|
@admin_level_required(4)
|
|
def badge_grant_get(v):
|
|
|
|
badge_types = g.db.query(BadgeDef).all()
|
|
|
|
errors = {"already_owned": "That user already has that badge.",
|
|
"no_user": "That user doesn't exist."
|
|
}
|
|
|
|
return render_template("admin/badge_grant.html",
|
|
v=v,
|
|
badge_types=badge_types,
|
|
error=errors.get(
|
|
request.args.get("error"),
|
|
None) if request.args.get('error') else None,
|
|
msg="Badge successfully assigned" if request.args.get(
|
|
"msg") else None
|
|
)
|
|
|
|
|
|
@app.post("/admin/badge_grant")
|
|
@admin_level_required(4)
|
|
@validate_formkey
|
|
def badge_grant_post(v):
|
|
|
|
user = get_user(request.form.get("username"), graceful=True)
|
|
if not user: return redirect("/badge_grant?error=no_user")
|
|
|
|
try: badge_id = int(request.form.get("badge_id"))
|
|
except: abort(400)
|
|
|
|
if user.has_badge(badge_id):
|
|
g.db.query(Badge).filter_by(badge_id=badge_id, user_id=user.id,).delete()
|
|
return redirect("/admin/badge_grant")
|
|
|
|
new_badge = Badge(badge_id=badge_id,
|
|
user_id=user.id,
|
|
)
|
|
|
|
desc = request.form.get("description")
|
|
if desc: new_badge.description = desc
|
|
|
|
url = request.form.get("url")
|
|
if url: new_badge.url = url
|
|
|
|
g.db.add(new_badge)
|
|
g.db.flush()
|
|
|
|
text = f"""
|
|
@{v.username} has given you the following profile badge:
|
|
\n\n
|
|
\n\n{new_badge.name}
|
|
"""
|
|
send_notification(NOTIFICATIONS_ACCOUNT, user, text)
|
|
|
|
if badge_id == 16 and user.has_badge(17):
|
|
g.db.query(Badge).filter_by(badge_id=17, user_id=user.id).delete()
|
|
|
|
elif badge_id in [21,22,23,24,28]:
|
|
user.patron = int(str(badge_id)[-1])
|
|
|
|
grant_awards = {}
|
|
|
|
if badge_id == 21:
|
|
if user.discord_id: add_role(user, "1")
|
|
grant_awards["shit"] = 1
|
|
grant_awards["stars"] = 1
|
|
elif badge_id == 22:
|
|
if user.discord_id: add_role(user, "2")
|
|
grant_awards["shit"] = 3
|
|
grant_awards["stars"] = 3
|
|
elif badge_id == 23:
|
|
if user.discord_id: add_role(user, "3")
|
|
grant_awards["shit"] = 5
|
|
grant_awards["stars"] = 5
|
|
grant_awards["ban"] = 1
|
|
elif badge_id in [24, 28]:
|
|
if user.discord_id: add_role(user, "4")
|
|
grant_awards["shit"] = 10
|
|
grant_awards["stars"] = 10
|
|
grant_awards["ban"] = 3
|
|
elif badge_id == 25:
|
|
if user.discord_id: add_role(user, "5")
|
|
grant_awards["shit"] = 20
|
|
grant_awards["stars"] = 20
|
|
grant_awards["ban"] = 6
|
|
|
|
if len(grant_awards):
|
|
|
|
_awards = []
|
|
|
|
thing = g.db.query(AwardRelationship).order_by(AwardRelationship.id.desc()).first().id
|
|
|
|
for name in grant_awards:
|
|
for count in range(grant_awards[name]):
|
|
|
|
thing += 1
|
|
|
|
_awards.append(AwardRelationship(
|
|
id=thing,
|
|
user_id=user.id,
|
|
kind=name
|
|
))
|
|
|
|
g.db.bulk_save_objects(_awards)
|
|
|
|
text = "You were given the following awards:\n\n"
|
|
|
|
for key, value in grant_awards.items():
|
|
text += f" - **{value}** {AWARDS[key]['title']} {'Awards' if value != 1 else 'Award'}\n"
|
|
|
|
send_notification(NOTIFICATIONS_ACCOUNT, user, text)
|
|
|
|
|
|
g.db.add(user)
|
|
|
|
return redirect("/admin/badge_grant")
|
|
|
|
|
|
@app.get("/admin/users")
|
|
@admin_level_required(2)
|
|
def users_list(v):
|
|
|
|
page = int(request.args.get("page", 1))
|
|
|
|
users = g.db.query(User).filter_by(is_banned=0
|
|
).order_by(User.created_utc.desc()
|
|
).offset(25 * (page - 1)).limit(26)
|
|
|
|
users = [x for x in users]
|
|
|
|
next_exists = (len(users) == 26)
|
|
users = users[:25]
|
|
|
|
return render_template("admin/new_users.html",
|
|
v=v,
|
|
users=users,
|
|
next_exists=next_exists,
|
|
page=page,
|
|
)
|
|
|
|
@app.get("/admin/alt_votes")
|
|
@admin_level_required(4)
|
|
def alt_votes_get(v):
|
|
|
|
if not request.args.get("u1") or not request.args.get("u2"):
|
|
return render_template("admin/alt_votes.html", v=v)
|
|
|
|
u1 = request.args.get("u1")
|
|
u2 = request.args.get("u2")
|
|
|
|
if not u1 or not u2:
|
|
return redirect("/admin/alt_votes")
|
|
|
|
u1 = get_user(u1)
|
|
u2 = get_user(u2)
|
|
|
|
u1_post_ups = g.db.query(
|
|
Vote.submission_id).filter_by(
|
|
user_id=u1.id,
|
|
vote_type=1).all()
|
|
u1_post_downs = g.db.query(
|
|
Vote.submission_id).filter_by(
|
|
user_id=u1.id,
|
|
vote_type=-1).all()
|
|
u1_comment_ups = g.db.query(
|
|
CommentVote.comment_id).filter_by(
|
|
user_id=u1.id,
|
|
vote_type=1).all()
|
|
u1_comment_downs = g.db.query(
|
|
CommentVote.comment_id).filter_by(
|
|
user_id=u1.id,
|
|
vote_type=-1).all()
|
|
u2_post_ups = g.db.query(
|
|
Vote.submission_id).filter_by(
|
|
user_id=u2.id,
|
|
vote_type=1).all()
|
|
u2_post_downs = g.db.query(
|
|
Vote.submission_id).filter_by(
|
|
user_id=u2.id,
|
|
vote_type=-1).all()
|
|
u2_comment_ups = g.db.query(
|
|
CommentVote.comment_id).filter_by(
|
|
user_id=u2.id,
|
|
vote_type=1).all()
|
|
u2_comment_downs = g.db.query(
|
|
CommentVote.comment_id).filter_by(
|
|
user_id=u2.id,
|
|
vote_type=-1).all()
|
|
|
|
data = {}
|
|
data['u1_only_post_ups'] = len(
|
|
[x for x in u1_post_ups if x not in u2_post_ups])
|
|
data['u2_only_post_ups'] = len(
|
|
[x for x in u2_post_ups if x not in u1_post_ups])
|
|
data['both_post_ups'] = len(list(set(u1_post_ups) & set(u2_post_ups)))
|
|
|
|
data['u1_only_post_downs'] = len(
|
|
[x for x in u1_post_downs if x not in u2_post_downs])
|
|
data['u2_only_post_downs'] = len(
|
|
[x for x in u2_post_downs if x not in u1_post_downs])
|
|
data['both_post_downs'] = len(
|
|
list(set(u1_post_downs) & set(u2_post_downs)))
|
|
|
|
data['u1_only_comment_ups'] = len(
|
|
[x for x in u1_comment_ups if x not in u2_comment_ups])
|
|
data['u2_only_comment_ups'] = len(
|
|
[x for x in u2_comment_ups if x not in u1_comment_ups])
|
|
data['both_comment_ups'] = len(
|
|
list(set(u1_comment_ups) & set(u2_comment_ups)))
|
|
|
|
data['u1_only_comment_downs'] = len(
|
|
[x for x in u1_comment_downs if x not in u2_comment_downs])
|
|
data['u2_only_comment_downs'] = len(
|
|
[x for x in u2_comment_downs if x not in u1_comment_downs])
|
|
data['both_comment_downs'] = len(
|
|
list(set(u1_comment_downs) & set(u2_comment_downs)))
|
|
|
|
data['u1_post_ups_unique'] = 100 * \
|
|
data['u1_only_post_ups'] // len(u1_post_ups) if u1_post_ups else 0
|
|
data['u2_post_ups_unique'] = 100 * \
|
|
data['u2_only_post_ups'] // len(u2_post_ups) if u2_post_ups else 0
|
|
data['u1_post_downs_unique'] = 100 * \
|
|
data['u1_only_post_downs'] // len(
|
|
u1_post_downs) if u1_post_downs else 0
|
|
data['u2_post_downs_unique'] = 100 * \
|
|
data['u2_only_post_downs'] // len(
|
|
u2_post_downs) if u2_post_downs else 0
|
|
|
|
data['u1_comment_ups_unique'] = 100 * \
|
|
data['u1_only_comment_ups'] // len(
|
|
u1_comment_ups) if u1_comment_ups else 0
|
|
data['u2_comment_ups_unique'] = 100 * \
|
|
data['u2_only_comment_ups'] // len(
|
|
u2_comment_ups) if u2_comment_ups else 0
|
|
data['u1_comment_downs_unique'] = 100 * \
|
|
data['u1_only_comment_downs'] // len(
|
|
u1_comment_downs) if u1_comment_downs else 0
|
|
data['u2_comment_downs_unique'] = 100 * \
|
|
data['u2_only_comment_downs'] // len(
|
|
u2_comment_downs) if u2_comment_downs else 0
|
|
|
|
return render_template("admin/alt_votes.html",
|
|
u1=u1,
|
|
u2=u2,
|
|
v=v,
|
|
data=data
|
|
)
|
|
|
|
|
|
@app.post("/admin/link_accounts")
|
|
@admin_level_required(4)
|
|
@validate_formkey
|
|
def admin_link_accounts(v):
|
|
|
|
u1 = int(request.form.get("u1"))
|
|
u2 = int(request.form.get("u2"))
|
|
|
|
new_alt = Alt(
|
|
user1=u1,
|
|
user2=u2,
|
|
is_manual=True
|
|
)
|
|
|
|
g.db.add(new_alt)
|
|
|
|
return redirect(f"/admin/alt_votes?u1={g.db.query(User).get(u1).username}&u2={g.db.query(User).get(u2).username}")
|
|
|
|
|
|
@app.get("/admin/removed")
|
|
@admin_level_required(3)
|
|
def admin_removed(v):
|
|
|
|
page = int(request.args.get("page", 1))
|
|
|
|
ids = g.db.query(Submission.id).options(lazyload('*')).filter_by(is_banned=True).order_by(
|
|
Submission.id.desc()).offset(25 * (page - 1)).limit(26).all()
|
|
|
|
ids=[x[0] for x in ids]
|
|
|
|
next_exists = len(ids) == 26
|
|
|
|
ids = ids[:25]
|
|
|
|
posts = get_posts(ids, v=v)
|
|
|
|
return render_template("admin/removed_posts.html",
|
|
v=v,
|
|
listing=posts,
|
|
page=page,
|
|
next_exists=next_exists
|
|
)
|
|
|
|
|
|
@app.post("/admin/image_purge")
|
|
@admin_level_required(5)
|
|
def admin_image_purge(v):
|
|
|
|
name = request.form.get("url")
|
|
image = g.db.query(Image).filter(Image.text == name).first()
|
|
if image:
|
|
requests.delete(f'https://api.imgur.com/3/image/{image.deletehash}', headers = {"Authorization": f"Client-ID {IMGUR_KEY}"})
|
|
headers = {"Authorization": f"Bearer {CF_KEY}", "Content-Type": "application/json"}
|
|
data = {'files': [name]}
|
|
url = f"https://api.cloudflare.com/client/v4/zones/{CF_ZONE}/purge_cache"
|
|
requests.post(url, headers=headers, json=data)
|
|
return redirect("/admin/image_purge")
|
|
|
|
|
|
@app.post("/admin/image_ban")
|
|
@admin_level_required(4)
|
|
@validate_formkey
|
|
def admin_image_ban(v):
|
|
|
|
i=request.files['file']
|
|
|
|
|
|
#make phash
|
|
tempname = f"admin_image_ban_{v.username}_{int(time.time())}"
|
|
|
|
i.save(tempname)
|
|
|
|
h=imagehash.phash(IMAGE.open(tempname))
|
|
|
|
value = int(str(h), 16)
|
|
bindigits = []
|
|
|
|
# Seed digit: 2**0
|
|
digit = (value % 2)
|
|
value //= 2
|
|
bindigits.append(digit)
|
|
|
|
while value > 0:
|
|
# Next power of 2**n
|
|
digit = (value % 2)
|
|
value //= 2
|
|
bindigits.append(digit)
|
|
|
|
h = ''.join([str(d) for d in bindigits])
|
|
|
|
#check db for existing
|
|
badpic = g.db.query(BadPic).filter_by(
|
|
phash=h
|
|
).first()
|
|
|
|
remove(tempname)
|
|
|
|
if badpic:
|
|
return render_template("admin/image_ban.html", v=v, existing=badpic)
|
|
|
|
new_bp=BadPic(
|
|
phash=h,
|
|
ban_reason=request.form.get("ban_reason"),
|
|
ban_time=int(request.form.get("ban_length",0))
|
|
)
|
|
|
|
g.db.add(new_bp)
|
|
|
|
return render_template("admin/image_ban.html", v=v, success=True)
|
|
|
|
|
|
@app.post("/agendaposter/<user_id>")
|
|
@admin_level_required(6)
|
|
@validate_formkey
|
|
def agendaposter(user_id, v):
|
|
user = g.db.query(User).filter_by(id=user_id).first()
|
|
|
|
expiry = request.form.get("days", 0)
|
|
if expiry:
|
|
expiry = int(expiry)
|
|
expiry = g.timestamp + expiry*60*60*24
|
|
else:
|
|
expiry = 0
|
|
|
|
user.agendaposter = not user.agendaposter
|
|
user.agendaposter_expires_utc = expiry
|
|
g.db.add(user)
|
|
for alt in user.alts:
|
|
if alt.admin_level > 0: break
|
|
alt.agendaposter = user.agendaposter
|
|
alt.agendaposter_expires_utc = expiry
|
|
g.db.add(alt)
|
|
|
|
note = None
|
|
|
|
if not user.agendaposter: kind = "unagendaposter"
|
|
else:
|
|
kind = "agendaposter"
|
|
note = f"for {request.form.get('days')} days" if expiry else "never expires"
|
|
|
|
ma = ModAction(
|
|
kind=kind,
|
|
user_id=v.id,
|
|
target_user_id=user.id,
|
|
note = note
|
|
)
|
|
g.db.add(ma)
|
|
|
|
user.refresh_selfset_badges()
|
|
|
|
g.db.flush()
|
|
|
|
if user.agendaposter: send_notification(NOTIFICATIONS_ACCOUNT, user, f"You have been marked by an admin as an agendaposter ({note}).")
|
|
else: send_notification(NOTIFICATIONS_ACCOUNT, user, f"You have been unmarked by an admin as an agendaposter.")
|
|
|
|
return (redirect(user.url), user)
|
|
|
|
@app.post("/shadowban/<user_id>")
|
|
@admin_level_required(6)
|
|
@validate_formkey
|
|
def shadowban(user_id, v):
|
|
user = g.db.query(User).filter_by(id=user_id).first()
|
|
if user.admin_level != 0: abort(403)
|
|
user.shadowbanned = True
|
|
g.db.add(user)
|
|
for alt in user.alts:
|
|
if alt.admin_level > 0: break
|
|
alt.shadowbanned = True
|
|
g.db.add(alt)
|
|
ma = ModAction(
|
|
kind="shadowban",
|
|
user_id=v.id,
|
|
target_user_id=user.id,
|
|
)
|
|
g.db.add(ma)
|
|
|
|
cache.delete_memoized(frontlist)
|
|
|
|
return {"message": "User shadowbanned!"}
|
|
|
|
|
|
@app.post("/unshadowban/<user_id>")
|
|
@admin_level_required(6)
|
|
@validate_formkey
|
|
def unshadowban(user_id, v):
|
|
user = g.db.query(User).filter_by(id=user_id).first()
|
|
if user.admin_level != 0: abort(403)
|
|
user.shadowbanned = False
|
|
g.db.add(user)
|
|
for alt in user.alts:
|
|
alt.shadowbanned = False
|
|
g.db.add(alt)
|
|
|
|
ma = ModAction(
|
|
kind="unshadowban",
|
|
user_id=v.id,
|
|
target_user_id=user.id,
|
|
)
|
|
g.db.add(ma)
|
|
|
|
cache.delete_memoized(frontlist)
|
|
|
|
return {"message": "User unshadowbanned!"}
|
|
|
|
@app.post("/admin/verify/<user_id>")
|
|
@admin_level_required(6)
|
|
@validate_formkey
|
|
def verify(user_id, v):
|
|
user = g.db.query(User).filter_by(id=user_id).first()
|
|
user.verified = "Verified"
|
|
g.db.add(user)
|
|
return {"message": "User verfied!"}
|
|
|
|
@app.post("/admin/unverify/<user_id>")
|
|
@admin_level_required(6)
|
|
@validate_formkey
|
|
def unverify(user_id, v):
|
|
user = g.db.query(User).filter_by(id=user_id).first()
|
|
user.verified = None
|
|
g.db.add(user)
|
|
return {"message": "User unverified!"}
|
|
|
|
|
|
@app.post("/admin/title_change/<user_id>")
|
|
@admin_level_required(6)
|
|
@validate_formkey
|
|
def admin_title_change(user_id, v):
|
|
|
|
user = g.db.query(User).filter_by(id=user_id).first()
|
|
|
|
if user.admin_level != 0: abort(403)
|
|
|
|
new_name=request.form.get("title").strip()
|
|
|
|
user.customtitleplain=new_name
|
|
new_name = sanitize(new_name)
|
|
|
|
user=g.db.query(User).with_for_update().options(lazyload('*')).filter_by(id=user.id).first()
|
|
user.customtitle=new_name
|
|
user.flairchanged = bool(request.form.get("locked"))
|
|
g.db.add(user)
|
|
|
|
if user.flairchanged: kind = "set_flair_locked"
|
|
else: kind = "set_flair_notlocked"
|
|
|
|
ma=ModAction(
|
|
kind=kind,
|
|
user_id=v.id,
|
|
target_user_id=user.id,
|
|
note=f'"{new_name}"'
|
|
)
|
|
g.db.add(ma)
|
|
|
|
return (redirect(user.url), user)
|
|
|
|
@app.post("/ban_user/<user_id>")
|
|
@admin_level_required(6)
|
|
@validate_formkey
|
|
def ban_user(user_id, v):
|
|
|
|
user = g.db.query(User).filter_by(id=user_id).first()
|
|
|
|
if user.admin_level >= v.admin_level: abort(403)
|
|
|
|
# check for number of days for suspension
|
|
days = int(request.form.get("days")) if request.form.get('days') else 0
|
|
reason = sanitize(request.values.get("reason", ""))
|
|
message = request.values.get("reason", "")
|
|
|
|
if not user: abort(400)
|
|
|
|
#if user.admin_level > 0: abort(403)
|
|
|
|
if days > 0:
|
|
if message:
|
|
text = f"Your account has been suspended for {days} days for the following reason:\n\n> {message}"
|
|
else:
|
|
text = f"Your account has been suspended for {days} days."
|
|
user.ban(admin=v, reason=reason, days=days)
|
|
|
|
else:
|
|
if message:
|
|
text = f"Your account has been permanently suspended for the following reason:\n\n> {message}"
|
|
else:
|
|
text = "Your account has been permanently suspended."
|
|
|
|
user.ban(admin=v, reason=reason)
|
|
|
|
if request.form.get("alts", ""):
|
|
for x in user.alts:
|
|
if x.admin_level > 0: break
|
|
x.ban(admin=v, reason=reason)
|
|
|
|
send_notification(NOTIFICATIONS_ACCOUNT, user, text)
|
|
|
|
if days == 0: duration = "permanent"
|
|
elif days == 1: duration = "1 day"
|
|
else: duration = f"{days} days"
|
|
ma=ModAction(
|
|
kind="exile_user",
|
|
user_id=v.id,
|
|
target_user_id=user.id,
|
|
note=f'reason: "{reason}", duration: {duration}'
|
|
)
|
|
g.db.add(ma)
|
|
|
|
if 'reason' in request.args:
|
|
if reason.startswith("/post/"):
|
|
post = reason.split("/post/")[1].split("/")[0]
|
|
post = get_post(post)
|
|
post.bannedfor = True
|
|
g.db.add(post)
|
|
elif reason.startswith("/comment/"):
|
|
comment = reason.split("/comment/")[1].split("/")[0]
|
|
comment = get_comment(comment)
|
|
comment.bannedfor = True
|
|
g.db.add(comment)
|
|
return {"message": f"@{user.username} was banned!"}
|
|
else: return redirect(user.url)
|
|
|
|
|
|
@app.post("/unban_user/<user_id>")
|
|
@admin_level_required(6)
|
|
@validate_formkey
|
|
def unban_user(user_id, v):
|
|
|
|
user = g.db.query(User).filter_by(id=user_id).first()
|
|
|
|
if not user:
|
|
abort(400)
|
|
|
|
user.unban()
|
|
|
|
if request.form.get("alts", ""):
|
|
for x in user.alts:
|
|
if x.admin_level == 0:
|
|
x.unban()
|
|
|
|
send_notification(NOTIFICATIONS_ACCOUNT, user,
|
|
"Your account has been reinstated. Please carefully review and abide by the [rules](/post/2510) to ensure that you don't get suspended again.")
|
|
|
|
ma=ModAction(
|
|
kind="unexile_user",
|
|
user_id=v.id,
|
|
target_user_id=user.id,
|
|
)
|
|
g.db.add(ma)
|
|
|
|
if "@" in request.referrer:
|
|
return redirect(user.url)
|
|
else:
|
|
return {"message": f"@{user.username} was unbanned!"}
|
|
|
|
@app.post("/ban_post/<post_id>")
|
|
@admin_level_required(3)
|
|
@validate_formkey
|
|
def ban_post(post_id, v):
|
|
|
|
post = g.db.query(Submission).filter_by(id=post_id).first()
|
|
|
|
if not post:
|
|
abort(400)
|
|
|
|
post.is_banned = True
|
|
post.is_approved = 0
|
|
post.stickied = False
|
|
post.is_pinned = False
|
|
|
|
ban_reason=request.form.get("reason", "")
|
|
ban_reason = ban_reason.replace("\n", "\n\n").replace("\n\n\n\n\n\n", "\n\n").replace("\n\n\n\n", "\n\n").replace("\n\n\n", "\n\n")
|
|
with CustomRenderer() as renderer:
|
|
ban_reason = renderer.render(mistletoe.Document(ban_reason))
|
|
ban_reason = sanitize(ban_reason)
|
|
|
|
post.ban_reason = ban_reason
|
|
|
|
g.db.add(post)
|
|
|
|
|
|
|
|
ma=ModAction(
|
|
kind="ban_post",
|
|
user_id=v.id,
|
|
target_submission_id=post.id,
|
|
)
|
|
g.db.add(ma)
|
|
|
|
cache.delete_memoized(frontlist)
|
|
|
|
return {"message": "Post removed!"}
|
|
|
|
|
|
@app.post("/unban_post/<post_id>")
|
|
@admin_level_required(3)
|
|
@validate_formkey
|
|
def unban_post(post_id, v):
|
|
|
|
post = g.db.query(Submission).filter_by(id=post_id).first()
|
|
|
|
if not post:
|
|
abort(400)
|
|
|
|
if post.is_banned:
|
|
ma=ModAction(
|
|
kind="unban_post",
|
|
user_id=v.id,
|
|
target_submission_id=post.id,
|
|
)
|
|
g.db.add(ma)
|
|
|
|
post.is_banned = False
|
|
post.is_approved = v.id
|
|
|
|
g.db.add(post)
|
|
|
|
cache.delete_memoized(frontlist)
|
|
|
|
return {"message": "Post approved!"}
|
|
|
|
|
|
@app.post("/distinguish/<post_id>")
|
|
@admin_level_required(1)
|
|
@validate_formkey
|
|
def api_distinguish_post(post_id, v):
|
|
|
|
post = g.db.query(Submission).filter_by(id=post_id).first()
|
|
|
|
if not post:
|
|
abort(404)
|
|
|
|
if not post.author_id == v.id:
|
|
abort(403)
|
|
|
|
if post.distinguish_level:
|
|
post.distinguish_level = 0
|
|
else:
|
|
post.distinguish_level = v.admin_level
|
|
|
|
g.db.add(post)
|
|
|
|
return {"message": "Post distinguished!"}
|
|
|
|
|
|
@app.post("/sticky/<post_id>")
|
|
@admin_level_required(3)
|
|
def api_sticky_post(post_id, v):
|
|
|
|
post = g.db.query(Submission).filter_by(id=post_id).first()
|
|
if post:
|
|
post.stickied = not (post.stickied)
|
|
g.db.add(post)
|
|
|
|
ma=ModAction(
|
|
kind="pin_post" if post.stickied else "unpin_post",
|
|
user_id=v.id,
|
|
target_submission_id=post.id
|
|
)
|
|
g.db.add(ma)
|
|
|
|
cache.delete_memoized(frontlist)
|
|
|
|
g.db.flush()
|
|
if post.stickied: return {"message": "Post pinned!"}
|
|
else: return {"message": "Post unpinned!"}
|
|
|
|
@app.post("/pin/<post_id>")
|
|
@auth_required
|
|
def api_pin_post(post_id, v):
|
|
|
|
post = g.db.query(Submission).filter_by(id=post_id).first()
|
|
if post:
|
|
post.is_pinned = not (post.is_pinned)
|
|
g.db.add(post)
|
|
g.db.flush()
|
|
|
|
if post.is_pinned: return {"message": "Post pinned!"}
|
|
else: return {"message": "Post unpinned!"}
|
|
|
|
@app.post("/ban_comment/<c_id>")
|
|
@admin_level_required(1)
|
|
def api_ban_comment(c_id, v):
|
|
|
|
comment = g.db.query(Comment).filter_by(id=c_id).first()
|
|
if not comment:
|
|
abort(404)
|
|
|
|
comment.is_banned = True
|
|
comment.is_approved = 0
|
|
|
|
g.db.add(comment)
|
|
ma=ModAction(
|
|
kind="ban_comment",
|
|
user_id=v.id,
|
|
target_comment_id=comment.id,
|
|
)
|
|
g.db.add(ma)
|
|
return {"message": "Comment removed!"}
|
|
|
|
|
|
@app.post("/unban_comment/<c_id>")
|
|
@admin_level_required(1)
|
|
def api_unban_comment(c_id, v):
|
|
|
|
comment = g.db.query(Comment).filter_by(id=c_id).first()
|
|
if not comment:
|
|
abort(404)
|
|
g.db.add(comment)
|
|
|
|
if comment.is_banned:
|
|
ma=ModAction(
|
|
kind="unban_comment",
|
|
user_id=v.id,
|
|
target_comment_id=comment.id,
|
|
)
|
|
g.db.add(ma)
|
|
|
|
comment.is_banned = False
|
|
comment.is_approved = v.id
|
|
|
|
|
|
return {"message": "Comment approved!"}
|
|
|
|
|
|
@app.post("/distinguish_comment/<c_id>")
|
|
@auth_required
|
|
def admin_distinguish_comment(c_id, v):
|
|
|
|
if v.admin_level == 0: abort(403)
|
|
|
|
comment = get_comment(c_id, v=v)
|
|
|
|
if comment.author_id != v.id:
|
|
abort(403)
|
|
|
|
comment.distinguish_level = 0 if comment.distinguish_level else v.admin_level
|
|
|
|
g.db.add(comment)
|
|
html=render_template(
|
|
"comments.html",
|
|
v=v,
|
|
comments=[comment],
|
|
render_replies=False,
|
|
)
|
|
|
|
html=str(BeautifulSoup(html, features="html.parser").find(id=f"comment-{comment.id}-only"))
|
|
|
|
return html
|
|
|
|
@app.get("/admin/dump_cache")
|
|
@admin_level_required(6)
|
|
def admin_dump_cache(v):
|
|
cache.clear()
|
|
return {"message": "Internal cache cleared."}
|
|
|
|
|
|
@app.get("/admin/banned_domains/")
|
|
@admin_level_required(4)
|
|
def admin_banned_domains(v):
|
|
|
|
banned_domains = g.db.query(BannedDomain).all()
|
|
return render_template("admin/banned_domains.html", v=v, banned_domains=banned_domains)
|
|
|
|
@app.post("/admin/banned_domains")
|
|
@admin_level_required(4)
|
|
@validate_formkey
|
|
def admin_toggle_ban_domain(v):
|
|
|
|
domain=request.form.get("DOMAIN").strip()
|
|
if not domain: abort(400)
|
|
|
|
reason=request.form.get("reason", "").strip()
|
|
|
|
d = g.db.query(BannedDomain).filter_by(domain=domain).first()
|
|
if d: g.db.delete(d)
|
|
else:
|
|
d = BannedDomain(domain=domain, reason=reason)
|
|
g.db.add(d)
|
|
return redirect("/admin/banned_domains/")
|
|
|
|
|
|
@app.post("/admin/nuke_user")
|
|
@admin_level_required(4)
|
|
@validate_formkey
|
|
def admin_nuke_user(v):
|
|
|
|
user=get_user(request.form.get("user"))
|
|
|
|
for post in g.db.query(Submission).filter_by(author_id=user.id).all():
|
|
if post.is_banned:
|
|
continue
|
|
|
|
post.is_banned=True
|
|
g.db.add(post)
|
|
|
|
for comment in g.db.query(Comment).filter_by(author_id=user.id).all():
|
|
if comment.is_banned:
|
|
continue
|
|
|
|
comment.is_banned=True
|
|
g.db.add(comment)
|
|
|
|
ma=ModAction(
|
|
kind="nuke_user",
|
|
user_id=v.id,
|
|
target_user_id=user.id,
|
|
)
|
|
g.db.add(ma)
|
|
|
|
return redirect(user.url)
|
|
|
|
@app.post("/admin/unnuke_user")
|
|
@admin_level_required(4)
|
|
@validate_formkey
|
|
def admin_nunuke_user(v):
|
|
|
|
user=get_user(request.form.get("user"))
|
|
|
|
for post in g.db.query(Submission).filter_by(author_id=user.id).all():
|
|
if not post.is_banned:
|
|
continue
|
|
|
|
post.is_banned=False
|
|
g.db.add(post)
|
|
|
|
for comment in g.db.query(Comment).filter_by(author_id=user.id).all():
|
|
if not comment.is_banned:
|
|
continue
|
|
|
|
comment.is_banned=False
|
|
g.db.add(comment)
|
|
|
|
ma=ModAction(
|
|
kind="unnuke_user",
|
|
user_id=v.id,
|
|
target_user_id=user.id,
|
|
)
|
|
g.db.add(ma)
|
|
|
|
return redirect(user.url)
|
|
|
|
@app.get("/admin/user_stat_data")
|
|
@admin_level_required(2)
|
|
def user_stat_data(v):
|
|
|
|
days = int(request.args.get("days", 25))
|
|
|
|
now = time.gmtime()
|
|
midnight_this_morning = time.struct_time((now.tm_year,
|
|
now.tm_mon,
|
|
now.tm_mday,
|
|
0,
|
|
0,
|
|
0,
|
|
now.tm_wday,
|
|
now.tm_yday,
|
|
0)
|
|
)
|
|
today_cutoff = calendar.timegm(midnight_this_morning)
|
|
|
|
day = 3600 * 24
|
|
|
|
day_cutoffs = [today_cutoff - day * i for i in range(days)]
|
|
day_cutoffs.insert(0, calendar.timegm(now))
|
|
|
|
daily_signups = [{"date": time.strftime("%d", time.gmtime(day_cutoffs[i + 1])),
|
|
"day_start":day_cutoffs[i + 1],
|
|
"signups": g.db.query(User).filter(User.created_utc < day_cutoffs[i],
|
|
User.created_utc > day_cutoffs[i + 1] ).count()
|
|
} for i in range(len(day_cutoffs) - 1)
|
|
]
|
|
|
|
user_stats = {'current_users': g.db.query(User).filter_by(is_banned=0, reserved=None).count(),
|
|
'banned_users': g.db.query(User).filter(User.is_banned != 0).count(),
|
|
'reserved_users': g.db.query(User).filter(User.reserved != None).count(),
|
|
'email_verified_users': g.db.query(User).filter_by(is_banned=0, is_activated=True).count(),
|
|
}
|
|
|
|
post_stats = [{"date": time.strftime("%d", time.gmtime(day_cutoffs[i + 1])),
|
|
"day_start":day_cutoffs[i + 1],
|
|
"posts": g.db.query(Submission).filter(Submission.created_utc < day_cutoffs[i],
|
|
Submission.created_utc > day_cutoffs[i + 1],
|
|
Submission.is_banned == False
|
|
).count()
|
|
} for i in range(len(day_cutoffs) - 1)
|
|
]
|
|
|
|
comment_stats = [{"date": time.strftime("%d", time.gmtime(day_cutoffs[i + 1])),
|
|
"day_start": day_cutoffs[i + 1],
|
|
"comments": g.db.query(Comment).filter(Comment.created_utc < day_cutoffs[i],
|
|
Comment.created_utc > day_cutoffs[i + 1],
|
|
Comment.is_banned == False,
|
|
Comment.author_id != 1
|
|
).count()
|
|
} for i in range(len(day_cutoffs) - 1)
|
|
]
|
|
|
|
x = create_plot(sign_ups={'daily_signups': daily_signups},
|
|
posts={'post_stats': post_stats},
|
|
comments={'comment_stats': comment_stats},
|
|
)
|
|
|
|
final = {
|
|
"multi_plot": x,
|
|
"user_stats": user_stats,
|
|
"signup_data": daily_signups,
|
|
"post_data": post_stats,
|
|
"comment_data": comment_stats,
|
|
}
|
|
|
|
return final
|
|
|
|
|
|
def create_plot(**kwargs):
|
|
|
|
if not kwargs:
|
|
return abort(400)
|
|
|
|
# create multiple charts
|
|
daily_signups = [d["signups"] for d in kwargs["sign_ups"]['daily_signups']][1:][::-1]
|
|
post_stats = [d["posts"] for d in kwargs["posts"]['post_stats']][1:][::-1]
|
|
comment_stats = [d["comments"] for d in kwargs["comments"]['comment_stats']][1:][::-1]
|
|
daily_times = [d["date"] for d in kwargs["sign_ups"]['daily_signups']][1:][::-1]
|
|
|
|
multi_plots = multiple_plots(sign_ups=daily_signups,
|
|
posts=post_stats,
|
|
comments=comment_stats,
|
|
daily_times=daily_times)
|
|
|
|
return multi_plots
|
|
|
|
|
|
def multiple_plots(**kwargs):
|
|
|
|
# create multiple charts
|
|
signup_chart = plt.subplot2grid((20, 4), (0, 0), rowspan=5, colspan=4)
|
|
posts_chart = plt.subplot2grid((20, 4), (7, 0), rowspan=5, colspan=4)
|
|
comments_chart = plt.subplot2grid((20, 4), (14, 0), rowspan=5, colspan=4)
|
|
|
|
signup_chart.grid(), posts_chart.grid(), comments_chart.grid()
|
|
|
|
signup_chart.plot(
|
|
kwargs['daily_times'],
|
|
kwargs['sign_ups'],
|
|
color='red')
|
|
posts_chart.plot(
|
|
kwargs['daily_times'],
|
|
kwargs['posts'],
|
|
color='green')
|
|
comments_chart.plot(
|
|
kwargs['daily_times'],
|
|
kwargs['comments'],
|
|
color='gold')
|
|
|
|
signup_chart.set_ylabel("Signups")
|
|
posts_chart.set_ylabel("Posts")
|
|
comments_chart.set_ylabel("Comments")
|
|
comments_chart.set_xlabel("Time (UTC)")
|
|
|
|
signup_chart.legend(loc='upper left', frameon=True)
|
|
posts_chart.legend(loc='upper left', frameon=True)
|
|
comments_chart.legend(loc='upper left', frameon=True)
|
|
|
|
plt.savefig("image.png")
|
|
plt.clf()
|
|
|
|
return upload_imgur(png=True)
|