forked from kemonomimi/nyabot
Yuki Joou
95139026e5
This commit adds a data_manager submodule that handles reading/writing
data. For now it's backed by a single json file, but in the future we
should move to a sqlite3 database
The marriage system uses Discord Interactions and views, and works for
basic things
TODO:
- Poly marriage support
- Listing marriages through a command
51 lines
1.3 KiB
Python
51 lines
1.3 KiB
Python
# TODO: If you need to grow this class, consider moving to a sqlite3 database!
|
|
|
|
import io
|
|
import json
|
|
|
|
DATA_PATH = "data.json"
|
|
|
|
|
|
def __load_json_or_make_empty_object(fp: io.TextIOWrapper):
|
|
data = {}
|
|
fp.seek(0)
|
|
try:
|
|
data = json.load(fp)
|
|
except ValueError as exception:
|
|
print(exception, DATA_PATH, "wasn't a JSON file, making it one...")
|
|
# fp doesn't point to a file with valid json, replacing the content with "{}"
|
|
fp.seek(0)
|
|
fp.write("{}")
|
|
fp.truncate()
|
|
fp.flush()
|
|
return {}
|
|
return data
|
|
|
|
|
|
def get_data() -> dict:
|
|
with open(DATA_PATH, 'a+') as fp:
|
|
return __load_json_or_make_empty_object(fp)
|
|
|
|
|
|
# Helper class for writing to the data file in a sage way
|
|
class DataWriter:
|
|
def __init__(self) -> None:
|
|
self.__data_fp: io.TextIOWrapper | None = None
|
|
|
|
def __enter__(self):
|
|
self.__data_fp = open(DATA_PATH, 'w+')
|
|
return self
|
|
|
|
def __exit__(self, *_):
|
|
if self.__data_fp is not None:
|
|
self.__data_fp.close()
|
|
|
|
# Set a new value to the data
|
|
def set_data(self, json_data: dict):
|
|
if self.__data_fp is None:
|
|
return
|
|
|
|
self.__data_fp.seek(0)
|
|
json.dump(json_data, self.__data_fp)
|
|
self.__data_fp.truncate()
|
|
self.__data_fp.flush()
|