From 54455ba0dc214f6efdb3a50d96fcfac2042ba5e0 Mon Sep 17 00:00:00 2001 From: Hykilpikonna Date: Sun, 22 Nov 2020 14:13:28 -0500 Subject: [PATCH] [+] database: Create load() --- src/database.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/src/database.py b/src/database.py index 9c5618d..a039953 100644 --- a/src/database.py +++ b/src/database.py @@ -1,4 +1,5 @@ import json +import os from src.constants import dbPath from src.utils import toJson @@ -9,12 +10,21 @@ class Database: self.users = [] self.userRequests = {} self.userStatus = {} + self.load() def save(self): f = open(dbPath, 'w') f.write(toJson(self)) f.close() + def load(self): + if os.path.isfile(dbPath): + f = open(dbPath, 'r') + database = json.loads(f.read()) + self.__dict__ = database + f.close() + print("Database loaded.") + def checkUser(self, user): if user not in self.users: self.users.append(user)