From a5b1e1770e54579ad7f0b62fa890b614bb8468e4 Mon Sep 17 00:00:00 2001 From: RickyHao Date: Thu, 2 Nov 2017 17:28:29 +0800 Subject: [PATCH] Fix db parameter if host url do not contain database name. --- flask_mongoengine/connection.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/flask_mongoengine/connection.py b/flask_mongoengine/connection.py index bbeda7f9..e4b1ddec 100644 --- a/flask_mongoengine/connection.py +++ b/flask_mongoengine/connection.py @@ -38,7 +38,9 @@ def _sanitize_settings(settings): uri_to_check = uri_to_check.replace('mongomock://', 'mongodb://') uri_dict = uri_parser.parse_uri(uri_to_check) - resolved_settings['db'] = uri_dict['database'] + if uri_dict['database'] is not None: + # If mongodb://... don't provide database name, don't put None to resolved_settings['db'] + resolved_settings['db'] = uri_dict['database'] # Add a default name param or use the "db" key if exists if resolved_settings.get('db'):