From e80ec3730edecc09f738a00b7821c34171383e92 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=D0=94=D0=B0=D0=BD=D1=8F=20=D0=92=D0=B0=D0=BA=D1=83=D0=BB?=
 =?UTF-8?q?=D0=B5=D0=BD=D0=BA=D0=BE=D0=B2?= <danyadjan@Epsilon.local>
Date: Sun, 15 Dec 2024 01:21:54 +0300
Subject: [PATCH] Add dbm defs and remove memcache

---
 bot/dbm_defs.py      | 14 ++++++++++++++
 bot/memcached_def.py | 12 ------------
 2 files changed, 14 insertions(+), 12 deletions(-)
 create mode 100644 bot/dbm_defs.py
 delete mode 100644 bot/memcached_def.py

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