Git-Mirrors/blockchain-data-engineering-toolkit
1
0
Fork 0
mirror of https://github.com/autistic-symposium/blockchain-data-engineering-toolkit.git synced 2025-04-25 10:19:13 -04:00
Code Issues Packages Projects Releases Wiki Activity

add db async methods

Browse Source
This commit is contained in:
osiris account 2023-03-12 12:29:29 -07:00
parent 2d346fec48
commit 8b46c0c6b2
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
token-scanner-api/src/server/database.py
View File

@ -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]