MarseyWorld/files/routes/settings.py

828 lines
27 KiB
Python
Raw Normal View History

2022-05-04 23:09:46 +00:00
from __future__ import unicode_literals
from files.helpers.alerts import *
from files.helpers.sanitize import *
from files.helpers.const import *
from files.helpers.regex import *
from files.helpers.actions import *
from files.helpers.useractions import *
2022-06-24 13:19:53 +00:00
from files.helpers.get import *
2022-10-06 22:59:50 +00:00
from files.helpers.security import *
2022-05-04 23:09:46 +00:00
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 shutil import copyfile
import requests
@app.get("/settings")
@auth_required
def settings(v):
return redirect("/settings/personal")
@app.get("/settings/personal")
@auth_required
def settings_personal(v):
return render_template("settings/personal.html", v=v)
2022-05-04 23:09:46 +00:00
@app.delete('/settings/background')
@limiter.limit(DEFAULT_RATELIMIT_SLOWER)
2022-11-06 12:08:39 +00:00
@auth_required
@ratelimit_user()
def remove_background(v):
if v.background:
v.background = None
g.db.add(v)
2022-05-04 23:09:46 +00:00
return {"message": "Background removed!"}
@app.post("/settings/personal")
@limiter.limit(DEFAULT_RATELIMIT_SLOWER)
2022-05-04 23:09:46 +00:00
@auth_required
@ratelimit_user()
def settings_personal_post(v):
2022-05-04 23:09:46 +00:00
updated = False
2022-11-07 03:28:27 +00:00
# begin common selectors #
def update_flag(column_name:str, request_name:str):
2022-11-07 02:03:09 +00:00
if not request.values.get(request_name, ''): return False
request_flag = request.values.get(request_name, '') == 'true'
if request_flag != getattr(v, column_name):
setattr(v, column_name, request_flag)
return True
return False
def update_potentially_permanent_flag(column_name:str, request_name:str, friendly_name:str, badge_id:Optional[int]):
if not request.values.get(request_name): return False
current_value = getattr(v, column_name)
if FEATURES['USERS_PERMANENT_WORD_FILTERS'] and current_value > 1:
abort(403, f"Cannot change the {friendly_name} setting after you've already set it permanently!")
request_flag = int(request.values.get(request_name, '') == 'true')
if current_value and request_flag and request.values.get("permanent", '') == 'true' and request.values.get("username") == v.username:
if v.client: abort(403, f"Cannot set {friendly_name} permanently from the API")
request_flag = int(time.time())
setattr(v, column_name, request_flag)
if badge_id: badge_grant(v, badge_id)
return render_template("settings/personal.html", v=v, msg=f"You have set the {friendly_name} permanently! Enjoy your new badge!")
elif current_value != request_flag:
setattr(v, column_name, request_flag)
return True
return False
2022-11-07 03:28:27 +00:00
def set_selector_option(column_name:str, api_name:str, valid_values:Iterable[str], error_msg:str="value"):
opt = request.values.get(api_name)
if opt: opt = opt.strip()
if not opt: return False
if opt in valid_values:
setattr(v, column_name, opt)
return True
abort(400, f"'{opt}' is not a valid {error_msg}")
# end common selectors #
background = request.values.get("background", v.background)
2022-11-07 03:50:38 +00:00
if background != v.background and background.endswith(".webp") and len(background) <= 20:
2022-05-04 23:09:46 +00:00
v.background = request.values.get("background")
2022-11-07 03:28:27 +00:00
updated = True
2022-05-04 23:09:46 +00:00
elif request.values.get("reddit", v.reddit) != v.reddit:
reddit = request.values.get("reddit")
if reddit in {'old.reddit.com', 'reddit.com', 'i.reddit.com', 'teddit.net', 'libredd.it', 'unddit.com'}:
updated = True
v.reddit = reddit
2022-10-25 16:07:40 +00:00
elif request.values.get("poor", v.poor) != v.poor:
2022-05-29 07:08:28 +00:00
updated = True
2022-11-07 03:28:27 +00:00
v.poor = request.values.get("poor", v.poor) == 'true'
2022-10-25 16:07:40 +00:00
session['poor'] = v.poor
slur_filter_updated = updated or update_potentially_permanent_flag("slurreplacer", "slurreplacer", "slur replacer", 192)
if isinstance(slur_filter_updated, bool):
updated = slur_filter_updated
else:
g.db.add(v)
return slur_filter_updated
profanity_filter_updated = updated or update_potentially_permanent_flag("profanityreplacer", "profanityreplacer", "profanity replacer", 190)
if isinstance(profanity_filter_updated, bool):
updated = profanity_filter_updated
else:
g.db.add(v)
return profanity_filter_updated
updated = updated or update_flag("hidevotedon", "hidevotedon")
updated = updated or update_flag("cardview", "cardview")
updated = updated or update_flag("highlightcomments", "highlightcomments")
updated = updated or update_flag("newtab", "newtab")
updated = updated or update_flag("newtabexternal", "newtabexternal")
updated = updated or update_flag("nitter", "nitter")
updated = updated or update_flag("imginn", "imginn")
updated = updated or update_flag("controversial", "controversial")
updated = updated or update_flag("sigs_disabled", "sigs_disabled")
2022-11-07 05:45:09 +00:00
updated = updated or update_flag("over_18", "over_18")
updated = updated or update_flag("is_private", "private")
if not updated and request.values.get("spider", v.spider) != v.spider and v.spider <= 1:
2022-09-25 02:44:19 +00:00
updated = True
v.spider = int(request.values.get("spider") == 'true')
if v.spider: badge_grant(user=v, badge_id=179)
else:
badge = v.has_badge(179)
2022-09-25 02:44:19 +00:00
if badge: g.db.delete(badge)
elif not updated and request.values.get("bio") == "":
2022-05-04 23:09:46 +00:00
v.bio = None
v.bio_html = None
g.db.add(v)
return render_template("settings/personal.html", v=v, msg="Your bio has been updated.")
2022-05-04 23:09:46 +00:00
elif not updated and request.values.get("sig") == "":
2022-05-04 23:09:46 +00:00
v.sig = None
v.sig_html = None
g.db.add(v)
return render_template("settings/personal.html", v=v, msg="Your sig has been updated.")
2022-05-04 23:09:46 +00:00
elif not updated and request.values.get("friends") == "":
2022-05-04 23:09:46 +00:00
v.friends = None
v.friends_html = None
g.db.add(v)
return render_template("settings/personal.html", v=v, msg="Your friends list has been updated.")
2022-05-04 23:09:46 +00:00
elif not updated and request.values.get("enemies") == "":
2022-05-04 23:09:46 +00:00
v.enemies = None
v.enemies_html = None
g.db.add(v)
return render_template("settings/personal.html", v=v, msg="Your enemies list has been updated.")
2022-05-04 23:09:46 +00:00
elif not updated and v.patron and request.values.get("sig"):
2022-07-17 17:24:58 +00:00
sig = request.values.get("sig")[:200].replace('\n','').replace('\r','')
2022-05-04 23:09:46 +00:00
sig_html = sanitize(sig)
if len(sig_html) > 1000:
return render_template("settings/personal.html",
2022-09-04 23:15:37 +00:00
v=v,
error="Your sig is too long")
2022-05-04 23:09:46 +00:00
v.sig = sig[:200]
v.sig_html=sig_html
g.db.add(v)
return render_template("settings/personal.html",
2022-09-04 23:15:37 +00:00
v=v,
msg="Your sig has been updated.")
2022-05-04 23:09:46 +00:00
elif not updated and FEATURES['USERS_PROFILE_BODYTEXT'] and request.values.get("friends"):
2022-05-04 23:09:46 +00:00
friends = request.values.get("friends")[:500]
friends_html = sanitize(friends)
if len(friends_html) > 2000:
return render_template("settings/personal.html",
2022-09-04 23:15:37 +00:00
v=v,
error="Your friends list is too long")
2022-05-04 23:09:46 +00:00
notify_users = NOTIFY_USERS(friends, v)
if notify_users:
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)
return render_template("settings/personal.html",
2022-09-04 23:15:37 +00:00
v=v,
msg="Your friends list has been updated.")
2022-05-04 23:09:46 +00:00
elif not updated and FEATURES['USERS_PROFILE_BODYTEXT'] and request.values.get("enemies"):
2022-05-04 23:09:46 +00:00
enemies = request.values.get("enemies")[:500]
enemies_html = sanitize(enemies)
if len(enemies_html) > 2000:
return render_template("settings/personal.html",
2022-09-04 23:15:37 +00:00
v=v,
error="Your enemies list is too long")
2022-05-04 23:09:46 +00:00
notify_users = NOTIFY_USERS(enemies, v)
if notify_users:
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)
return render_template("settings/personal.html",
2022-09-04 23:15:37 +00:00
v=v,
msg="Your enemies list has been updated.")
2022-05-04 23:09:46 +00:00
elif not updated and FEATURES['USERS_PROFILE_BODYTEXT'] and \
(request.values.get("bio") or request.files.get('file')):
2022-05-04 23:09:46 +00:00
bio = request.values.get("bio")[:1500]
bio += process_files()
2022-05-22 10:26:59 +00:00
bio = bio.strip()
2022-05-04 23:09:46 +00:00
bio_html = sanitize(bio)
if len(bio_html) > 10000:
return render_template("settings/personal.html",
2022-09-04 23:15:37 +00:00
v=v,
error="Your bio is too long")
2022-05-04 23:09:46 +00:00
if len(bio_html) > 10000: abort(400)
v.bio = bio[:1500]
v.bio_html=bio_html
g.db.add(v)
return render_template("settings/personal.html",
2022-09-04 23:15:37 +00:00
v=v,
msg="Your bio has been updated.")
2022-05-04 23:09:46 +00:00
frontsize = request.values.get("frontsize")
if frontsize:
frontsize = int(frontsize)
if frontsize in PAGE_SIZES:
v.frontsize = frontsize
2022-05-04 23:09:46 +00:00
updated = True
cache.delete_memoized(frontlist)
else: abort(400)
2022-11-07 03:28:27 +00:00
updated = updated or set_selector_option("defaultsortingcomments", "defaultsortingcomments", COMMENT_SORTS, "comment sort")
updated = updated or set_selector_option("defaultsorting", "defaultsorting", SORTS, "post sort")
updated = updated or set_selector_option("defaulttime", "defaulttime", TIME_FILTERS, "time filter")
2022-05-04 23:09:46 +00:00
theme = request.values.get("theme")
2022-11-07 03:28:27 +00:00
if not updated and theme:
if theme in THEMES:
2022-05-04 23:09:46 +00:00
if theme == "transparent" and not v.background:
abort(409, "You need to set a background to use the transparent theme")
2022-05-04 23:09:46 +00:00
v.theme = theme
if theme == "win98": v.themecolor = "30409f"
updated = True
else: abort(400, f"{theme} is not a valid theme")
2022-05-04 23:09:46 +00:00
house = request.values.get("house")
2022-11-07 03:28:27 +00:00
if not updated and house and house in HOUSES and FEATURES['HOUSES']:
2022-08-27 02:57:19 +00:00
if v.bite: abort(403)
if v.house:
if v.house.replace(' Founder', '') == house: abort(409, f"You're already in House {house}")
cost = HOUSE_SWITCH_COST
else:
cost = HOUSE_JOIN_COST
2022-05-04 23:09:46 +00:00
2022-11-01 05:25:19 +00:00
success = v.charge_account('coins', cost)
if not success:
success = v.charge_account('procoins', cost)
if not success: abort(403)
2022-05-04 23:09:46 +00:00
2022-09-17 12:12:30 +00:00
if house == "None": house = ''
2022-05-04 23:09:46 +00:00
v.house = house
if v.house == "Vampire":
send_repeatable_notification(DAD_ID, f"@{v.username} has joined House Vampire!")
updated = True
if updated:
g.db.add(v)
return {"message": "Your settings have been updated."}
else:
abort(400, "You didn't change anything.")
2022-05-04 23:09:46 +00:00
@app.post("/settings/filters")
@auth_required
def filters(v):
filters=request.values.get("filters")[:1000].strip()
if filters == v.custom_filter_list:
return render_template("settings/advanced.html", v=v, error="You didn't change anything")
2022-05-04 23:09:46 +00:00
v.custom_filter_list=filters
g.db.add(v)
return render_template("settings/advanced.html", v=v, msg="Your custom filters have been updated.")
2022-05-04 23:09:46 +00:00
def set_color(v:User, attr:str, color:Optional[str]):
current = getattr(v, attr)
color = color.strip().lower() if color else None
if color:
if color.startswith('#'): color = color[1:]
if not color_regex.fullmatch(color):
return render_template("settings/personal.html", v=v, error="Invalid color hex code")
if color and current != color:
setattr(v, attr, color)
g.db.add(v)
return redirect("/settings/personal")
2022-05-04 23:09:46 +00:00
@app.post("/settings/namecolor")
@limiter.limit(DEFAULT_RATELIMIT_SLOWER)
2022-05-04 23:09:46 +00:00
@auth_required
@ratelimit_user()
2022-05-04 23:09:46 +00:00
def namecolor(v):
return set_color(v, "namecolor", request.values.get("namecolor"))
2022-05-04 23:09:46 +00:00
@app.post("/settings/themecolor")
@limiter.limit(DEFAULT_RATELIMIT_SLOWER)
2022-05-04 23:09:46 +00:00
@auth_required
@ratelimit_user()
2022-05-04 23:09:46 +00:00
def themecolor(v):
return set_color(v, "themecolor", request.values.get("themecolor"))
2022-05-04 23:09:46 +00:00
@app.post("/settings/gumroad")
@limiter.limit(DEFAULT_RATELIMIT_SLOWER)
2022-05-04 23:09:46 +00:00
@auth_required
@ratelimit_user()
2022-05-04 23:09:46 +00:00
def gumroad(v):
if not (v.email and v.is_activated):
abort(400, f"You must have a verified email to verify {patron} status and claim your rewards!")
2022-05-04 23:09:46 +00:00
data = {'access_token': GUMROAD_TOKEN, 'email': v.email}
response = requests.get('https://api.gumroad.com/v2/sales', data=data, timeout=5).json()["sales"]
2022-05-04 23:09:46 +00:00
if len(response) == 0: abort(404, "Email not found")
2022-05-04 23:09:46 +00:00
2022-06-18 01:59:33 +00:00
response = [x for x in response if x['variants_and_quantity']]
2022-05-04 23:09:46 +00:00
response = response[0]
tier = tiers[response["variants_and_quantity"]]
if v.patron == tier: abort(400, f"{patron} rewards already claimed")
2022-05-04 23:09:46 +00:00
procoins = procoins_li[tier] - procoins_li[v.patron]
if procoins < 0: abort(400, f"{patron} rewards already claimed")
2022-05-04 23:09:46 +00:00
2022-09-01 19:05:37 +00:00
existing = g.db.query(User.id).filter(User.email == v.email, User.is_activated == True, User.patron >= tier).first()
if existing: abort(400, f"{patron} rewards already claimed on another account")
2022-05-04 23:09:46 +00:00
v.patron = tier
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).")
g.db.add(v)
badge_grant(badge_id=20+tier, user=v)
2022-05-04 23:09:46 +00:00
return {"message": f"{patron} rewards claimed!"}
@app.post("/settings/titlecolor")
@limiter.limit(DEFAULT_RATELIMIT_SLOWER)
2022-05-04 23:09:46 +00:00
@auth_required
@ratelimit_user()
2022-05-04 23:09:46 +00:00
def titlecolor(v):
2022-11-07 04:31:38 +00:00
return set_color(v, "titlecolor", request.values.get("titlecolor"))
2022-05-04 23:09:46 +00:00
@app.post("/settings/verifiedcolor")
@limiter.limit(DEFAULT_RATELIMIT_SLOWER)
2022-05-04 23:09:46 +00:00
@auth_required
@ratelimit_user()
2022-05-04 23:09:46 +00:00
def verifiedcolor(v):
if not v.verified: abort(403, "You don't have a checkmark")
return set_color(v, "verifiedcolor", "verifiedcolor")
2022-05-04 23:09:46 +00:00
@app.post("/settings/security")
@limiter.limit(DEFAULT_RATELIMIT_SLOWER)
2022-05-04 23:09:46 +00:00
@auth_required
@ratelimit_user()
2022-05-04 23:09:46 +00:00
def settings_security_post(v):
if request.values.get("new_password"):
if request.values.get("new_password") != request.values.get("cnf_password"):
return render_template("settings/security.html", v=v, error="Passwords do not match.")
2022-05-04 23:09:46 +00:00
if not valid_password_regex.fullmatch(request.values.get("new_password")):
return render_template("settings/security.html", v=v, error="Password must be between 8 and 100 characters.")
2022-05-04 23:09:46 +00:00
if not v.verifyPass(request.values.get("old_password")):
return render_template("settings/security.html", v=v, error="Incorrect password")
2022-05-04 23:09:46 +00:00
2022-10-06 22:59:50 +00:00
v.passhash = hash_password(request.values.get("new_password"))
2022-05-04 23:09:46 +00:00
g.db.add(v)
return render_template("settings/security.html", v=v, msg="Your password has been changed.")
2022-05-04 23:09:46 +00:00
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.")
2022-05-04 23:09:46 +00:00
new_email = request.values.get("new_email","").strip().lower()
if new_email == v.email:
return render_template("settings/security.html", v=v, error="This email is already yours!")
2022-05-04 23:09:46 +00:00
url = f"{SITE_FULL}/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,
2022-09-04 23:15:37 +00:00
subject="Verify your email address.",
html=render_template("email/email_change.html",
action_url=link,
v=v)
)
2022-05-04 23:09:46 +00:00
2022-11-09 17:58:32 +00:00
return render_template("settings/security.html", v=v, msg="Check your email and click the verification link to complete the email change.")
2022-05-04 23:09:46 +00:00
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.")
2022-05-04 23:09:46 +00:00
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.")
2022-05-04 23:09:46 +00:00
v.mfa_secret = secret
g.db.add(v)
return render_template("settings/security.html", v=v, msg="Two-factor authentication enabled.")
2022-05-04 23:09:46 +00:00
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.")
2022-05-04 23:09:46 +00:00
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.")
2022-05-04 23:09:46 +00:00
v.mfa_secret = None
g.db.add(v)
return render_template("settings/security.html", v=v, msg="Two-factor authentication disabled.")
2022-05-04 23:09:46 +00:00
@app.post("/settings/log_out_all_others")
@limiter.limit(DEFAULT_RATELIMIT_SLOWER)
2022-05-04 23:09:46 +00:00
@auth_required
@ratelimit_user()
2022-05-04 23:09:46 +00:00
def settings_log_out_others(v):
submitted_password = request.values.get("password", "").strip()
if not v.verifyPass(submitted_password):
return render_template("settings/security.html", v=v, error="Incorrect Password"), 401
2022-05-04 23:09:46 +00:00
v.login_nonce += 1
session["login_nonce"] = v.login_nonce
g.db.add(v)
return render_template("settings/security.html", v=v, msg="All other devices have been logged out")
2022-05-04 23:09:46 +00:00
@app.post("/settings/images/profile")
@limiter.limit(DEFAULT_RATELIMIT_SLOWER)
2022-05-04 23:09:46 +00:00
@auth_required
@ratelimit_user()
2022-05-04 23:09:46 +00:00
def settings_images_profile(v):
if request.headers.get("cf-ipcountry") == "T1": abort(403, "Image uploads are not allowed through TOR.")
2022-05-04 23:09:46 +00:00
file = request.files["profile"]
name = f'/images/{time.time()}'.replace('.','') + '.webp'
file.save(name)
2022-10-06 07:30:38 +00:00
highres = process_image(name, patron=v.patron)
2022-05-04 23:09:46 +00:00
if not highres: abort(400)
name2 = name.replace('.webp', 'r.webp')
copyfile(name, name2)
2022-06-18 20:41:00 +00:00
imageurl = process_image(name2, resize=100)
2022-05-04 23:09:46 +00:00
if not imageurl: abort(400)
if v.highres and '/images/' in v.highres:
fpath = '/images/' + v.highres.split('/images/')[1]
if path.isfile(fpath): os.remove(fpath)
if v.profileurl and '/images/' in v.profileurl:
fpath = '/images/' + v.profileurl.split('/images/')[1]
if path.isfile(fpath): os.remove(fpath)
v.highres = highres
v.profileurl = imageurl
g.db.add(v)
return render_template("settings/personal.html", v=v, msg="Profile picture successfully updated.")
2022-05-04 23:09:46 +00:00
@app.post("/settings/images/banner")
@limiter.limit(DEFAULT_RATELIMIT_SLOWER)
2022-05-04 23:09:46 +00:00
@auth_required
@ratelimit_user()
@feature_required('USERS_PROFILE_BANNER')
2022-05-04 23:09:46 +00:00
def settings_images_banner(v):
if request.headers.get("cf-ipcountry") == "T1": abort(403, "Image uploads are not allowed through TOR.")
2022-05-04 23:09:46 +00:00
file = request.files["banner"]
name = f'/images/{time.time()}'.replace('.','') + '.webp'
file.save(name)
2022-10-06 07:30:38 +00:00
bannerurl = process_image(name, patron=v.patron)
2022-05-04 23:09:46 +00:00
if bannerurl:
if v.bannerurl and '/images/' in v.bannerurl:
fpath = '/images/' + v.bannerurl.split('/images/')[1]
if path.isfile(fpath): os.remove(fpath)
v.bannerurl = bannerurl
g.db.add(v)
return render_template("settings/personal.html", v=v, msg="Banner successfully updated.")
2022-05-04 23:09:46 +00:00
@app.get("/settings/css")
@auth_required
def settings_css_get(v):
return render_template("settings/css.html", v=v)
2022-05-04 23:09:46 +00:00
@app.post("/settings/css")
@limiter.limit(DEFAULT_RATELIMIT_SLOWER)
2022-05-04 23:09:46 +00:00
@auth_required
@ratelimit_user()
2022-05-04 23:09:46 +00:00
def settings_css(v):
if v.agendaposter: abort(400, "Agendapostered users can't edit CSS!")
2022-11-08 03:40:11 +00:00
css = request.values.get("css", v.css).strip().replace('\\', '').strip()[:4000]
2022-07-01 21:24:58 +00:00
if '</style' in css.lower():
abort(400, "Please message @Aevann if you get this error")
2022-05-04 23:09:46 +00:00
v.css = css
g.db.add(v)
return render_template("settings/css.html", v=v)
2022-05-04 23:09:46 +00:00
@app.post("/settings/profilecss")
@limiter.limit(DEFAULT_RATELIMIT_SLOWER)
2022-05-04 23:09:46 +00:00
@auth_required
@ratelimit_user()
2022-05-04 23:09:46 +00:00
def settings_profilecss(v):
2022-11-08 03:40:11 +00:00
profilecss = request.values.get("profilecss", v.profilecss).strip().replace('\\', '').strip()[:4000]
valid, error = validate_css(profilecss)
if not valid:
return render_template("settings/css.html", error=error, v=v)
2022-05-04 23:09:46 +00:00
v.profilecss = profilecss
g.db.add(v)
2022-11-08 02:53:57 +00:00
return redirect('/settings/css')
@app.get("/settings/security")
@auth_required
def settings_security(v):
return render_template("settings/security.html",
v=v,
mfa_secret=pyotp.random_base32() if not v.mfa_secret else None,
now=int(time.time())
)
2022-05-04 23:09:46 +00:00
@app.post("/settings/block")
@limiter.limit("1/second;20/day")
@limiter.limit("1/second;20/day", key_func=lambda:f'{SITE}-{session.get("lo_user")}')
2022-05-04 23:09:46 +00:00
@auth_required
def settings_block_user(v):
user = get_user(request.values.get("username"), graceful=True)
if not user: abort(404, "This user doesn't exist.")
2022-05-04 23:09:46 +00:00
if user.unblockable:
if not v.shadowbanned:
send_notification(user.id, f"@{v.username} has tried to block you and failed because of your unblockable status!")
2022-11-12 10:11:46 +00:00
abort(403, f"@{user.username} is unblockable!")
2022-05-04 23:09:46 +00:00
if user.id == v.id: abort(400, "You can't block yourself")
2022-11-12 10:11:46 +00:00
if user.id == AUTOJANNY_ID: abort(403, f"You can't block @{user.username}")
if v.has_blocked(user): abort(409, f"You have already blocked @{user.username}")
2022-05-04 23:09:46 +00:00
new_block = UserBlock(user_id=v.id, target_id=user.id)
2022-05-04 23:09:46 +00:00
g.db.add(new_block)
if user.admin_level >= PERMS['USER_BLOCKS_VISIBLE']:
send_notification(user.id, f"@{v.username} has blocked you!")
2022-05-04 23:09:46 +00:00
cache.delete_memoized(frontlist)
return {"message": f"@{user.username} blocked."}
@app.post("/settings/unblock")
@limiter.limit(DEFAULT_RATELIMIT_SLOWER)
2022-05-04 23:09:46 +00:00
@auth_required
@ratelimit_user()
2022-05-04 23:09:46 +00:00
def settings_unblock_user(v):
user = get_user(request.values.get("username"))
x = v.has_blocked(user)
if not x: abort(409, "You can't unblock someone you haven't blocked")
2022-05-04 23:09:46 +00:00
g.db.delete(x)
if not v.shadowbanned and user.admin_level >= PERMS['USER_BLOCKS_VISIBLE']:
send_notification(user.id, f"@{v.username} has unblocked you!")
2022-05-04 23:09:46 +00:00
cache.delete_memoized(frontlist)
return {"message": f"@{user.username} unblocked."}
@app.get("/settings/apps")
@auth_required
def settings_apps(v):
return render_template("settings/apps.html", v=v)
2022-05-04 23:09:46 +00:00
@app.get("/settings/advanced")
2022-05-04 23:09:46 +00:00
@auth_required
def settings_advanced_get(v):
return render_template("settings/advanced.html", v=v)
2022-05-04 23:09:46 +00:00
@app.post("/settings/name_change")
@limiter.limit(DEFAULT_RATELIMIT_SLOWER)
2022-05-04 23:09:46 +00:00
@is_not_permabanned
@ratelimit_user()
2022-05-04 23:09:46 +00:00
def settings_name_change(v):
new_name=request.values.get("name").strip()
if new_name==v.username:
return render_template("settings/personal.html",
2022-09-04 23:15:37 +00:00
v=v,
error="You didn't change anything")
2022-05-04 23:09:46 +00:00
if not valid_username_regex.fullmatch(new_name):
return render_template("settings/personal.html",
2022-09-04 23:15:37 +00:00
v=v,
error="This isn't a valid username.")
2022-05-04 23:09:46 +00:00
search_name = new_name.replace('\\', '').replace('_','\_').replace('%','')
2022-09-04 23:15:37 +00:00
x = g.db.query(User).filter(
2022-05-04 23:09:46 +00:00
or_(
User.username.ilike(search_name),
User.original_username.ilike(search_name)
)
).one_or_none()
if x and x.id != v.id:
return render_template("settings/personal.html",
2022-09-04 23:15:37 +00:00
v=v,
error=f"Username `{new_name}` is already in use.")
2022-05-04 23:09:46 +00:00
2022-06-24 13:19:53 +00:00
v=get_account(v.id)
2022-05-04 23:09:46 +00:00
v.username=new_name
v.name_changed_utc=int(time.time())
g.db.add(v)
return redirect("/settings/personal")
2022-05-24 23:26:50 +00:00
@app.post("/settings/song_change_mp3")
@limiter.limit("3/second;10/day")
2022-07-13 18:14:37 +00:00
@limiter.limit("3/second;10/day", key_func=lambda:f'{SITE}-{session.get("lo_user")}')
2022-05-24 23:26:50 +00:00
@auth_required
@feature_required('USERS_PROFILE_SONG')
2022-05-24 23:26:50 +00:00
def settings_song_change_mp3(v):
file = request.files['file']
if file.content_type != 'audio/mpeg':
return render_template("settings/personal.html", v=v, error="Not a valid MP3 file")
2022-05-24 23:26:50 +00:00
2022-07-10 15:43:27 +00:00
song = str(time.time()).replace('.','')
name = f'/songs/{song}.mp3'
2022-05-24 23:26:50 +00:00
file.save(name)
size = os.stat(name).st_size
if size > 8 * 1024 * 1024:
os.remove(name)
return render_template("settings/personal.html", v=v, error="MP3 file must be smaller than 8MB")
2022-05-24 23:26:50 +00:00
2022-07-10 15:43:27 +00:00
if path.isfile(f"/songs/{v.song}.mp3") and g.db.query(User).filter_by(song=v.song).count() == 1:
os.remove(f"/songs/{v.song}.mp3")
v.song = song
2022-05-24 23:26:50 +00:00
g.db.add(v)
return redirect("/settings/personal")
2022-05-24 23:26:50 +00:00
2022-05-04 23:09:46 +00:00
@app.post("/settings/song_change")
2022-05-17 16:16:56 +00:00
@limiter.limit("3/second;10/day")
2022-07-13 18:14:37 +00:00
@limiter.limit("3/second;10/day", key_func=lambda:f'{SITE}-{session.get("lo_user")}')
2022-05-04 23:09:46 +00:00
@auth_required
2022-10-12 04:06:11 +00:00
@feature_required('USERS_PROFILE_SONG')
2022-05-04 23:09:46 +00:00
def settings_song_change(v):
song=request.values.get("song").strip()
if song == "" and v.song:
2022-05-09 11:21:49 +00:00
if path.isfile(f"/songs/{v.song}.mp3") and g.db.query(User).filter_by(song=v.song).count() == 1:
2022-05-04 23:09:46 +00:00
os.remove(f"/songs/{v.song}.mp3")
v.song = None
g.db.add(v)
return redirect("/settings/personal")
2022-05-04 23:09:46 +00:00
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:
return render_template("settings/personal.html", v=v, error="Not a youtube link.")
2022-05-04 23:09:46 +00:00
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)
return redirect("/settings/personal")
2022-05-04 23:09:46 +00:00
req = requests.get(f"https://www.googleapis.com/youtube/v3/videos?id={id}&key={YOUTUBE_KEY}&part=contentDetails", timeout=5).json()
2022-05-04 23:09:46 +00:00
duration = req['items'][0]['contentDetails']['duration']
if duration == 'P0D':
return render_template("settings/personal.html", v=v, error="Can't use a live youtube video!")
2022-05-04 23:09:46 +00:00
if "H" in duration:
return render_template("settings/personal.html", v=v, error="Duration of the video must not exceed 15 minutes.")
2022-05-04 23:09:46 +00:00
if "M" in duration:
duration = int(duration.split("PT")[1].split("M")[0])
if duration > 15:
return render_template("settings/personal.html", v=v, error="Duration of the video must not exceed 15 minutes.")
2022-05-04 23:09:46 +00:00
2022-05-09 11:21:49 +00:00
if v.song and path.isfile(f"/songs/{v.song}.mp3") and g.db.query(User).filter_by(song=v.song).count() == 1:
2022-05-04 23:09:46 +00:00
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:
2022-05-22 22:45:04 +00:00
print(e, flush=True)
return render_template("settings/personal.html",
2022-09-04 23:15:37 +00:00
v=v,
error="Age-restricted videos aren't allowed.")
2022-05-04 23:09:46 +00:00
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)
return redirect("/settings/personal")
2022-05-04 23:09:46 +00:00
@app.post("/settings/title_change")
@limiter.limit(DEFAULT_RATELIMIT_SLOWER)
2022-05-04 23:09:46 +00:00
@auth_required
@ratelimit_user()
2022-05-04 23:09:46 +00:00
def settings_title_change(v):
if v.flairchanged: abort(403)
customtitleplain = sanitize_settings_text(request.values.get("title"), 100)
2022-08-26 22:01:36 +00:00
if customtitleplain == v.customtitleplain:
return render_template("settings/personal.html", v=v, error="You didn't change anything")
2022-05-04 23:09:46 +00:00
customtitle = filter_emojis_only(customtitleplain)
customtitle = censor_slurs(customtitle, None)
2022-05-04 23:09:46 +00:00
2022-08-26 22:01:36 +00:00
if len(customtitle) > 1000:
return render_template("settings/personal.html", v=v, error="Flair too long!")
2022-07-12 20:09:59 +00:00
2022-08-26 22:01:36 +00:00
v.customtitleplain = customtitleplain
v.customtitle = customtitle
g.db.add(v)
2022-05-04 23:09:46 +00:00
return redirect("/settings/personal")
2022-05-04 23:09:46 +00:00
2022-07-11 16:46:08 +00:00
@app.post("/settings/pronouns_change")
@feature_required('PRONOUNS')
@limiter.limit(DEFAULT_RATELIMIT_SLOWER)
2022-07-11 16:46:08 +00:00
@auth_required
@ratelimit_user()
2022-07-11 16:46:08 +00:00
def settings_pronouns_change(v):
pronouns = sanitize_settings_text(request.values.get("pronouns"))
2022-07-11 16:46:08 +00:00
if len(pronouns) > 11:
return render_template("settings/personal.html", v=v, error="Your pronouns exceed the character limit (11 characters)")
2022-07-11 16:46:08 +00:00
if pronouns == v.pronouns:
return render_template("settings/personal.html", v=v, error="You didn't change anything.")
2022-07-11 16:46:08 +00:00
if not pronouns_regex.fullmatch(pronouns):
return render_template("settings/personal.html", v=v, error="The pronouns you entered don't match the required format.")
2022-07-11 16:46:08 +00:00
2022-08-16 16:16:04 +00:00
bare_pronouns = pronouns.lower().replace('/', '')
if 'nig' in bare_pronouns: pronouns = 'BI/POC'
elif 'fag' in bare_pronouns: pronouns = 'cute/twink'
2022-07-11 16:46:08 +00:00
v.pronouns = pronouns
g.db.add(v)
return redirect("/settings/personal")
2022-07-11 16:46:08 +00:00
2022-05-10 07:20:49 +00:00
@app.post("/settings/checkmark_text")
@limiter.limit(DEFAULT_RATELIMIT_SLOWER)
2022-05-10 07:20:49 +00:00
@auth_required
@ratelimit_user()
2022-05-10 07:20:49 +00:00
def settings_checkmark_text(v):
if not v.verified: abort(403)
new_name = sanitize_settings_text(request.values.get("checkmark-text"), 100)
2022-05-10 07:20:49 +00:00
if not new_name: abort(400)
if new_name == v.verified: return render_template("settings/personal.html", v=v, error="You didn't change anything")
2022-05-10 07:20:49 +00:00
v.verified = new_name
g.db.add(v)
return redirect("/settings/personal")