forked from MarseyWorld/MarseyWorld
101 lines
2.9 KiB
Python
101 lines
2.9 KiB
Python
from flask import g
|
|
import time
|
|
from files.helpers.alerts import send_repeatable_notification
|
|
from files.helpers.const import *
|
|
from files.helpers.discord import remove_role
|
|
from files.classes.badges import Badge
|
|
from files.classes.user import User
|
|
|
|
def award_timers(v, bot=False):
|
|
now = time.time()
|
|
dirty = False
|
|
|
|
def notify_if_not_bot(msg):
|
|
if not bot:
|
|
send_repeatable_notification(v.id,msg)
|
|
|
|
if v.patron_utc and v.patron_utc < now:
|
|
v.patron = 0
|
|
v.patron_utc = 0
|
|
notify_if_not_bot(f"Your {patron} status has expired!")
|
|
if not bot and v.discord_id: remove_role(v, "1")
|
|
dirty = True
|
|
if v.unban_utc and v.unban_utc < now:
|
|
v.is_banned = 0
|
|
v.unban_utc = 0
|
|
v.ban_evade = 0
|
|
v.ban_reason = None
|
|
notify_if_not_bot("You have been unbanned!")
|
|
dirty = True
|
|
if v.agendaposter and v.agendaposter != 1 and v.agendaposter < now:
|
|
v.agendaposter = 0
|
|
notify_if_not_bot("Your chud theme has expired!")
|
|
badge = v.has_badge(28)
|
|
if badge: g.db.delete(badge)
|
|
dirty = True
|
|
if v.flairchanged and v.flairchanged < now:
|
|
v.flairchanged = None
|
|
notify_if_not_bot("Your flair lock has expired. You can now change your flair!")
|
|
badge = v.has_badge(96)
|
|
if badge: g.db.delete(badge)
|
|
dirty = True
|
|
if v.marseyawarded and v.marseyawarded < now:
|
|
v.marseyawarded = None
|
|
notify_if_not_bot("Your marsey award has expired!")
|
|
badge = v.has_badge(98)
|
|
if badge: g.db.delete(badge)
|
|
dirty = True
|
|
if v.longpost and v.longpost < now:
|
|
v.longpost = None
|
|
notify_if_not_bot("Your pizzashill award has expired!")
|
|
badge = v.has_badge(97)
|
|
if badge: g.db.delete(badge)
|
|
dirty = True
|
|
if v.bird and v.bird < now:
|
|
v.bird = None
|
|
notify_if_not_bot("Your bird site award has expired!")
|
|
badge = v.has_badge(95)
|
|
if badge: g.db.delete(badge)
|
|
dirty = True
|
|
if v.progressivestack and v.progressivestack < now:
|
|
v.progressivestack = None
|
|
notify_if_not_bot("Your progressive stack has expired!")
|
|
badge = v.has_badge(94)
|
|
if badge: g.db.delete(badge)
|
|
dirty = True
|
|
if v.rehab and v.rehab < now:
|
|
v.rehab = None
|
|
notify_if_not_bot("Your rehab has finished!")
|
|
badge = v.has_badge(109)
|
|
if badge: g.db.delete(badge)
|
|
dirty = True
|
|
if v.deflector and v.deflector < now:
|
|
v.deflector = None
|
|
notify_if_not_bot("Your deflector has expired!")
|
|
dirty = True
|
|
if v.bite and v.bite < now:
|
|
v.bite = None
|
|
notify_if_not_bot("Your vampire status has ended!")
|
|
v.house = v.old_house
|
|
dirty = True
|
|
if v.earlylife and v.earlylife < now:
|
|
v.earlylife = None
|
|
notify_if_not_bot("Your earlylife status has expired!")
|
|
dirty = True
|
|
if v.owoify and v.owoify < now:
|
|
v.owoify = None
|
|
notify_if_not_bot("Your owoify status has expired!")
|
|
dirty = True
|
|
if v.marsify and v.marsify < now:
|
|
v.marsify = None
|
|
if SITE_NAME != 'rDrama': notify_if_not_bot("Your marsify status has expired!")
|
|
dirty = True
|
|
|
|
if dirty:
|
|
g.db.add(v)
|
|
|
|
def award_timers_bots_task():
|
|
accs = g.db.query(User).filter(User.id.in_(bots))
|
|
for u in accs:
|
|
award_timers(u, bot=True)
|