79 lines
2.4 KiB
Python
79 lines
2.4 KiB
Python
import json
|
|
import os.path
|
|
from gui.modules.core.popup import popup
|
|
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:")
|
|
if os.path.isfile(Config.get().database):
|
|
print(open(Config.get().database).read())
|
|
with open("error.baseback", 'w') as f:
|
|
f.write(open(Config.get().database).read())
|
|
popup('Error', 'Error happened while getting database, writing default \n'
|
|
'Old database wrote to error.baseback file')
|
|
else:
|
|
print('No database file, writing new with name from config')
|
|
with open(Config.get().database, 'w') as f:
|
|
json.dump(default_database, f, indent=4)
|
|
return DatabaseModel.from_dict(default_database)
|
|
|
|
@staticmethod
|
|
def get_profile():
|
|
return Database.get().profiles[Config.get().profile]
|
|
|
|
@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)
|
|
|
|
@staticmethod
|
|
def set_owned(item_name: str):
|
|
db = Database.get()
|
|
db.profiles[Config.get().profile].owned_items.append(item_name)
|
|
|
|
Database.write(db)
|
|
|
|
@staticmethod
|
|
def set_unowned(item_name: str):
|
|
db = Database.get()
|
|
if item_name in db.profiles[Config.get().profile].owned_items:
|
|
db.profiles[Config.get().profile].owned_items.remove(item_name)
|
|
|
|
Database.write(db)
|