diff --git a/bot/dbm_defs.py b/bot/dbm_defs.py new file mode 100644 index 0000000..c51fc1b --- /dev/null +++ b/bot/dbm_defs.py @@ -0,0 +1,14 @@ +import dbm + +DB_FILE = 'data.dbm' + +def add_rec(key, value): + with dbm.open(DB_FILE, 'c') as db: + db[str(key)] = str(value) + +def get_rec(key): + with dbm.open(DB_FILE, 'r') as db: + value = db.get(str(key)) + if value is None: + raise KeyError(f"Key '{key}' not found in the database.") + return int(value) \ No newline at end of file diff --git a/bot/memcached_def.py b/bot/memcached_def.py deleted file mode 100644 index 74008a1..0000000 --- a/bot/memcached_def.py +++ /dev/null @@ -1,12 +0,0 @@ -import os -from pymemcache.client import base - - -client = base.Client((os.getenv('MEMCACHED_HOST', 'localhost'), os.getenv('MEMCACHED_PORT', 11211))) - -def add_rec(key, value): - client.set(str(key), value) - -def get_rec(key): - value = client.get(str(key)) - return (int(value)) \ No newline at end of file