Adding the database name

python/create_database.py

 #!/usr/bin/env python3
 import mongo_connect
 
-# Create a client connection to the MongoDB database.
+# Get a client connection to the MongoDB database.
 client = mongo_connect.get_client()
 
 # Create a new database.
-db = client['sales']
+db = client[mongo_connect.get_database_name()]
 
 # Create a collection.
 customers = db['Customers']

python/mongo_connect.py

@@ -43,3 +43,10 @@ def get_client():
         return None
 
     return client
+
+
+def get_database_name():
+    """
+    A function to return the MongoDB database name.
+    """
+    return os.getenv('MONGODB_DB', 'test')