34 lines
915 B
Python
34 lines
915 B
Python
import json
|
|
from modules.database.model import DatabaseModel, default_database, Item
|
|
from modules.config import Config
|
|
|
|
|
|
class Database:
|
|
@staticmethod
|
|
def get():
|
|
try:
|
|
return DatabaseModel.from_dict(json.load(open(Config.get().database)))
|
|
except:
|
|
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 remove_item(item_name: str):
|
|
db = Database.get()
|
|
db.items.pop(item_name)
|
|
|
|
Database.write(db)
|
|
|
|
@staticmethod
|
|
def add_item(item: Item):
|
|
db = Database.get()
|
|
db.items[item.item_name] = item
|
|
|
|
Database.write(db)
|