turdbot/py/bin/storage.py

10 lines
390 B
Python
Raw Normal View History

2023-12-14 18:47:29 -07:00
class storage:
2023-10-26 23:25:09 -06:00
def __init__(self, serverid, db):
self.serverdb = db[str(serverid)]
2023-01-19 21:37:39 -07:00
2023-10-26 23:25:09 -06:00
def db(self, module, key, value=None):
2024-03-12 15:59:59 -06:00
try : r = self.serverdb.find_one({"module": module}).get(key)
except AttributeError: r = None
2023-10-26 23:25:09 -06:00
if value is not None:
2024-03-12 15:59:59 -06:00
self.serverdb.update_one({"module": module}, {"$set": {key: value}}, upsert=True)
return r