settings for mongodb with auth

Albert-Gao · Mar 26, 2020 · b84ac40 · b84ac40
1 parent 7e1d4c4
commit b84ac40
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/settings.py b/settings.py
@@ -1,5 +1,7 @@
 MONGO_HOST = 'localhost'
 MONGO_PORT = 27017
+MONGO_USER = ''
+MONGO_PASSWORD = ''
 DB_NAME = 'WorldCities'
 
 CITY_COLLECTION_NAME = 'cities'

diff --git a/utils/mongodb.py b/utils/mongodb.py
@@ -1,8 +1,14 @@
 from pymongo import MongoClient
 from utils.parser import cities, countries, cities_dict, countries_dict
+from urllib.parse import quote_plus
 import settings
 
-MONGO_CLIENT = MongoClient(settings.MONGO_HOST, settings.MONGO_PORT)
+if settings.MONGO_USER and settings.MONGO_PASSWORD:
+    URI = "mongodb://%s:%s@%s:%s/%s" % (quote_plus(settings.MONGO_USER), quote_plus(settings.MONGO_PASSWORD), settings.MONGO_HOST, settings.MONGO_PORT, settings.DB_NAME)
+else:
+    URI = "mongodb://%s:%s/%s" % (settings.MONGO_HOST, settings.MONGO_PORT, settings.DB_NAME)
+
+MONGO_CLIENT = MongoClient(URI)
 DB = MONGO_CLIENT[settings.DB_NAME]
 CITY_COLLECTION = DB[settings.CITY_COLLECTION_NAME]
 COUNTRY_COLLECTION = DB[settings.COUNTRY_COLLECTION_NAME]