From 8b46c0c6b20a028b4a1d8b043f9f421d7c402cf7 Mon Sep 17 00:00:00 2001 From: osiris account Date: Sun, 12 Mar 2023 12:29:29 -0700 Subject: [PATCH] add db async methods --- token-scanner-api/src/server/database.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/token-scanner-api/src/server/database.py b/token-scanner-api/src/server/database.py index 40abe4e..f0ceb13 100644 --- a/token-scanner-api/src/server/database.py +++ b/token-scanner-api/src/server/database.py @@ -4,12 +4,7 @@ import pymongo - - -env_vars = os_utils.load_config() -url = env_vars['MONGODB_URL'] -db_name = env_vars['MONGODB_DB_NAME'] -collection = env_vars['MONGODB_COLLECTION_NAME'] +from src.utils import os_utils ############################################ @@ -19,8 +14,13 @@ collection = env_vars['MONGODB_COLLECTION_NAME'] def _get_db_collection(): """Connect to the database.""" + env_vars = os_utils.load_config() + url = env_vars['MONGODB_URL'] + db_name = env_vars['MONGODB_DB_NAME'] + collection = env_vars['MONGODB_COLLECTION_NAME'] + client = pymongo.MongoClient(url) - database = client.balances + database = client[db_name] return database[collection]