summaryrefslogtreecommitdiff
path: root/day7/db.py
diff options
context:
space:
mode:
Diffstat (limited to 'day7/db.py')
-rw-r--r--day7/db.py13
1 files changed, 7 insertions, 6 deletions
diff --git a/day7/db.py b/day7/db.py
index 11d2a45..4502bfd 100644
--- a/day7/db.py
+++ b/day7/db.py
@@ -1,21 +1,22 @@
import shelve
+from config import CONFIG_DB_PATH
def set_cookie(key, value):
- _db[key] = value
+ _cookies_db[key] = value
def get_cookie(key, default=None):
- return _db.get(key, default)
+ return _cookies_db.get(key, default)
def set_config_entry(key, value):
- _config[key] = value
+ _config_db[key] = value
def get_config_entry(key, default=None):
- return _config.get(key, default)
+ return _config_db.get(key, default)
-_db = shelve.open('db/cookies.db')
-_config = shelve.open('db/config.db')
+_config_db = shelve.open(CONFIG_DB_PATH)
+_cookies_db = shelve.open(get_config_entry('cookies_db_path'))