Add admina base
Signed-off-by: Nikolaos Karaolidis <nick@karaolidis.com>
This commit is contained in:
15
src/util/database.py
Normal file
15
src/util/database.py
Normal file
@@ -0,0 +1,15 @@
|
||||
import pymongo
|
||||
from os import environ
|
||||
|
||||
MONGO_HOST = environ.get("MONGO_HOST")
|
||||
MONGO_PORT = environ.get("MONGO_PORT")
|
||||
MONGO_USERNAME = environ.get("MONGO_USERNAME")
|
||||
MONGO_PASSWORD = environ.get("MONGO_PASSWORD")
|
||||
MONGO_DATABASE = environ.get("MONGO_DATABASE")
|
||||
|
||||
if not MONGO_HOST or not MONGO_PORT or not MONGO_USERNAME or not MONGO_PASSWORD or not MONGO_DATABASE:
|
||||
raise ValueError(
|
||||
"MONGO_HOST, MONGO_PORT, MONGO_USERNAME, MONGO_PASSWORD and MONGO_DATABASE must be set")
|
||||
|
||||
database = pymongo.MongoClient(MONGO_HOST, int(
|
||||
MONGO_PORT), username=MONGO_USERNAME, password=MONGO_PASSWORD, authSource=MONGO_DATABASE)[MONGO_DATABASE].client
|
Reference in New Issue
Block a user