File size: 2,249 Bytes
f45efbd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 |
import threading
from datetime import datetime
from sqlalchemy import BigInteger, Boolean, Column, DateTime, UnicodeText
from Database.sql import BASE, SESSION
class AFK(BASE):
__tablename__ = "afk_user"
user_id = Column(BigInteger, primary_key=True)
is_afk = Column(Boolean)
reason = Column(UnicodeText)
time = Column(DateTime)
def __init__(self, user_id: int, reason: str = "", is_afk: bool = True):
self.user_id = user_id
self.reason = reason
self.is_afk = is_afk
self.time = datetime.now()
def __repr__(self):
return "afk_status for {}".format(self.user_id)
AFK.__table__.create(checkfirst=True)
INSERTION_LOCK = threading.RLock()
AFK_USERS = {}
def is_afk(user_id):
return user_id in AFK_USERS
def check_afk_status(user_id):
try:
return SESSION.query(AFK).get(user_id)
finally:
SESSION.close()
def set_afk(user_id, reason=""):
with INSERTION_LOCK:
curr = SESSION.query(AFK).get(user_id)
if not curr:
curr = AFK(user_id, reason, True)
else:
curr.is_afk = True
AFK_USERS[user_id] = {"reason": reason, "time": curr.time}
SESSION.add(curr)
SESSION.commit()
def rm_afk(user_id):
with INSERTION_LOCK:
curr = SESSION.query(AFK).get(user_id)
if curr:
if user_id in AFK_USERS: # sanity check
del AFK_USERS[user_id]
SESSION.delete(curr)
SESSION.commit()
return True
SESSION.close()
return False
def toggle_afk(user_id, reason=""):
with INSERTION_LOCK:
curr = SESSION.query(AFK).get(user_id)
if not curr:
curr = AFK(user_id, reason, True)
elif curr.is_afk:
curr.is_afk = False
elif not curr.is_afk:
curr.is_afk = True
SESSION.add(curr)
SESSION.commit()
def __load_afk_users():
global AFK_USERS
try:
all_afk = SESSION.query(AFK).all()
AFK_USERS = {
user.user_id: {"reason": user.reason, "time": user.time}
for user in all_afk
if user.is_afk
}
finally:
SESSION.close()
__load_afk_users()
|