Merge pull request #248 from IlyaSkriblovsky/mongodb+srv

support for mongodb+srv:// uri scheme
twisted · Mar 25, 2019 · bece8c2 · bece8c2
2 parents db7f5f3 + 8280fd9
commit bece8c2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -11,6 +11,9 @@
     keywords=["mongo", "mongodb", "pymongo", "gridfs", "txmongo"],
     packages=["txmongo", "txmongo._gridfs"],
     install_requires=["twisted>=14.0", "pymongo>=3.0"],
+    extras_require={
+        'srv': ['pymongo[srv]>=3.6'],
+    },
     license="Apache License, Version 2.0",
     include_package_data=True,
     test_suite="nose.collector",

diff --git a/txmongo/connection.py b/txmongo/connection.py
@@ -251,7 +251,7 @@ def __init__(self, uri="mongodb://127.0.0.1:27017", pool_size=1, ssl_context_fac
         assert isinstance(pool_size, int)
         assert pool_size >= 1
 
-        if not uri.startswith("mongodb://"):
+        if not uri.startswith("mongodb://") and not uri.startswith("mongodb+srv://"):
             uri = "mongodb://" + uri
 
         self.__uri = parse_uri(uri)