ENH: Support Plugin Accessors Via Entry Points

doc/source/whatsnew/v3.0.0.rst

@@ -83,6 +83,7 @@ Other enhancements
 - Improved deprecation message for offset aliases (:issue:`60820`)
 - Multiplying two :class:`DateOffset` objects will now raise a ``TypeError`` instead of a ``RecursionError`` (:issue:`59442`)
 - Restore support for reading Stata 104-format and enable reading 103-format dta files (:issue:`58554`)
+- Support :class:`DataFrame` plugin accessor via entry points (:issue:`29076`)
 - Support passing a :class:`Iterable[Hashable]` input to :meth:`DataFrame.drop_duplicates` (:issue:`59237`)
 - Support reading Stata 102-format (Stata 1) dta files (:issue:`58978`)
 - Support reading Stata 110-format (Stata 7) dta files (:issue:`47176`)

pandas/__init__.py

@@ -346,3 +346,8 @@
     "unique",
     "wide_to_long",
 ]
+
+from pandas.core.accessor import DataFrameAccessorLoader
+
+DataFrameAccessorLoader.load()
+del DataFrameAccessorLoader

pandas/core/accessor.py

@@ -10,6 +10,7 @@
 import functools
 from typing import (
     TYPE_CHECKING,
+    Any,
     final,
 )
 import warnings
@@ -25,6 +26,8 @@
     from pandas import Index
     from pandas.core.generic import NDFrame
 
+from importlib.metadata import entry_points
+
 
 class DirNamesMixin:
     _accessors: set[str] = set()
@@ -393,3 +396,26 @@ def register_index_accessor(name: str) -> Callable[[TypeT], TypeT]:
     from pandas import Index
 
     return _register_accessor(name, Index)
+
+
+class DataFrameAccessorLoader:
+    """Loader class for registering DataFrame accessors via entry points."""
+
+    ENTRY_POINT_GROUP: str = "pandas_dataframe_accessor"
+
+    @classmethod
+    def load(cls) -> None:
+        """loads and registers accessors defined by 'pandas_dataframe_accessor'."""
+        eps = entry_points(group=cls.ENTRY_POINT_GROUP)
+
+        for ep in eps:
+            name: str = ep.name
+
+            def make_property(ep):
+                def accessor(self) -> Any:
+                    cls_ = ep.load()
+                    return cls_(self)
+
+                return accessor
+
+            register_dataframe_accessor(name)(make_property(ep))

pandas/tests/test_plugis_entrypoint_loader.py

@@ -0,0 +1,35 @@
+import pandas as pd
+from pandas.core.accessor import DataFrameAccessorLoader
+
+
+def test_load_dataframe_accessors(monkeypatch):
+    # GH29076
+    # Mocked EntryPoint to simulate a plugin
+    class MockEntryPoint:
+        name = "test_accessor"
+
+        def load(self):
+            class TestAccessor:
+                def __init__(self, df):
+                    self._df = df
+
+                def test_method(self):
+                    return "success"
+
+            return TestAccessor
+
+    # Mock entry_points
+    def mock_entry_points(*, group):
+        if group == DataFrameAccessorLoader.ENTRY_POINT_GROUP:
+            return [MockEntryPoint()]
+        return []
+
+    # Patch entry_points in the correct module
+    monkeypatch.setattr("pandas.core.accessor.entry_points", mock_entry_points)
+
+    DataFrameAccessorLoader.load()
+
+    # Create DataFrame and verify that the accessor was registered
+    df = pd.DataFrame({"a": [1, 2, 3]})
+    assert hasattr(df, "test_accessor")
+    assert df.test_accessor.test_method() == "success"