o ̿Si @sFddlZddlZddlmZddlZGdddZddZddZdS) N) MongoClientc@s"eZdZdefddZddZdS)MongoDBconnection_stringcCst||_|jj|_dSN)rclient my_databasedb)selfrr -/var/www/html/Trade-python/app/db/database.py__init__s zMongoDB.__init__cCs|jdSr)rclose)r r r r r sz MongoDB.closeN)__name__ __module__ __qualname__strr r r r r r rs rccstdd}td}td}td}td}|s>|r6|r6tj|}d|d|d |d |}nd|d |}t|}||}z |VW|jdS|jw) N MONGO_HOST localhost MONGO_USER MONGO_PASSMONGO_DB MONGO_URI mongodb://:@:27017/) osgetenvurllibparse quote_pluspymongorrr  mongo_host mongo_user mongo_pass mongo_db_name mongo_urirrr r r get_mongo_dbs$         r(cCstdd}td}td}td}td}|s td|sC|r;|r;tj|}d|d |d |d |}nd|d |}t|}||}||fS) zCreate a MongoClient + DB handle for non-request contexts. Use this for background threads / long-running jobs. Caller must close the returned client. rrrrrrzMONGO_DB is not configuredrrrr)rr ValueErrorrrr r!rr"r r r create_mongo_client_and_db,s       r*)rr!r urllib.parserrr(r*r r r r s