27 lines
896 B
Python
27 lines
896 B
Python
# sample as https://github.com/mongodb-developer/mongodb-with-fastapi/blob/master/app.py
|
|
import os
|
|
import pymongo
|
|
|
|
|
|
async def get_mongo():
|
|
if os.getenv('env') == 'dev':
|
|
|
|
mongo_info = {
|
|
"host": 'localhost',
|
|
"user": "admin",
|
|
"pswd": "kingsome",
|
|
"port": 27017,
|
|
"db": "jump"
|
|
}
|
|
mongo_client = pymongo.MongoClient(mongo_info['host'],
|
|
mongo_info['port'])
|
|
mongo_auth = mongo_client['admin']
|
|
mongo_auth.authenticate(mongo_info['user'], mongo_info['pswd'])
|
|
mongo_db = mongo_client.jump
|
|
else:
|
|
mongo_info = {"host": '10.10.5.6', "port": 27017, "db": "jump"}
|
|
mongo_client = pymongo.MongoClient(mongo_info['host'],
|
|
mongo_info['port'])
|
|
mongo_db = mongo_client.jump
|
|
return mongo_db
|