forked from rDrama/rDrama
1111 lines
34 KiB
Python
1111 lines
34 KiB
Python
from __future__ import unicode_literals
|
|
from files.helpers.alerts import *
|
|
from files.helpers.sanitize import *
|
|
from files.helpers.filters import filter_comment_html
|
|
from files.helpers.markdown import *
|
|
from files.helpers.discord import remove_user, set_nick
|
|
from files.helpers.const import *
|
|
from files.mail import *
|
|
from files.__main__ import app, cache, limiter
|
|
import youtube_dl
|
|
from .front import frontlist
|
|
import os
|
|
from files.helpers.sanitize import filter_emojis_only
|
|
from files.helpers.discord import add_role
|
|
from shutil import copyfile
|
|
import requests
|
|
|
|
valid_username_regex = re.compile("^[a-zA-Z0-9_\-]{3,25}$")
|
|
valid_password_regex = re.compile("^.{8,100}$")
|
|
|
|
YOUTUBE_KEY = environ.get("YOUTUBE_KEY", "").strip()
|
|
COINS_NAME = environ.get("COINS_NAME").strip()
|
|
GUMROAD_TOKEN = environ.get("GUMROAD_TOKEN", "").strip()
|
|
SITE_NAME = environ.get("SITE_NAME", "").strip()
|
|
IMGUR_KEY = environ.get("IMGUR_KEY").strip()
|
|
GUMROAD_ID = environ.get("GUMROAD_ID", "tfcvri").strip()
|
|
|
|
tiers={
|
|
"(Paypig)": 1,
|
|
"(Renthog)": 2,
|
|
"(Landchad)": 3,
|
|
"(Terminally online turboautist)": 4,
|
|
"(Rich Bich)": 5,
|
|
}
|
|
|
|
@app.post("/settings/removebackground")
|
|
@limiter.limit("1/second")
|
|
@auth_required
|
|
@validate_formkey
|
|
def removebackground(v):
|
|
v.background = None
|
|
g.db.add(v)
|
|
g.db.commit()
|
|
return {"message": "Background removed!"}
|
|
|
|
@app.post("/settings/profile")
|
|
@limiter.limit("1/second")
|
|
@auth_required
|
|
@validate_formkey
|
|
def settings_profile_post(v):
|
|
if v and v.patron:
|
|
if request.content_length > 8 * 1024 * 1024: return {"error":"Max file size is 8 MB."}, 413
|
|
elif request.content_length > 4 * 1024 * 1024: return {"error":"Max file size is 4 MB."}, 413
|
|
|
|
updated = False
|
|
|
|
if request.values.get("background", v.background) != v.background:
|
|
updated = True
|
|
v.background = request.values.get("background", None)
|
|
|
|
elif request.values.get("slurreplacer", v.slurreplacer) != v.slurreplacer:
|
|
updated = True
|
|
v.slurreplacer = request.values.get("slurreplacer", None) == 'true'
|
|
|
|
elif request.values.get("hidevotedon", v.hidevotedon) != v.hidevotedon:
|
|
updated = True
|
|
v.hidevotedon = request.values.get("hidevotedon", None) == 'true'
|
|
|
|
elif request.values.get("cardview", v.cardview) != v.cardview:
|
|
updated = True
|
|
v.cardview = request.values.get("cardview", None) == 'true'
|
|
|
|
elif request.values.get("highlightcomments", v.highlightcomments) != v.highlightcomments:
|
|
updated = True
|
|
v.highlightcomments = request.values.get("highlightcomments", None) == 'true'
|
|
|
|
elif request.values.get("newtab", v.newtab) != v.newtab:
|
|
updated = True
|
|
v.newtab = request.values.get("newtab", None) == 'true'
|
|
|
|
elif request.values.get("newtabexternal", v.newtabexternal) != v.newtabexternal:
|
|
updated = True
|
|
v.newtabexternal = request.values.get("newtabexternal", None) == 'true'
|
|
|
|
elif request.values.get("oldreddit", v.oldreddit) != v.oldreddit:
|
|
updated = True
|
|
v.oldreddit = request.values.get("oldreddit", None) == 'true'
|
|
|
|
elif request.values.get("oldsite", v.oldsite) != v.oldsite:
|
|
updated = True
|
|
v.oldsite = request.values.get("oldsite", None) == 'true'
|
|
|
|
elif request.values.get("teddit", v.teddit) != v.teddit:
|
|
updated = True
|
|
v.teddit = request.values.get("teddit", None) == 'true'
|
|
|
|
elif request.values.get("nitter", v.nitter) != v.nitter:
|
|
updated = True
|
|
v.nitter = request.values.get("nitter", None) == 'true'
|
|
|
|
elif request.values.get("controversial", v.controversial) != v.controversial:
|
|
updated = True
|
|
v.controversial = request.values.get("controversial", None) == 'true'
|
|
|
|
elif request.values.get("sigs_disabled", v.sigs_disabled) != v.sigs_disabled:
|
|
updated = True
|
|
v.sigs_disabled = request.values.get("sigs_disabled", None) == 'true'
|
|
|
|
elif request.values.get("over18", v.over_18) != v.over_18:
|
|
updated = True
|
|
v.over_18 = request.values.get("over18", None) == 'true'
|
|
|
|
elif request.values.get("private", v.is_private) != v.is_private:
|
|
updated = True
|
|
v.is_private = request.values.get("private", None) == 'true'
|
|
|
|
elif request.values.get("nofollow", v.is_nofollow) != v.is_nofollow:
|
|
updated = True
|
|
v.is_nofollow = request.values.get("nofollow", None) == 'true'
|
|
|
|
elif request.values.get("bio") == "":
|
|
v.bio = None
|
|
v.bio_html = None
|
|
g.db.add(v)
|
|
g.db.commit()
|
|
if not v or v.oldsite: template = ''
|
|
else: template = 'CHRISTMAS/'
|
|
return render_template(f"{template}settings_profile.html", v=v, msg="Your bio has been updated.")
|
|
|
|
elif request.values.get("sig") == "":
|
|
v.sig = None
|
|
v.sig_html = None
|
|
g.db.add(v)
|
|
g.db.commit()
|
|
if not v or v.oldsite: template = ''
|
|
else: template = 'CHRISTMAS/'
|
|
return render_template(f"{template}settings_profile.html", v=v, msg="Your sig has been updated.")
|
|
|
|
elif request.values.get("friends") == "":
|
|
v.friends = None
|
|
v.friends_html = None
|
|
g.db.add(v)
|
|
g.db.commit()
|
|
if not v or v.oldsite: template = ''
|
|
else: template = 'CHRISTMAS/'
|
|
return render_template(f"{template}settings_profile.html", v=v, msg="Your friends list has been updated.")
|
|
|
|
elif request.values.get("enemies") == "":
|
|
v.enemies = None
|
|
v.enemies_html = None
|
|
g.db.add(v)
|
|
g.db.commit()
|
|
if not v or v.oldsite: template = ''
|
|
else: template = 'CHRISTMAS/'
|
|
return render_template(f"{template}settings_profile.html", v=v, msg="Your enemies list has been updated.")
|
|
|
|
elif (v.patron or v.id == 1904) and request.values.get("sig"):
|
|
sig = request.values.get("sig")[:200]
|
|
|
|
for i in re.finditer('^(https:\/\/.*\.(png|jpg|jpeg|gif|webp|PNG|JPG|JPEG|GIF|WEBP|9999))', sig, re.MULTILINE):
|
|
if "wikipedia" not in i.group(1): sig = sig.replace(i.group(1), f'![]({i.group(1)})')
|
|
|
|
sig_html = CustomRenderer().render(mistletoe.Document(sig))
|
|
sig_html = sanitize(sig_html)
|
|
bans = filter_comment_html(sig_html)
|
|
|
|
|
|
if bans:
|
|
ban = bans[0]
|
|
reason = f"Remove the {ban.domain} link from your sig and try again."
|
|
if ban.reason:
|
|
reason += f" {ban.reason}"
|
|
|
|
return {"error": reason}, 401
|
|
|
|
if len(sig_html) > 1000:
|
|
if not v or v.oldsite: template = ''
|
|
else: template = 'CHRISTMAS/'
|
|
return render_template(f"{template}settings_profile.html",
|
|
v=v,
|
|
error="Your sig is too long")
|
|
|
|
v.sig = sig[:200]
|
|
v.sig_html=sig_html
|
|
g.db.add(v)
|
|
g.db.commit()
|
|
if not v or v.oldsite: template = ''
|
|
else: template = 'CHRISTMAS/'
|
|
return render_template(f"{template}settings_profile.html",
|
|
v=v,
|
|
msg="Your sig has been updated.")
|
|
|
|
|
|
|
|
|
|
elif request.values.get("friends"):
|
|
friends = request.values.get("friends")[:500]
|
|
|
|
for i in re.finditer('^(https:\/\/.*\.(png|jpg|jpeg|gif|webp|PNG|JPG|JPEG|GIF|WEBP|9999))', friends, re.MULTILINE):
|
|
if "wikipedia" not in i.group(1): friends = friends.replace(i.group(1), f'![]({i.group(1)})')
|
|
|
|
friends_html = CustomRenderer().render(mistletoe.Document(friends))
|
|
friends_html = sanitize(friends_html)
|
|
bans = filter_comment_html(friends_html)
|
|
|
|
if bans:
|
|
ban = bans[0]
|
|
reason = f"Remove the {ban.domain} link from your friends list and try again."
|
|
if ban.reason: reason += f" {ban.reason}"
|
|
return {"error": reason}, 401
|
|
|
|
if len(friends_html) > 2000:
|
|
if not v or v.oldsite: template = ''
|
|
else: template = 'CHRISTMAS/'
|
|
return render_template(f"{template}settings_profile.html",
|
|
v=v,
|
|
error="Your friends list is too long")
|
|
|
|
|
|
notify_users = NOTIFY_USERS(friends_html, v)
|
|
|
|
cid = notif_comment(f"@{v.username} has added you to their friends list!")
|
|
|
|
for x in notify_users:
|
|
add_notif(cid, x)
|
|
|
|
v.friends = friends[:500]
|
|
v.friends_html=friends_html
|
|
g.db.add(v)
|
|
g.db.commit()
|
|
if not v or v.oldsite: template = ''
|
|
else: template = 'CHRISTMAS/'
|
|
return render_template(f"{template}settings_profile.html",
|
|
v=v,
|
|
msg="Your friends list has been updated.")
|
|
|
|
|
|
elif request.values.get("enemies"):
|
|
enemies = request.values.get("enemies")[:500]
|
|
|
|
for i in re.finditer('^(https:\/\/.*\.(png|jpg|jpeg|gif|webp|PNG|JPG|JPEG|GIF|WEBP|9999))', enemies, re.MULTILINE):
|
|
if "wikipedia" not in i.group(1): enemies = enemies.replace(i.group(1), f'![]({i.group(1)})')
|
|
|
|
enemies_html = CustomRenderer().render(mistletoe.Document(enemies))
|
|
enemies_html = sanitize(enemies_html)
|
|
bans = filter_comment_html(enemies_html)
|
|
|
|
if bans:
|
|
ban = bans[0]
|
|
reason = f"Remove the {ban.domain} link from your enemies list and try again."
|
|
if ban.reason: reason += f" {ban.reason}"
|
|
return {"error": reason}, 401
|
|
|
|
if len(enemies_html) > 2000:
|
|
if not v or v.oldsite: template = ''
|
|
else: template = 'CHRISTMAS/'
|
|
return render_template(f"{template}settings_profile.html",
|
|
v=v,
|
|
error="Your enemies list is too long")
|
|
|
|
|
|
notify_users = NOTIFY_USERS(enemies_html, v)
|
|
|
|
cid = notif_comment(f"@{v.username} has added you to their enemies list!")
|
|
|
|
for x in notify_users:
|
|
add_notif(cid, x)
|
|
|
|
v.enemies = enemies[:500]
|
|
v.enemies_html=enemies_html
|
|
g.db.add(v)
|
|
g.db.commit()
|
|
if not v or v.oldsite: template = ''
|
|
else: template = 'CHRISTMAS/'
|
|
return render_template(f"{template}settings_profile.html",
|
|
v=v,
|
|
msg="Your enemies list has been updated.")
|
|
|
|
|
|
elif request.values.get("bio") or request.files.get('file') and request.headers.get("cf-ipcountry") != "T1":
|
|
bio = request.values.get("bio")[:1500]
|
|
|
|
for i in re.finditer('^(https:\/\/.*\.(png|jpg|jpeg|gif|webp|PNG|JPG|JPEG|GIF|WEBP|9999))', bio, re.MULTILINE):
|
|
if "wikipedia" not in i.group(1): bio = bio.replace(i.group(1), f'![]({i.group(1)})')
|
|
|
|
if request.files.get('file'):
|
|
file = request.files['file']
|
|
if file.content_type.startswith('image/'):
|
|
name = f'/images/{time.time()}'.replace('.','')[:-5] + '.webp'
|
|
file.save(name)
|
|
url = process_image(name)
|
|
bio += f"\n\n![]({url})"
|
|
elif file.content_type.startswith('video/'):
|
|
file.save("video.mp4")
|
|
with open("video.mp4", 'rb') as f:
|
|
url = requests.request("POST", "https://api.imgur.com/3/upload", headers={'Authorization': f'Client-ID {IMGUR_KEY}'}, files=[('video', f)]).json()['data']['link']
|
|
bio += f"\n\n{url}"
|
|
else:
|
|
if request.headers.get("Authorization"): return {"error": f"Image/Video files only"}, 400
|
|
if not v or v.oldsite: template = ''
|
|
else: template = 'CHRISTMAS/'
|
|
return render_template(f"{template}settings_profile.html", v=v, error=f"Image/Video files only."), 400
|
|
|
|
bio_html = CustomRenderer().render(mistletoe.Document(bio))
|
|
bio_html = sanitize(bio_html)
|
|
bans = filter_comment_html(bio_html)
|
|
|
|
if len(bio_html) > 10000:
|
|
if not v or v.oldsite: template = ''
|
|
else: template = 'CHRISTMAS/'
|
|
return render_template(f"{template}settings_profile.html",
|
|
v=v,
|
|
error="Your bio is too long")
|
|
|
|
if bans:
|
|
ban = bans[0]
|
|
reason = f"Remove the {ban.domain} link from your bio and try again."
|
|
if ban.reason:
|
|
reason += f" {ban.reason}"
|
|
|
|
return {"error": reason}, 401
|
|
|
|
if len(bio_html) > 10000: abort(400)
|
|
|
|
v.bio = bio[:1500]
|
|
v.bio_html=bio_html
|
|
g.db.add(v)
|
|
g.db.commit()
|
|
if not v or v.oldsite: template = ''
|
|
else: template = 'CHRISTMAS/'
|
|
return render_template(f"{template}settings_profile.html",
|
|
v=v,
|
|
msg="Your bio has been updated.")
|
|
|
|
|
|
frontsize = request.values.get("frontsize")
|
|
if frontsize:
|
|
if frontsize in ["25", "50", "100"]:
|
|
v.frontsize = int(frontsize)
|
|
updated = True
|
|
cache.delete_memoized(frontlist)
|
|
else: abort(400)
|
|
|
|
defaultsortingcomments = request.values.get("defaultsortingcomments")
|
|
if defaultsortingcomments:
|
|
if defaultsortingcomments in ["new", "old", "controversial", "top", "bottom"]:
|
|
v.defaultsortingcomments = defaultsortingcomments
|
|
updated = True
|
|
else: abort(400)
|
|
|
|
defaultsorting = request.values.get("defaultsorting")
|
|
if defaultsorting:
|
|
if defaultsorting in ["hot", "new", "old", "comments", "controversial", "top", "bottom"]:
|
|
v.defaultsorting = defaultsorting
|
|
updated = True
|
|
else: abort(400)
|
|
|
|
defaulttime = request.values.get("defaulttime")
|
|
if defaulttime:
|
|
if defaulttime in ["hour", "day", "week", "month", "year", "all"]:
|
|
v.defaulttime = defaulttime
|
|
updated = True
|
|
else: abort(400)
|
|
|
|
theme = request.values.get("theme")
|
|
if theme:
|
|
v.theme = theme
|
|
if theme == "win98": v.themecolor = "30409f"
|
|
updated = True
|
|
|
|
theme2 = request.values.get("theme2")
|
|
if theme2:
|
|
v.theme2 = theme2
|
|
updated = True
|
|
|
|
quadrant = request.values.get("quadrant")
|
|
if quadrant and 'pcmemes.net' == request.host.lower():
|
|
v.quadrant = quadrant
|
|
v.customtitle = quadrant
|
|
if quadrant=="Centrist":
|
|
v.namecolor = "7f8fa6"
|
|
v.titlecolor = "7f8fa6"
|
|
elif quadrant=="LibLeft":
|
|
v.namecolor = "62ca56"
|
|
v.titlecolor = "62ca56"
|
|
elif quadrant=="LibRight":
|
|
v.namecolor = "f8db58"
|
|
v.titlecolor = "f8db58"
|
|
elif quadrant=="AuthLeft":
|
|
v.namecolor = "ff0000"
|
|
v.titlecolor = "ff0000"
|
|
elif quadrant=="AuthRight":
|
|
v.namecolor = "2a96f3"
|
|
v.titlecolor = "2a96f3"
|
|
elif quadrant=="LibCenter":
|
|
v.namecolor = "add357"
|
|
v.titlecolor = "add357"
|
|
elif quadrant=="AuthCenter":
|
|
v.namecolor = "954b7a"
|
|
v.titlecolor = "954b7a"
|
|
elif quadrant=="Left":
|
|
v.namecolor = "b1652b"
|
|
v.titlecolor = "b1652b"
|
|
elif quadrant=="Right":
|
|
v.namecolor = "91b9A6"
|
|
v.titlecolor = "91b9A6"
|
|
|
|
updated = True
|
|
|
|
if updated:
|
|
g.db.add(v)
|
|
g.db.commit()
|
|
|
|
return {"message": "Your settings have been updated."}
|
|
|
|
else:
|
|
return {"error": "You didn't change anything."}, 400
|
|
|
|
|
|
@app.post("/settings/filters")
|
|
@auth_required
|
|
@validate_formkey
|
|
def filters(v):
|
|
filters=request.values.get("filters")[:1000].strip()
|
|
|
|
if filters == v.custom_filter_list:
|
|
if not v or v.oldsite: template = ''
|
|
else: template = 'CHRISTMAS/'
|
|
return render_template(f"{template}settings_filters.html", v=v, error="You didn't change anything")
|
|
|
|
v.custom_filter_list=filters
|
|
g.db.add(v)
|
|
g.db.commit()
|
|
if not v or v.oldsite: template = ''
|
|
else: template = 'CHRISTMAS/'
|
|
return render_template(f"{template}settings_filters.html", v=v, msg="Your custom filters have been updated.")
|
|
|
|
@app.post("/changelogsub")
|
|
@auth_required
|
|
@validate_formkey
|
|
def changelogsub(v):
|
|
v.changelogsub = not v.changelogsub
|
|
g.db.add(v)
|
|
|
|
cache.delete_memoized(frontlist)
|
|
|
|
g.db.commit()
|
|
if v.changelogsub: return {"message": "You have subscribed to the changelog!"}
|
|
else: return {"message": "You have unsubscribed from the changelog!"}
|
|
|
|
@app.post("/settings/namecolor")
|
|
@limiter.limit("1/second")
|
|
@auth_required
|
|
@validate_formkey
|
|
def namecolor(v):
|
|
if not v or v.oldsite: template = ''
|
|
else: template = 'CHRISTMAS/'
|
|
|
|
color = str(request.values.get("color", "")).strip()
|
|
if color.startswith('#'): color = color[1:]
|
|
if len(color) != 6: return render_template(f"{template}settings_security.html", v=v, error="Invalid color code")
|
|
v.namecolor = color
|
|
g.db.add(v)
|
|
g.db.commit()
|
|
return redirect("/settings/profile")
|
|
|
|
@app.post("/settings/themecolor")
|
|
@limiter.limit("1/second")
|
|
@auth_required
|
|
@validate_formkey
|
|
def themecolor(v):
|
|
if not v or v.oldsite: template = ''
|
|
else: template = 'CHRISTMAS/'
|
|
|
|
themecolor = str(request.values.get("themecolor", "")).strip()
|
|
if themecolor.startswith('#'): themecolor = themecolor[1:]
|
|
if len(themecolor) != 6: return render_template(f"{template}settings_security.html", v=v, error="Invalid color code")
|
|
v.themecolor = themecolor
|
|
g.db.add(v)
|
|
g.db.commit()
|
|
return redirect("/settings/profile")
|
|
|
|
@app.post("/settings/gumroad")
|
|
@limiter.limit("1/second")
|
|
@auth_required
|
|
@validate_formkey
|
|
def gumroad(v):
|
|
if SITE_NAME == 'Drama': patron = 'Paypig'
|
|
else: patron = 'Patron'
|
|
|
|
if not (v.email and v.is_activated): return {"error": f"You must have a verified email to verify {patron} status and claim your rewards"}, 400
|
|
|
|
data = {'access_token': GUMROAD_TOKEN,}
|
|
|
|
response = [x['email'] for x in requests.get(f'https://api.gumroad.com/v2/products/{GUMROAD_ID}/subscribers', data=data, timeout=5).json()["subscribers"]]
|
|
emails = []
|
|
|
|
for email in response:
|
|
if email.endswith("@gmail.com"):
|
|
email=email.split('@')[0]
|
|
email=email.split('+')[0]
|
|
email=email.replace('.','').replace('_','')
|
|
email=f"{email}@gmail.com"
|
|
emails.append(email.lower())
|
|
|
|
if v.email.lower() not in emails: return {"error": "Email not found"}, 404
|
|
|
|
response = requests.get('https://api.gumroad.com/v2/sales', data=data, timeout=5).json()["sales"][0]
|
|
tier = tiers[response["variants_and_quantity"]]
|
|
|
|
if v.patron == tier: return {"error": f"{patron} rewards already claimed"}, 400
|
|
|
|
existing = g.db.query(User.id).filter_by(email=v.email, is_activated=True, patron=tier).one_or_none()
|
|
if existing: return {"error": f"{patron} rewards already claimed on another account"}, 400
|
|
|
|
if v.patron:
|
|
badge = v.has_badge(20+v.patron)
|
|
if badge: g.db.delete(badge)
|
|
|
|
v.patron = tier
|
|
if v.discord_id: add_role(v, f"{tier}")
|
|
|
|
if v.patron == 1: procoins = 2500
|
|
elif v.patron == 2: procoins = 5000
|
|
elif v.patron == 3: procoins = 10000
|
|
elif v.patron == 4: procoins = 25000
|
|
elif v.patron == 5: procoins = 50000
|
|
|
|
v.procoins += procoins
|
|
send_repeatable_notification(v.id, f"You have received {procoins} Marseybux! You can use them to buy awards in the [shop](/shop).")
|
|
|
|
if v.truecoins > 150 and v.patron > 0: v.cluballowed = True
|
|
if v.patron > 3 and v.verified == None: v.verified = "Verified"
|
|
|
|
g.db.add(v)
|
|
|
|
if not v.has_badge(20+tier):
|
|
new_badge = Badge(badge_id=20+tier, user_id=v.id)
|
|
g.db.add(new_badge)
|
|
send_notification(v.id, f"@AutoJanny has given you the following profile badge:\n\n![]({new_badge.path})\n\n{new_badge.name}")
|
|
|
|
g.db.commit()
|
|
|
|
return {"message": f"{patron} rewards claimed!"}
|
|
|
|
@app.post("/settings/titlecolor")
|
|
@limiter.limit("1/second")
|
|
@auth_required
|
|
@validate_formkey
|
|
def titlecolor(v):
|
|
if not v or v.oldsite: template = ''
|
|
else: template = 'CHRISTMAS/'
|
|
|
|
titlecolor = str(request.values.get("titlecolor", "")).strip()
|
|
if titlecolor.startswith('#'): titlecolor = titlecolor[1:]
|
|
if len(titlecolor) != 6: return render_template(f"{template}settings_profile.html", v=v, error="Invalid color code")
|
|
v.titlecolor = titlecolor
|
|
g.db.add(v)
|
|
g.db.commit()
|
|
return redirect("/settings/profile")
|
|
|
|
@app.post("/settings/verifiedcolor")
|
|
@limiter.limit("1/second")
|
|
@auth_required
|
|
@validate_formkey
|
|
def verifiedcolor(v):
|
|
if not v or v.oldsite: template = ''
|
|
else: template = 'CHRISTMAS/'
|
|
|
|
verifiedcolor = str(request.values.get("verifiedcolor", "")).strip()
|
|
if verifiedcolor.startswith('#'): verifiedcolor = verifiedcolor[1:]
|
|
if len(verifiedcolor) != 6: return render_template(f"{template}settings_profile.html", v=v, error="Invalid color code")
|
|
v.verifiedcolor = verifiedcolor
|
|
g.db.add(v)
|
|
g.db.commit()
|
|
return redirect("/settings/profile")
|
|
|
|
@app.post("/settings/security")
|
|
@limiter.limit("1/second")
|
|
@auth_required
|
|
@validate_formkey
|
|
def settings_security_post(v):
|
|
if request.values.get("new_password"):
|
|
if v.id == PW_ID: abort(403)
|
|
|
|
if request.values.get("new_password") != request.values.get("cnf_password"):
|
|
return render_template("settings_security.html", v=v, error="Passwords do not match.")
|
|
|
|
if not re.match(valid_password_regex, request.values.get("new_password")):
|
|
return render_template("settings_security.html", v=v, error="Password must be between 8 and 100 characters.")
|
|
|
|
if not v.verifyPass(request.values.get("old_password")):
|
|
return render_template("settings_security.html", v=v, error="Incorrect password")
|
|
|
|
v.passhash = v.hash_password(request.values.get("new_password"))
|
|
|
|
g.db.add(v)
|
|
|
|
g.db.commit()
|
|
|
|
return render_template("settings_security.html", v=v, error="Your password has been changed.")
|
|
|
|
if request.values.get("new_email"):
|
|
|
|
if not v.verifyPass(request.values.get('password')):
|
|
return render_template("settings_security.html", v=v, error="Invalid password.")
|
|
|
|
new_email = request.values.get("new_email","").strip().lower()
|
|
|
|
if new_email.endswith("@gmail.com"):
|
|
new_email=new_email.split('@')[0]
|
|
new_email=new_email.split('+')[0]
|
|
new_email=new_email.replace('.','').replace('_','')
|
|
new_email=f"{new_email}@gmail.com"
|
|
|
|
if new_email == v.email:
|
|
return render_template("settings_security.html", v=v, error="That email is already yours!")
|
|
|
|
existing = g.db.query(User.id).filter(User.id != v.id,
|
|
func.lower(User.email) == new_email.lower()).first()
|
|
if existing:
|
|
return render_template("settings_security.html", v=v, error="That email address is already in use.")
|
|
|
|
url = f"https://{app.config['SERVER_NAME']}/activate"
|
|
|
|
now = int(time.time())
|
|
|
|
token = generate_hash(f"{new_email}+{v.id}+{now}")
|
|
params = f"?email={quote(new_email)}&id={v.id}&time={now}&token={token}"
|
|
|
|
link = url + params
|
|
|
|
send_mail(to_address=new_email,
|
|
subject="Verify your email address.",
|
|
html=render_template("email/email_change.html",
|
|
action_url=link,
|
|
v=v)
|
|
)
|
|
|
|
return render_template("settings_security.html", v=v, error="Check your email and click the verification link to complete the email change.")
|
|
|
|
if request.values.get("2fa_token", ""):
|
|
|
|
if not v.verifyPass(request.values.get('password')):
|
|
return render_template("settings_security.html", v=v, error="Invalid password or token.")
|
|
|
|
secret = request.values.get("2fa_secret")
|
|
x = pyotp.TOTP(secret)
|
|
if not x.verify(request.values.get("2fa_token"), valid_window=1):
|
|
return render_template("settings_security.html", v=v, error="Invalid password or token.")
|
|
|
|
v.mfa_secret = secret
|
|
g.db.add(v)
|
|
|
|
g.db.commit()
|
|
|
|
return render_template("settings_security.html", v=v, error="Two-factor authentication enabled.")
|
|
|
|
if request.values.get("2fa_remove", ""):
|
|
|
|
if not v.verifyPass(request.values.get('password')):
|
|
return render_template("settings_security.html", v=v, error="Invalid password or token.")
|
|
|
|
token = request.values.get("2fa_remove")
|
|
|
|
if not v.validate_2fa(token):
|
|
return render_template("settings_security.html", v=v, error="Invalid password or token.")
|
|
|
|
v.mfa_secret = None
|
|
g.db.add(v)
|
|
|
|
g.db.commit()
|
|
|
|
return render_template("settings_security.html", v=v, error="Two-factor authentication disabled.")
|
|
|
|
@app.post("/settings/log_out_all_others")
|
|
@limiter.limit("1/second")
|
|
@auth_required
|
|
@validate_formkey
|
|
def settings_log_out_others(v):
|
|
|
|
submitted_password = request.values.get("password", "").strip()
|
|
|
|
if not v.verifyPass(submitted_password):
|
|
if not v or v.oldsite: template = ''
|
|
else: template = 'CHRISTMAS/'
|
|
return render_template(f"{template}settings_security.html", v=v, error="Incorrect Password"), 401
|
|
|
|
v.login_nonce += 1
|
|
|
|
session["login_nonce"] = v.login_nonce
|
|
|
|
g.db.add(v)
|
|
|
|
g.db.commit()
|
|
|
|
if not v or v.oldsite: template = ''
|
|
else: template = 'CHRISTMAS/'
|
|
return render_template(f"{template}settings_security.html", v=v, msg="All other devices have been logged out")
|
|
|
|
|
|
@app.post("/settings/images/profile")
|
|
@limiter.limit("1/second")
|
|
@auth_required
|
|
@validate_formkey
|
|
def settings_images_profile(v):
|
|
if v and v.patron:
|
|
if request.content_length > 8 * 1024 * 1024: return {"error":"Max file size is 8 MB."}, 413
|
|
elif request.content_length > 4 * 1024 * 1024: return {"error":"Max file size is 4 MB."}, 413
|
|
|
|
if request.headers.get("cf-ipcountry") == "T1": return {"error":"Image uploads are not allowed through TOR."}, 403
|
|
|
|
file = request.files["profile"]
|
|
|
|
name = f'/images/{time.time()}'.replace('.','')[:-5] + '.webp'
|
|
file.save(name)
|
|
highres = process_image(name)
|
|
|
|
if not highres: abort(400)
|
|
|
|
name2 = name.replace('.webp', 'r.webp')
|
|
copyfile(name, name2)
|
|
imageurl = process_image(name2, True)
|
|
|
|
if not imageurl: abort(400)
|
|
|
|
if v.highres and '/images/' in v.highres : os.remove('/images/' + v.highres.split('/images/')[1])
|
|
if v.profileurl and '/images/' in v.profileurl : os.remove('/images/' + v.profileurl.split('/images/')[1])
|
|
v.highres = highres
|
|
v.profileurl = imageurl
|
|
g.db.add(v)
|
|
|
|
g.db.commit()
|
|
|
|
if not v or v.oldsite: template = ''
|
|
else: template = 'CHRISTMAS/'
|
|
return render_template(f"{template}settings_profile.html", v=v, msg="Profile picture successfully updated.")
|
|
|
|
|
|
@app.post("/settings/images/banner")
|
|
@limiter.limit("1/second")
|
|
@auth_required
|
|
@validate_formkey
|
|
def settings_images_banner(v):
|
|
if v and v.patron:
|
|
if request.content_length > 8 * 1024 * 1024: return {"error":"Max file size is 8 MB."}, 413
|
|
elif request.content_length > 4 * 1024 * 1024: return {"error":"Max file size is 4 MB."}, 413
|
|
|
|
if request.headers.get("cf-ipcountry") == "T1": return {"error":"Image uploads are not allowed through TOR."}, 403
|
|
|
|
file = request.files["banner"]
|
|
|
|
name = f'/images/{time.time()}'.replace('.','')[:-5] + '.webp'
|
|
file.save(name)
|
|
bannerurl = process_image(name)
|
|
|
|
if bannerurl:
|
|
if v.bannerurl and '/images/' in v.bannerurl : os.remove('/images/' + v.bannerurl.split('/images/')[1])
|
|
v.bannerurl = bannerurl
|
|
g.db.add(v)
|
|
g.db.commit()
|
|
|
|
if not v or v.oldsite: template = ''
|
|
else: template = 'CHRISTMAS/'
|
|
return render_template(f"{template}settings_profile.html", v=v, msg="Banner successfully updated.")
|
|
|
|
|
|
@app.post("/settings/delete/profile")
|
|
@limiter.limit("1/second")
|
|
@auth_required
|
|
@validate_formkey
|
|
def settings_delete_profile(v):
|
|
|
|
if v.profileurl or v.highres:
|
|
v.deletepfp()
|
|
g.db.add(v)
|
|
g.db.commit()
|
|
|
|
if not v or v.oldsite: template = ''
|
|
else: template = 'CHRISTMAS/'
|
|
return render_template(f"{template}settings_profile.html", v=v,
|
|
msg="Profile picture successfully removed.")
|
|
|
|
@app.post("/settings/delete/banner")
|
|
@limiter.limit("1/second")
|
|
@auth_required
|
|
@validate_formkey
|
|
def settings_delete_banner(v):
|
|
|
|
if v.bannerurl:
|
|
v.deletebanner()
|
|
g.db.add(v)
|
|
g.db.commit()
|
|
|
|
if not v or v.oldsite: template = ''
|
|
else: template = 'CHRISTMAS/'
|
|
return render_template(f"{template}settings_profile.html", v=v, msg="Banner successfully removed.")
|
|
|
|
|
|
@app.get("/settings/blocks")
|
|
@auth_required
|
|
def settings_blockedpage(v):
|
|
|
|
if not v or v.oldsite: template = ''
|
|
else: template = 'CHRISTMAS/'
|
|
return render_template(f"{template}settings_blocks.html", v=v)
|
|
|
|
@app.get("/settings/css")
|
|
@auth_required
|
|
def settings_css_get(v):
|
|
|
|
if not v or v.oldsite: template = ''
|
|
else: template = 'CHRISTMAS/'
|
|
return render_template(f"{template}settings_css.html", v=v)
|
|
|
|
@app.post("/settings/css")
|
|
@limiter.limit("1/second")
|
|
@auth_required
|
|
@validate_formkey
|
|
def settings_css(v):
|
|
if v.agendaposter: return {"error": "Agendapostered users can't edit css!"}
|
|
|
|
css = request.values.get("css").strip().replace('\\', '').strip()[:4000]
|
|
v.css = css
|
|
g.db.add(v)
|
|
g.db.commit()
|
|
|
|
if not v or v.oldsite: template = ''
|
|
else: template = 'CHRISTMAS/'
|
|
return render_template(f"{template}settings_css.html", v=v)
|
|
|
|
@app.get("/settings/profilecss")
|
|
@auth_required
|
|
def settings_profilecss_get(v):
|
|
|
|
if v.truecoins < 1000 and not v.patron and v.admin_level == 0 : return f"You must have +1000 {COINS_NAME} or be a paypig to set profile css."
|
|
if not v or v.oldsite: template = ''
|
|
else: template = 'CHRISTMAS/'
|
|
return render_template(f"{template}settings_profilecss.html", v=v)
|
|
|
|
@app.post("/settings/profilecss")
|
|
@limiter.limit("1/second")
|
|
@auth_required
|
|
@validate_formkey
|
|
def settings_profilecss(v):
|
|
if v.truecoins < 1000 and not v.patron: return f"You must have +1000 {COINS_NAME} or be a paypig to set profile css."
|
|
profilecss = request.values.get("profilecss").strip().replace('\\', '').strip()[:4000]
|
|
v.profilecss = profilecss
|
|
g.db.add(v)
|
|
g.db.commit()
|
|
|
|
if not v or v.oldsite: template = ''
|
|
else: template = 'CHRISTMAS/'
|
|
return render_template(f"{template}settings_profilecss.html", v=v)
|
|
|
|
@app.post("/settings/block")
|
|
@limiter.limit("1/second")
|
|
@auth_required
|
|
@validate_formkey
|
|
def settings_block_user(v):
|
|
|
|
user = get_user(request.values.get("username"), graceful=True)
|
|
|
|
if not user: return {"error": "That user doesn't exist."}, 404
|
|
|
|
if user.unblockable: return {"error": "This user is unblockable."}, 403
|
|
|
|
if user.id == v.id:
|
|
return {"error": "You can't block yourself."}, 409
|
|
|
|
if v.is_blocking(user):
|
|
return {"error": f"You have already blocked @{user.username}."}, 409
|
|
|
|
if user.id == NOTIFICATIONS_ID:
|
|
return {"error": "You can't block this user."}, 409
|
|
|
|
new_block = UserBlock(user_id=v.id,
|
|
target_id=user.id,
|
|
)
|
|
g.db.add(new_block)
|
|
|
|
send_notification(user.id, f"@{v.username} has blocked you!")
|
|
|
|
cache.delete_memoized(frontlist)
|
|
|
|
g.db.commit()
|
|
|
|
return {"message": f"@{user.username} blocked."}
|
|
|
|
|
|
@app.post("/settings/unblock")
|
|
@limiter.limit("1/second")
|
|
@auth_required
|
|
@validate_formkey
|
|
def settings_unblock_user(v):
|
|
|
|
user = get_user(request.values.get("username"))
|
|
|
|
x = v.is_blocking(user)
|
|
|
|
if not x: abort(409)
|
|
|
|
g.db.delete(x)
|
|
|
|
send_notification(user.id, f"@{v.username} has unblocked you!")
|
|
|
|
cache.delete_memoized(frontlist)
|
|
|
|
g.db.commit()
|
|
|
|
return {"message": f"@{user.username} unblocked."}
|
|
|
|
|
|
@app.get("/settings/apps")
|
|
@auth_required
|
|
def settings_apps(v):
|
|
|
|
if not v or v.oldsite: template = ''
|
|
else: template = 'CHRISTMAS/'
|
|
return render_template(f"{template}settings_apps.html", v=v)
|
|
|
|
|
|
@app.post("/settings/remove_discord")
|
|
@limiter.limit("1/second")
|
|
@auth_required
|
|
@validate_formkey
|
|
def settings_remove_discord(v):
|
|
|
|
remove_user(v)
|
|
|
|
v.discord_id=None
|
|
g.db.add(v)
|
|
|
|
g.db.commit()
|
|
|
|
return redirect("/settings/profile")
|
|
|
|
@app.get("/settings/content")
|
|
@auth_required
|
|
def settings_content_get(v):
|
|
|
|
if not v or v.oldsite: template = ''
|
|
else: template = 'CHRISTMAS/'
|
|
return render_template(f"{template}settings_filters.html", v=v)
|
|
|
|
@app.post("/settings/name_change")
|
|
@limiter.limit("1/second")
|
|
@auth_required
|
|
@validate_formkey
|
|
def settings_name_change(v):
|
|
|
|
if v.is_banned and not v.unban_utc: return {"error": "forbidden."}, 403
|
|
|
|
new_name=request.values.get("name").strip()
|
|
|
|
if new_name==v.username:
|
|
if not v or v.oldsite: template = ''
|
|
else: template = 'CHRISTMAS/'
|
|
return render_template(f"{template}settings_profile.html",
|
|
v=v,
|
|
error="You didn't change anything")
|
|
|
|
if not re.match(valid_username_regex, new_name):
|
|
if not v or v.oldsite: template = ''
|
|
else: template = 'CHRISTMAS/'
|
|
return render_template(f"{template}settings_profile.html",
|
|
v=v,
|
|
error=f"This isn't a valid username.")
|
|
|
|
name=new_name.replace('_','\_')
|
|
|
|
x= g.db.query(User).filter(
|
|
or_(
|
|
User.username.ilike(name),
|
|
User.original_username.ilike(name)
|
|
)
|
|
).first()
|
|
|
|
if x and x.id != v.id:
|
|
if not v or v.oldsite: template = ''
|
|
else: template = 'CHRISTMAS/'
|
|
return render_template(f"{template}settings_profile.html",
|
|
v=v,
|
|
error=f"Username `{new_name}` is already in use.")
|
|
|
|
v=g.db.query(User).with_for_update().filter_by(id=v.id).first()
|
|
|
|
v.username=new_name
|
|
v.name_changed_utc=int(time.time())
|
|
|
|
set_nick(v, new_name)
|
|
|
|
g.db.add(v)
|
|
|
|
g.db.commit()
|
|
|
|
return redirect("/settings/profile")
|
|
|
|
@app.post("/settings/song_change")
|
|
@limiter.limit("5/day;1/second")
|
|
@auth_required
|
|
@validate_formkey
|
|
def settings_song_change(v):
|
|
song=request.values.get("song").strip()
|
|
|
|
if song == "" and v.song and path.isfile(f"/songs/{v.song}.mp3") and g.db.query(User.id).filter_by(song=v.song).count() == 1:
|
|
os.remove(f"/songs/{v.song}.mp3")
|
|
v.song = None
|
|
g.db.add(v)
|
|
g.db.commit()
|
|
return redirect("/settings/profile")
|
|
|
|
song = song.replace("https://music.youtube.com", "https://youtube.com")
|
|
if song.startswith(("https://www.youtube.com/watch?v=", "https://youtube.com/watch?v=", "https://m.youtube.com/watch?v=")):
|
|
id = song.split("v=")[1]
|
|
elif song.startswith("https://youtu.be/"):
|
|
id = song.split("https://youtu.be/")[1]
|
|
else:
|
|
if not v or v.oldsite: template = ''
|
|
else: template = 'CHRISTMAS/'
|
|
return render_template(f"{template}settings_profile.html", v=v, error=f"Not a youtube link.")
|
|
|
|
if "?" in id: id = id.split("?")[0]
|
|
if "&" in id: id = id.split("&")[0]
|
|
|
|
if path.isfile(f'/songs/{id}.mp3'):
|
|
v.song = id
|
|
g.db.add(v)
|
|
g.db.commit()
|
|
return redirect("/settings/profile")
|
|
|
|
|
|
req = requests.get(f"https://www.googleapis.com/youtube/v3/videos?id={id}&key={YOUTUBE_KEY}&part=contentDetails", timeout=5).json()
|
|
duration = req['items'][0]['contentDetails']['duration']
|
|
if duration == 'P0D':
|
|
if not v or v.oldsite: template = ''
|
|
else: template = 'CHRISTMAS/'
|
|
return render_template(f"{template}settings_profile.html", v=v, error=f"Can't use a live youtube video!")
|
|
|
|
if "H" in duration:
|
|
if not v or v.oldsite: template = ''
|
|
else: template = 'CHRISTMAS/'
|
|
return render_template(f"{template}settings_profile.html", v=v, error=f"Duration of the video must not exceed 10 minutes.")
|
|
|
|
if "M" in duration:
|
|
duration = int(duration.split("PT")[1].split("M")[0])
|
|
if duration > 10:
|
|
if not v or v.oldsite: template = ''
|
|
else: template = 'CHRISTMAS/'
|
|
return render_template(f"{template}settings_profile.html", v=v, error=f"Duration of the video must not exceed 10 minutes.")
|
|
|
|
|
|
if v.song and path.isfile(f"/songs/{v.song}.mp3") and g.db.query(User.id).filter_by(song=v.song).count() == 1:
|
|
os.remove(f"/songs/{v.song}.mp3")
|
|
|
|
ydl_opts = {
|
|
'outtmpl': '/songs/%(title)s.%(ext)s',
|
|
'format': 'bestaudio/best',
|
|
'postprocessors': [{
|
|
'key': 'FFmpegExtractAudio',
|
|
'preferredcodec': 'mp3',
|
|
'preferredquality': '192',
|
|
}],
|
|
}
|
|
|
|
with youtube_dl.YoutubeDL(ydl_opts) as ydl:
|
|
try: ydl.download([f"https://youtube.com/watch?v={id}"])
|
|
except Exception as e:
|
|
print(e)
|
|
if not v or v.oldsite: template = ''
|
|
else: template = 'CHRISTMAS/'
|
|
return render_template(f"{template}settings_profile.html",
|
|
v=v,
|
|
error=f"Age-restricted videos aren't allowed.")
|
|
|
|
files = os.listdir("/songs/")
|
|
paths = [path.join("/songs/", basename) for basename in files]
|
|
songfile = max(paths, key=path.getctime)
|
|
os.rename(songfile, f"/songs/{id}.mp3")
|
|
|
|
v.song = id
|
|
g.db.add(v)
|
|
|
|
g.db.commit()
|
|
|
|
return redirect("/settings/profile")
|
|
|
|
@app.post("/settings/title_change")
|
|
@limiter.limit("1/second")
|
|
@auth_required
|
|
@validate_formkey
|
|
def settings_title_change(v):
|
|
if not v or v.oldsite: template = ''
|
|
else: template = 'CHRISTMAS/'
|
|
|
|
if v.flairchanged: abort(403)
|
|
|
|
new_name=request.values.get("title").strip()[:100].replace("𒐪","")
|
|
|
|
if new_name==v.customtitle: return render_template(f"{template}settings_profile.html", v=v, error="You didn't change anything")
|
|
|
|
v.customtitleplain = new_name
|
|
|
|
v.customtitle = filter_emojis_only(new_name)
|
|
|
|
if len(v.customtitle) < 1000:
|
|
g.db.add(v)
|
|
g.db.commit()
|
|
|
|
return redirect("/settings/profile")
|