51 lines
1.4 KiB
Python
51 lines
1.4 KiB
Python
import json
|
|
from modules.database.model import DatabaseModel, default_database, Item, Profile
|
|
from modules.config import Config
|
|
|
|
|
|
class Database:
|
|
@staticmethod
|
|
def get():
|
|
try:
|
|
return DatabaseModel.from_dict(json.load(open(Config.get().database)))
|
|
except Exception as e:
|
|
print(f"Cannot load database: {e}. Writing default database")
|
|
print("Old data:")
|
|
print(open(Config.get().database).read())
|
|
with open(Config.get().database, 'w') as f:
|
|
json.dump(default_database, f, indent=4)
|
|
return DatabaseModel.from_dict(default_database)
|
|
|
|
@staticmethod
|
|
def write(db: DatabaseModel):
|
|
with open(Config.get().database, 'w') as f:
|
|
json.dump(db.to_dict(), f, indent=4, sort_keys=True)
|
|
|
|
@staticmethod
|
|
def add_item(item: Item):
|
|
db = Database.get()
|
|
db.items[item.item_name] = item
|
|
|
|
Database.write(db)
|
|
|
|
@staticmethod
|
|
def remove_item(item_name: str):
|
|
db = Database.get()
|
|
db.items.pop(item_name)
|
|
|
|
Database.write(db)
|
|
|
|
@staticmethod
|
|
def add_profile(profile: Profile):
|
|
db = Database.get()
|
|
db.profiles[profile.profile_name] = profile
|
|
|
|
Database.write(db)
|
|
|
|
@staticmethod
|
|
def remove_profile(profile_name: str):
|
|
db = Database.get()
|
|
db.profiles.pop(profile_name)
|
|
|
|
Database.write(db)
|