61 lines
1.9 KiB
Python
61 lines
1.9 KiB
Python
import random
|
|
import base64
|
|
import json
|
|
import os
|
|
import datetime
|
|
|
|
|
|
class Model:
|
|
def __init__(self, filename = "tehmodel.json"):
|
|
self.sockets = {}
|
|
self.filename = filename
|
|
self.sessions = None
|
|
if os.path.isfile(filename):
|
|
with open(filename) as f:
|
|
try:
|
|
self.sessions = json.load(f)
|
|
except:
|
|
self.sessions = {}
|
|
else:
|
|
self.sessions = {}
|
|
|
|
def handle_post(self, data):
|
|
print("I have received P O S T data: " + str(data))
|
|
print("let me tell everyone")
|
|
for k, v in self.sockets.items():
|
|
v.send("some data was received!")
|
|
|
|
# handle post request data
|
|
|
|
def __enter__(self):
|
|
return self
|
|
|
|
def __exit__(self, exception_type, exception_value, traceback):
|
|
if not (exception_type or exception_value or traceback):
|
|
with open(self.filename, "w") as f:
|
|
json.dump(self.sessions, f)
|
|
if not os.path.isdir("backups"):
|
|
try:
|
|
os.mkdir("backups")
|
|
except FileExistsError:
|
|
print("backups is a file, no directory. Please delete yourself")
|
|
datestring = datetime.datetime.strftime(datetime.datetime.now(), "%Y-%m-%d-%H%M%S")
|
|
with open(f"backups/{datestring}_{self.filename}", "w") as f:
|
|
json.dump(self.sessions, f)
|
|
|
|
|
|
def create_session(self) -> str:
|
|
sessionname = base64.b32encode(bytearray(random.randint(0, 0xFF) for _ in range(10)))[:16].decode().lower()
|
|
self.sessions[sessionname] = {}
|
|
return sessionname
|
|
|
|
def subscribe(self, clientid, socket):
|
|
self.sockets[clientid] = socket
|
|
# todo subscribe socket, match with id
|
|
|
|
|
|
def unsubscribe(self, socket):
|
|
for k,v in self.sockets.items():
|
|
if v == socket:
|
|
self.sockets.pop(k)
|