stubfix?

Author: evertlammerts

duckdb/_duckdb/__init__.pyi

@@ -0,0 +1,71 @@
+"""
+This module contains classes and methods related to functions and udf
+"""
+
+from __future__ import annotations
+import typing
+
+__all__: list[str] = ["ARROW", "DEFAULT", "FunctionNullHandling", "NATIVE", "PythonUDFType", "SPECIAL"]
+
+class FunctionNullHandling:
+    """
+    Members:
+
+      DEFAULT
+
+      SPECIAL
+    """
+
+    DEFAULT: typing.ClassVar[FunctionNullHandling]  # value = <FunctionNullHandling.DEFAULT: 0>
+    SPECIAL: typing.ClassVar[FunctionNullHandling]  # value = <FunctionNullHandling.SPECIAL: 1>
+    __members__: typing.ClassVar[
+        dict[str, FunctionNullHandling]
+    ]  # value = {'DEFAULT': <FunctionNullHandling.DEFAULT: 0>, 'SPECIAL': <FunctionNullHandling.SPECIAL: 1>}
+    def __eq__(self, other: typing.Any) -> bool: ...
+    def __getstate__(self) -> int: ...
+    def __hash__(self) -> int: ...
+    def __index__(self) -> int: ...
+    def __init__(self, value: typing.SupportsInt) -> None: ...
+    def __int__(self) -> int: ...
+    def __ne__(self, other: typing.Any) -> bool: ...
+    def __repr__(self) -> str: ...
+    def __setstate__(self, state: typing.SupportsInt) -> None: ...
+    def __str__(self) -> str: ...
+    @property
+    def name(self) -> str: ...
+    @property
+    def value(self) -> int: ...
+
+class PythonUDFType:
+    """
+    Members:
+
+      NATIVE
+
+      ARROW
+    """
+
+    ARROW: typing.ClassVar[PythonUDFType]  # value = <PythonUDFType.ARROW: 1>
+    NATIVE: typing.ClassVar[PythonUDFType]  # value = <PythonUDFType.NATIVE: 0>
+    __members__: typing.ClassVar[
+        dict[str, PythonUDFType]
+    ]  # value = {'NATIVE': <PythonUDFType.NATIVE: 0>, 'ARROW': <PythonUDFType.ARROW: 1>}
+    def __eq__(self, other: typing.Any) -> bool: ...
+    def __getstate__(self) -> int: ...
+    def __hash__(self) -> int: ...
+    def __index__(self) -> int: ...
+    def __init__(self, value: typing.SupportsInt) -> None: ...
+    def __int__(self) -> int: ...
+    def __ne__(self, other: typing.Any) -> bool: ...
+    def __repr__(self) -> str: ...
+    def __setstate__(self, state: typing.SupportsInt) -> None: ...
+    def __str__(self) -> str: ...
+    @property
+    def name(self) -> str: ...
+    @property
+    def value(self) -> int: ...
+
+ARROW: PythonUDFType  # value = <PythonUDFType.ARROW: 1>
+DEFAULT: FunctionNullHandling  # value = <FunctionNullHandling.DEFAULT: 0>
+NATIVE: PythonUDFType  # value = <PythonUDFType.NATIVE: 0>
+SPECIAL: FunctionNullHandling  # value = <FunctionNullHandling.SPECIAL: 1>

duckdb/_duckdb/functional.pyi

