YasinOnm08/interstellar_ai
0
0
Fork 0
forked from React-Group/interstellar_ai
Code Pull requests Activity

It saves data now yay.

Browse source
This commit is contained in:
Patrick_Pluto 2024-09-20 16:06:40 +02:00
parent 840654f9d1
commit 1d1705ebfa
2 changed files with 15 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
py/api.py
View file

@ -13,6 +13,7 @@ class API:
self.ai_response = {} self.ai_response = {}
self.ai = AI() self.ai = AI()
self.db = DB() self.db = DB()
self.db.load_database()
CORS(self.app) CORS(self.app)
def run(self): def run(self):

15
py/db.py
View file

@ -1,5 +1,5 @@
import json
import hashlib import hashlib
import json
class DB: class DB:
@ -26,6 +26,7 @@ class DB:
return False return False
self.database[username]['data'] = data self.database[username]['data'] = data
self.save_database()
return True return True
def update_password(self, data): def update_password(self, data):
@ -36,6 +37,7 @@ class DB:
hashed_new_password = self.hash_password(new_password) hashed_new_password = self.hash_password(new_password)
self.database[username].update({"hashed_password": hashed_new_password}) self.database[username].update({"hashed_password": hashed_new_password})
self.save_database()
return True return True
def check_credentials(self, data): def check_credentials(self, data):
@ -55,3 +57,14 @@ class DB:
send_back = self.database[username].get['data'] send_back = self.database[username].get['data']
return send_back return send_back
def save_database(self):
with open("database.json", 'w') as file:
json.dump(self.database, file)
def load_database(self):
try:
with open("database.json", 'r') as file:
self.database = json.load(file)
except FileNotFoundError:
pass