add namespace protocol

Author: MarcoGorelli

spec/API_specification/dataframe_api/_types.py

@@ -3,18 +3,22 @@
 """
 from __future__ import annotations
 
-from dataclasses import dataclass
 from typing import (
+    TYPE_CHECKING,
     Any,
     List,
     Literal,
+    Mapping,
     Optional,
+    Protocol,
     Sequence,
     Tuple,
     Union,
     TYPE_CHECKING,
 )
-from enum import Enum
+
+if TYPE_CHECKING:
+    from .dataframe_object import DataFrame
 
 if TYPE_CHECKING:
     from .dtypes import (
@@ -41,6 +45,100 @@
 NullType = Any
 
 
+class Namespace(Protocol):
+    __dataframe_api_version__: str
+
+    class DataFrame:
+        ...
+
+    class Column:
+        ...
+
+    class Int64:
+        ...
+
+    class Int32:
+        ...
+
+    class Int16:
+        ...
+
+    class Int8:
+        ...
+
+    class UInt64:
+        ...
+
+    class UInt32:
+        ...
+
+    class UInt16:
+        ...
+
+    class UInt8:
+        ...
+
+    class Float64:
+        ...
+
+    class Float32:
+        ...
+
+    class Bool:
+        ...
+
+    @staticmethod
+    def concat(dataframes: Sequence[DataFrame]) -> DataFrame:
+        ...
+
+    @staticmethod
+    def column_from_sequence(
+        sequence: Sequence[Any],
+        *,
+        dtype: Any,
+        name: str = "",
+        api_version: str | None = None,
+    ) -> Column:
+        ...
+
+    @staticmethod
+    def dataframe_from_dict(
+        data: Mapping[str, Column], *, api_version: str | None = None
+    ) -> DataFrame:
+        ...
+
+    @staticmethod
+    def column_from_1d_array(
+        array: Any, *, dtype: Any, name: str = "", api_version: str | None = None
+    ) -> Column:
+        ...
+
+    @staticmethod
+    def dataframe_from_2d_array(
+        array: Any,
+        *,
+        names: Sequence[str],
+        dtypes: Mapping[str, Any],
+        api_version: str | None = None,
+    ) -> DataFrame:
+        ...
+
+    @staticmethod
+    def is_null(value: object, /) -> bool:
+        ...
+
+    @staticmethod
+    def is_dtype(dtype: Any, kind: str | tuple[str, ...]) -> bool:
+        ...
+
+
+class SupportsDataFrameAPI(Protocol):
+    def __dataframe_consortium_standard__(
+        self, *, api_version: str | None = None
+    ) -> DataFrame:
+        ...
+
+
 __all__ = [
     "Any",
     "DataFrame",
@@ -58,5 +156,4 @@
     "device",
     "DType",
     "ellipsis",
-    "Enum",
 ]

spec/API_specification/dataframe_api/dataframe_object.py

@@ -6,7 +6,8 @@
 if TYPE_CHECKING:
     from .column_object import Column
     from .groupby_object import GroupBy
-    from ._types import NullType, Scalar, DType
+    from . import Bool
+    from ._types import NullType, Scalar, Namespace, DType
 
 
 __all__ = ["DataFrame"]
@@ -36,7 +37,7 @@ class DataFrame:
     **Methods and Attributes**
 
     """
-    def __dataframe_namespace__(self) -> Any:
+    def __dataframe_namespace__(self) -> Namespace:
         """
         Returns an object that has all the top-level dataframe API functions on it.