skorotkiewicz/rDrama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

automating marsey badges

Browse source
This commit is contained in:
Aevann1 2022-03-05 00:59:33 +02:00
parent eb9d874e66
commit d1901de786
2 changed files with 35 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

2
files/routes/admin.py
View file

@ -621,7 +621,7 @@ def badge_grant_post(v):
try: badge_id = int(request.values.get("badge_id")) try: badge_id = int(request.values.get("badge_id"))
except: abort(400) except: abort(400)
if badge_id in {94,95,96,97,98,109}: abort(403) if badge_id in {16,17,94,95,96,97,98,109}: abort(403)
if user.has_badge(badge_id): if user.has_badge(badge_id):
return render_template("admin/badge_grant.html", v=v, badge_types=badges, error="User already has that badge.") return render_template("admin/badge_grant.html", v=v, badge_types=badges, error="User already has that badge.")

37
files/routes/comments.py
View file

@ -273,20 +273,51 @@ def api_comment(v):
if not marsey_regex2.fullmatch(name): if not marsey_regex2.fullmatch(name):
return {"error": "Invalid name!"}, 403 return {"error": "Invalid name!"}, 403
if "author" in marsey: author_id = get_user(marsey["author"]).id if "author" in marsey: user = get_user(marsey["author"])
elif "author_id" in marsey: author_id = int(marsey["author_id"]) elif "author_id" in marsey: user = get_account(marsey["author_id"])
else: abort(400) else: abort(400)
existing = g.db.query(Marsey.name).filter_by(name=name).one_or_none() existing = g.db.query(Marsey.name).filter_by(name=name).one_or_none()
if existing: return {"error": "A marsey with this name already exists!"}, 403 if existing: return {"error": "A marsey with this name already exists!"}, 403
marsey = Marsey(name=marsey["name"], author_id=author_id, tags=marsey["tags"], count=0) marsey = Marsey(name=marsey["name"], author_id=user.id, tags=marsey["tags"], count=0)
g.db.add(marsey) g.db.add(marsey)
filename = f'files/assets/images/emojis/{name}.webp' filename = f'files/assets/images/emojis/{name}.webp'
copyfile(oldname, filename) copyfile(oldname, filename)
process_image(filename, 200) process_image(filename, 200)
all_by_author = g.db.query(Marsey.author_id).filter_by(author_id=user.id).count()
if all_by_author >= 10 and not user.has_badge(16):
new_badge = Badge(badge_id=16, user_id=user.id)
g.db.add(new_badge)
g.db.flush()
if v.id != user.id:
text = f"@AutoJanny has given you the following profile badge:\n\n![]({new_badge.path})\n\n{new_badge.name}"
send_notification(user.id, text)
old_badge = user.has_badge(17)
if badge: g.db.delete(badge)
if all_by_author < 10 and not user.has_badge(17):
new_badge = Badge(badge_id=17, user_id=user.id)
g.db.add(new_badge)
g.db.flush()
if v.id != user.id:
text = f"@AutoJanny has given you the following profile badge:\n\n![]({new_badge.path})\n\n{new_badge.name}"
send_notification(user.id, text)
requests.post(f'https://api.cloudflare.com/client/v4/zones/{CF_ZONE}/purge_cache', headers=CF_HEADERS, data={'files': [f"https://{request.host}/e/{name}.webp"]}, timeout=5) requests.post(f'https://api.cloudflare.com/client/v4/zones/{CF_ZONE}/purge_cache', headers=CF_HEADERS, data={'files': [f"https://{request.host}/e/{name}.webp"]}, timeout=5)
cache.delete_memoized(marsey_list) cache.delete_memoized(marsey_list)
except Exception as e: except Exception as e:
return {"error": str(e)}, 400 return {"error": str(e)}, 400
body += f"\n\n![]({image})" body += f"\n\n![]({image})"