async def setmile(cls, user, guild_id: int, time: str):
session = Session()
fitness = await cls.get_user(user.lower(), guild_id)
fitness.mile = time
session.add(fitness)
session.commit()
session.close()
async def auto_remove_event(cls, guild_id: int, event_id):
session = Session()
event = session.query(CountdownEvents).filter_by(id=event_id,
guild_id=guild_id).first()
session.delete(event)
session.commit()
session.close()
async def setohp(cls, user, guild_id: int, weight: float):
session = Session()
fitness = await cls.get_user(user.lower(), guild_id)
fitness.ohp = weight
session.add(fitness)
session.commit()
session.close()
async def setgender(cls, user, guild_id: int, gender: str):
session = Session()
fitness = await cls.get_user(user.lower(), guild_id)
fitness.gender = gender.lower()
session.add(fitness)
session.commit()
session.close()
async def setburpees(cls, user, guild_id: int, count: int):
session = Session()
fitness = await cls.get_user(user.lower(), guild_id)
fitness.burpess_1m = count
session.add(fitness)
session.commit()
session.close()
async def add_event(cls, guild_id: int, event_name: str, event_date: date):
session = Session()
event = CountdownEvents()
event.guild_id = guild_id
event.event_name = event_name
event.event_date = event_date
session.add(event)
session.commit()
session.close()
async def remove_event(cls, guild_id: int, event_name: str, event_date: date):
session = Session()
event = session.query(CountdownEvents).filter_by(
guild_id=guild_id, event_name=event_name, event_date=event_date).first()
if event is not None:
session.delete(event)
session.commit()
session.close()
return True
return False
async def add_reminder(cls, guild_id, user, reminder_msg, channel,
reminder_time):
session = Session()
reminder = Reminders()
reminder.guild_id = guild_id
reminder.user = user
reminder.message = reminder_msg
reminder.channel = channel
reminder.when = reminder_time
session.add(reminder)
session.commit()
session.close()
async def ban_user(cls, user: str, banned_by: str, ban_reason, ban_time):
session = Session()
ban = session.query(Bans).filter_by(user=user).first()
if ban is None:
ban = Bans()
ban.user = user
ban.banned_by = banned_by.lower()
ban.ban_reason = ban_reason
ban.banned_on = datetime.datetime.now()
ban.ban_time = ban_time
session.add(ban)
session.commit()
return
async def remove_reminder(cls, reminder_id):
session = Session()
reminder = session.query(Reminders).filter_by(id=reminder_id).first()
session.delete(reminder)
session.commit()
session.close()
async def unban_user(cls, user: str):
session = Session()
ban = session.query(Bans).filter_by(user=user).first()
session.delete(ban)
session.commit()
return
async def get_progress(cls, user, guild_id: int):
session = Session()
fitness = await cls.get_user(user.lower(), guild_id)
session.add(fitness)
session.commit()
return fitness
