Register dialect

Author: Peng Ren

MANIFEST.in

@@ -8,7 +8,6 @@ include requirements-test.txt
 
 # Include configuration files
 include pyproject.toml
-include .flake8
 
 # Exclude unnecessary files
 global-exclude *.pyc

pymongosql/sqlalchemy_mongodb/sqlalchemy_dialect.py

@@ -164,7 +164,7 @@ class PyMongoSQLDialect(default.DefaultDialect):
     Compatible with SQLAlchemy 1.4+ and 2.x versions.
     """
 
-    name = "pymongosql"
+    name = "mongodb"
     driver = "pymongosql"
 
     # Version compatibility

pyproject.toml

@@ -53,6 +53,10 @@ Repository = "https://github.com/passren/PyMongoSQL.git"
 Documentation = "https://github.com/passren/PyMongoSQL/wiki"
 "Bug Reports" = "https://github.com/passren/PyMongoSQL/issues"
 
+[project.entry-points."sqlalchemy.dialects"]
+mongodb = "pymongosql.sqlalchemy_mongodb.sqlalchemy_dialect:PyMongoSQLDialect"
+"mongodb+srv" = "pymongosql.sqlalchemy_mongodb.sqlalchemy_dialect:PyMongoSQLDialect"
+
 [tool.black]
 line-length = 120
 target-version = ['py39']

tests/test_sqlalchemy_dialect.py

@@ -60,7 +60,7 @@ def setUp(self):
 
     def test_dialect_name(self):
         """Test dialect name and driver."""
-        self.assertEqual(self.dialect.name, "pymongosql")
+        self.assertEqual(self.dialect.name, "mongodb")
         self.assertEqual(self.dialect.driver, "pymongosql")
 
     def test_dbapi(self):
@@ -312,7 +312,7 @@ def test_engine_creation(self, mock_connect):
         # This should not raise an exception
         engine = create_engine("mongodb://localhost:27017/testdb")
         self.assertIsNotNone(engine)
-        self.assertEqual(engine.dialect.name, "pymongosql")
+        self.assertEqual(engine.dialect.name, "mongodb")
 
         # Test version compatibility attributes
         if hasattr(engine.dialect, "_sqlalchemy_version"):

tests/test_sqlalchemy_integration.py

@@ -152,7 +152,7 @@ class TestSQLAlchemyIntegration:
     def test_engine_creation(self, sqlalchemy_engine):
         """Test that SQLAlchemy engine works with real MongoDB."""
         assert sqlalchemy_engine is not None
-        assert sqlalchemy_engine.dialect.name == "pymongosql"
+        assert sqlalchemy_engine.dialect.name == "mongodb"
 
         # Test that we can get a connection
         with sqlalchemy_engine.connect() as connection: