db_kyc_project/bot/dbm_defs.py

14 lines
344 B
Python
Raw Normal View History

2024-12-15 01:21:54 +03:00
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)