@@ -0,0 +1,106 @@
+"""
+This module contains classes and methods related to typing
+"""
+
+from __future__ import annotations
+import duckdb
+import typing
+
+__all__: list[str] = [
+    "BIGINT",
+    "BIT",
+    "BLOB",
+    "BOOLEAN",
+    "DATE",
+    "DOUBLE",
+    "DuckDBPyType",
+    "FLOAT",
+    "HUGEINT",
+    "INTEGER",
+    "INTERVAL",
+    "SMALLINT",
+    "SQLNULL",
+    "TIME",
+    "TIMESTAMP",
+    "TIMESTAMP_MS",
+    "TIMESTAMP_NS",
+    "TIMESTAMP_S",
+    "TIMESTAMP_TZ",
+    "TIME_TZ",
+    "TINYINT",
+    "UBIGINT",
+    "UHUGEINT",
+    "UINTEGER",
+    "USMALLINT",
+    "UTINYINT",
+    "UUID",
+    "VARCHAR",
+]
+
+class DuckDBPyType:
+    @typing.overload
+    def __eq__(self, other: DuckDBPyType) -> bool:
+        """
+        Compare two types for equality
+        """
+    @typing.overload
+    def __eq__(self, other: str) -> bool:
+        """
+        Compare two types for equality
+        """
+    def __getattr__(self, name: str) -> DuckDBPyType:
+        """
+        Get the child type by 'name'
+        """
+    def __getitem__(self, name: str) -> DuckDBPyType:
+        """
+        Get the child type by 'name'
+        """
+    def __hash__(self) -> int: ...
+    @typing.overload
+    def __init__(self, type_str: str, connection: typing.Optional[duckdb.DuckDBPyConnection] = None) -> None:
+        """
+        Construct a DuckDBPyType from a type name and optional connection.
+        """
+
+    @typing.overload
+    def __init__(self, obj: object) -> None:
+        """
+        Construct a DuckDBPyType from a Python type (types.*) or object.
+        """
+    def __repr__(self) -> str:
+        """
+        Stringified representation of the type object
+        """
+    @property
+    def children(self) -> list: ...
+    @property
+    def id(self) -> str: ...
+
+BIGINT: DuckDBPyType  # value = BIGINT
+BIT: DuckDBPyType  # value = BIT
+BLOB: DuckDBPyType  # value = BLOB
+BOOLEAN: DuckDBPyType  # value = BOOLEAN
+DATE: DuckDBPyType  # value = DATE
+DOUBLE: DuckDBPyType  # value = DOUBLE
+FLOAT: DuckDBPyType  # value = FLOAT
+HUGEINT: DuckDBPyType  # value = HUGEINT
+INTEGER: DuckDBPyType  # value = INTEGER
+INTERVAL: DuckDBPyType  # value = INTERVAL
+SMALLINT: DuckDBPyType  # value = SMALLINT
+SQLNULL: DuckDBPyType  # value = "NULL"
+TIME: DuckDBPyType  # value = TIME
+TIMESTAMP: DuckDBPyType  # value = TIMESTAMP
+TIMESTAMP_MS: DuckDBPyType  # value = TIMESTAMP_MS
+TIMESTAMP_NS: DuckDBPyType  # value = TIMESTAMP_NS
+TIMESTAMP_S: DuckDBPyType  # value = TIMESTAMP_S
+TIMESTAMP_TZ: DuckDBPyType  # value = TIMESTAMP WITH TIME ZONE
+TIME_TZ: DuckDBPyType  # value = TIME WITH TIME ZONE
+TINYINT: DuckDBPyType  # value = TINYINT
+UBIGINT: DuckDBPyType  # value = UBIGINT
+UHUGEINT: DuckDBPyType  # value = UHUGEINT
+UINTEGER: DuckDBPyType  # value = UINTEGER
+USMALLINT: DuckDBPyType  # value = USMALLINT
+UTINYINT: DuckDBPyType  # value = UTINYINT
+UUID: DuckDBPyType  # value = UUID
+VARCHAR: DuckDBPyType  # value = VARCHAR

duckdb/_duckdb/typing.pyi

@@ -297,6 +297,27 @@ dev = [ # tooling like uv will install this automatically when syncing the envir
 # - pytest
 ######################################################################################################
 
+[tool.mypy]
+packages = ["duckdb"]
+strict = true
+warn_unreachable = true
+pretty = true
+python_version = "3.9"
+exclude = [
+    "duckdb/experimental/",
+    "duckdb/functional/",
+    "duckdb/query_graph/",
+    "duckdb/typing/",
+    "duckdb/value/",
+]
+
+[[tool.mypy.overrides]]
+module = [
+#    "some_untyped_dependency.*",
+#    "another_untyped_lib"
+]
+ignore_missing_imports = true
+
 [tool.pytest.ini_options]
 minversion = "6.0"
 addopts = "-ra -q"