24 lines
662 B
Python
24 lines
662 B
Python
import json
|
|
from .model import DatabaseModel, default_database
|
|
from modules.config import Config
|
|
import os
|
|
|
|
|
|
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 remove_item(item_name: str):
|
|
db = Database.get()
|
|
db.items.pop(item_name)
|
|
|
|
with open(Config.get().database, 'w') as f:
|
|
json.dump(db.to_dict(), f, indent=4)
|