# 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