Fix driver registration issue

Author: Peng Ren

pymongosql/__init__.py

@@ -6,7 +6,7 @@
 if TYPE_CHECKING:
     from .connection import Connection
 
-__version__: str = "0.4.4"
+__version__: str = "0.4.5"
 
 # Globals https://www.python.org/dev/peps/pep-0249/#globals
 apilevel: str = "2.0"

pyproject.toml

@@ -58,6 +58,7 @@ Documentation = "https://github.com/passren/PyMongoSQL/wiki"
 [project.entry-points."sqlalchemy.dialects"]
 mongodb = "pymongosql.sqlalchemy_mongodb.sqlalchemy_dialect:PyMongoSQLDialect"
 "mongodb+srv" = "pymongosql.sqlalchemy_mongodb.sqlalchemy_dialect:PyMongoSQLDialect"
+"mongodb.srv" = "pymongosql.sqlalchemy_mongodb.sqlalchemy_dialect:PyMongoSQLDialect"
 
 [tool.black]
 line-length = 120

tests/test_sqlalchemy_dialect.py

@@ -658,3 +658,14 @@ def test_dialect_registration(self):
         except ImportError:
             # Skip if SQLAlchemy registry is not available
             self.skipTest("SQLAlchemy registry not available")
+
+    def test_srv_dialect_lookup(self):
+        """Test that mongodb.srv resolves correctly (mongodb+srv:// URLs)."""
+        if not HAS_SQLALCHEMY:
+            self.skipTest("SQLAlchemy not available")
+
+        from sqlalchemy.dialects import registry
+
+        loaded = registry.load("mongodb.srv")
+        self.assertIsNotNone(loaded)
+        self.assertEqual(loaded.name, "mongodb")