-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathuser.py
47 lines (37 loc) · 1.2 KB
/
user.py
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
import util, json
class User:
users = {}
def __init__(self, id, ip, websocket) -> None:
self.name = "anon_" + util.generate_uid(ip)[10:]
self.id = id
self.raw_ip = ip
self.uid = util.generate_uid(ip)
self.websocket = websocket
self.room = None
self.logged_in = False
self.mod = False
self.owner = False
def setMod(self, boolean: bool):
self.mod = boolean
def setOwner(self, boolean: bool):
self.owner = boolean
def setRoom(self, name: str):
self.room = name
def setName(self, name: str):
self.name = name
def setLoggedIn(self, boolean: bool):
self.logged_in = boolean
def set(self):
self.users[self.id] = self
def get(self, id: str):
if id in self.users:
return self.users[id]
def remove(self, id: str = None):
if id:
if id in self.users:
del self.users[id]
else:
del self.users[self.id]
def write(self, **commands):
data = json.dumps(commands).encode()
self.websocket.sendMessage(data)