Добавлена обработка аргументов командной строки

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'))