From 0c00a50417220d21cdb9008fcc9f1413df08f59a Mon Sep 17 00:00:00 2001 From: iLLiCiTiT Date: Fri, 1 Oct 2021 18:25:59 +0200 Subject: [PATCH] added validation of mongo url --- openpype/lib/mongo.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/openpype/lib/mongo.py b/openpype/lib/mongo.py index 7527fce3b9..64c46eac68 100644 --- a/openpype/lib/mongo.py +++ b/openpype/lib/mongo.py @@ -185,6 +185,14 @@ class OpenPypeMongoConnection: @classmethod def create_connection(cls, mongo_url, timeout=None, retry_attempts=None): + parsed = urlparse(mongo_url) + # Force validation of scheme + if parsed.scheme not in ["mongodb", "mongodb+srv"]: + raise pymongo.errors.InvalidURI(( + "Invalid URI scheme:" + " URI must begin with 'mongodb://' or 'mongodb+srv://'" + )) + if timeout is None: timeout = int(os.environ.get("AVALON_TIMEOUT") or 1000)