First pass at completing all stubs

Author: p-vdp

stubs/geojson/geojson/__init__.pyi

@@ -0,0 +1,28 @@
+from geojson._version import __version__, __version_info__
+from geojson.base import GeoJSON
+from geojson.codec import GeoJSONEncoder, dump, dumps, load, loads
+from geojson.feature import Feature, FeatureCollection
+from geojson.geometry import GeometryCollection, LineString, MultiLineString, MultiPoint, MultiPolygon, Point, Polygon
+from geojson.utils import coords, map_coords
+
+__all__ = [
+    "dump",
+    "dumps",
+    "load",
+    "loads",
+    "GeoJSONEncoder",
+    "coords",
+    "map_coords",
+    "Point",
+    "LineString",
+    "Polygon",
+    "MultiLineString",
+    "MultiPoint",
+    "MultiPolygon",
+    "GeometryCollection",
+    "Feature",
+    "FeatureCollection",
+    "GeoJSON",
+    "__version__",
+    "__version_info__",
+]

stubs/geojson/geojson/_version.pyi

@@ -1,4 +1,2 @@
-from _typeshed import Incomplete
-
 __version__: str
-__version_info__: Incomplete
+__version_info__: tuple[int, ...]

stubs/geojson/geojson/base.pyi

@@ -1,15 +1,16 @@
+from collections.abc import Callable, Iterable
 from typing import Any
 
 class GeoJSON(dict[str, Any]):
-    def __init__(self, iterable=(), **extra) -> None: ...
-    def __getattr__(self, name): ...
+    def __init__(self, iterable: Iterable[Any] = (), **extra) -> None: ...
+    def __getattr__(self, name) -> Any: ...
     def __setattr__(self, name, value) -> None: ...
     def __delattr__(self, name) -> None: ...
     @property
-    def __geo_interface__(self): ...
+    def __geo_interface__(self) -> None | GeoJSON: ...
     @classmethod
-    def to_instance(cls, ob, default=None, strict: bool = False): ...
+    def to_instance(cls, ob: Any, default: Any = None, strict: bool = False) -> Any: ...
     @property
-    def is_valid(self): ...
-    def check_list_errors(self, checkFunc, lst): ...
-    def errors(self) -> None: ...
+    def is_valid(self) -> bool: ...
+    def check_list_errors(self, checkFunc: Callable[..., Any], lst: list[Any]) -> list[str]: ...
+    def errors(self) -> list[str] | None: ...

stubs/geojson/geojson/codec.pyi

@@ -1,11 +1,32 @@
 import json
+from _typeshed import SupportsRead, SupportsWrite
+from collections.abc import Callable
+from typing import Any
+
+from geojson import GeoJSON
 
 class GeoJSONEncoder(json.JSONEncoder):
-    def default(self, obj): ...
+    def default(self, obj) -> GeoJSON: ...
 
-def dump(obj, fp, cls=..., allow_nan: bool = False, **kwargs): ...
-def dumps(obj, cls=..., allow_nan: bool = False, ensure_ascii: bool = False, **kwargs): ...
-def load(fp, cls=..., parse_constant=..., object_hook=..., **kwargs): ...
-def loads(s, cls=..., parse_constant=..., object_hook=..., **kwargs): ...
+def dump(
+    obj: Any, fp: SupportsWrite[str], cls: type[json.JSONEncoder] | None = json.JSONEncoder, allow_nan: bool = False, **kwargs
+) -> None: ...
+def dumps(
+    obj: Any, cls: type[json.JSONEncoder] | None = json.JSONEncoder, allow_nan: bool = False, ensure_ascii: bool = False, **kwargs
+) -> str: ...
+def load(
+    fp: SupportsRead[str],
+    cls: type[json.JSONDecoder] = json.JSONDecoder,
+    parse_constant: Callable[[Any], None] = ...,
+    object_hook: Callable[..., GeoJSON] = GeoJSON.to_instance,
+    **kwargs,
+) -> GeoJSON: ...
+def loads(
+    s: str,
+    cls: type[json.JSONDecoder] = json.JSONDecoder,
+    parse_constant: Callable[[Any], None] = ...,
+    object_hook: Callable[..., GeoJSON] = GeoJSON.to_instance,
+    **kwargs,
+) -> GeoJSON: ...
 
 PyGFPEncoder = GeoJSONEncoder

stubs/geojson/geojson/examples.pyi

@@ -1,11 +1,22 @@
-from _typeshed import Incomplete
+from typing import Any, Literal
+
+from geojson.geometry import Geometry
+
+from geojson import GeoJSON
 
 class SimpleWebFeature:
-    id: Incomplete
-    geometry: Incomplete
-    properties: Incomplete
-    def __init__(self, id=None, geometry=None, title=None, summary=None, link=None) -> None: ...
-    def as_dict(self): ...
-    __geo_interface__: Incomplete
-
-def create_simple_web_feature(o): ...
+    id: None | int | str
+    geometry: None | Geometry
+    properties: dict[Literal["title", "summary", "link"], str]
+    __geo_interface__: dict[str, Any]
+    def __init__(
+        self,
+        id: None | int | str = None,
+        geometry: None | dict[str, Any] = None,
+        title: None | str = None,
+        summary: None | str = None,
+        link: None | str = None,
+    ) -> None: ...
+    def as_dict(self) -> dict[str, Any]: ...
+
+def create_simple_web_feature(o: dict[str, Any]) -> GeoJSON: ...

stubs/geojson/geojson/feature.pyi

@@ -1,10 +1,15 @@
+from typing import Any
+
 from geojson.base import GeoJSON
+from geojson.geometry import Geometry
 
 class Feature(GeoJSON):
-    def __init__(self, id=None, geometry=None, properties=None, **extra) -> None: ...
-    def errors(self): ...
+    def __init__(
+        self, id: None | str | int = None, geometry: None | Geometry = None, properties: None | dict[str, Any] = None, **extra
+    ) -> None: ...
+    def errors(self) -> list[str] | None: ...
 
 class FeatureCollection(GeoJSON):
-    def __init__(self, features, **extra) -> None: ...
-    def errors(self): ...
-    def __getitem__(self, key): ...
+    def __init__(self, features: list[Feature | Geometry], **extra) -> None: ...
+    def errors(self) -> list[str] | None: ...
+    def __getitem__(self, key) -> Feature: ...

stubs/geojson/geojson/geometry.pyi

@@ -1,39 +1,48 @@
+from collections.abc import Iterable
+from typing import Any
+
 from geojson.base import GeoJSON
 
+from geojson import Feature
+
 DEFAULT_PRECISION: int
 
 class Geometry(GeoJSON):
-    def __init__(self, coordinates=None, validate: bool = False, precision=None, **extra) -> None: ...
+    def __init__(
+        self, coordinates: None | Feature | Iterable[Any] = None, validate: bool = False, precision: None | int = None, **extra
+    ) -> None: ...
     @classmethod
-    def clean_coordinates(cls, coords, precision): ...
+    def clean_coordinates(
+        cls, coords: Geometry | tuple[Any, ...] | list[tuple[Any, ...]], precision: None | int
+    ) -> None | tuple[Any, ...] | list[tuple[Any, ...]]: ...
 
 class GeometryCollection(GeoJSON):
-    def __init__(self, geometries=None, **extra) -> None: ...
-    def errors(self): ...
-    def __getitem__(self, key): ...
+    def __init__(self, geometries=..., **extra) -> None: ...
+    def errors(self) -> list[str] | None: ...
+    def __getitem__(self, key) -> Geometry | tuple[()] | None: ...
 
-def check_point(coord): ...
+def check_point(coord) -> str | None: ...
 
 class Point(Geometry):
-    def errors(self): ...
+    def errors(self) -> list[str] | None: ...
 
 class MultiPoint(Geometry):
-    def errors(self): ...
+    def errors(self) -> list[str] | None: ...
 
-def check_line_string(coord): ...
+def check_line_string(coord) -> str | None: ...
 
 class LineString(MultiPoint):
-    def errors(self): ...
+    def errors(self) -> list[str] | None: ...
 
 class MultiLineString(Geometry):
-    def errors(self): ...
+    def errors(self) -> list[str] | None: ...
 
-def check_polygon(coord): ...
+def check_polygon(coord) -> str | None: ...
 
 class Polygon(Geometry):
-    def errors(self): ...
+    def errors(self) -> list[str] | None: ...
 
 class MultiPolygon(Geometry):
-    def errors(self): ...
+    def errors(self) -> list[str] | None: ...
 
 class Default: ...

stubs/geojson/geojson/mapping.pyi

@@ -1,4 +1,6 @@
-GEO_INTERFACE_MARKER: str
+from typing import Any, Literal
 
-def is_mapping(obj): ...
-def to_mapping(obj): ...
+GEO_INTERFACE_MARKER: Literal["__geo_interface__"]
+
+def is_mapping(obj) -> bool: ...
+def to_mapping(obj) -> dict[str, Any]: ...

stubs/geojson/geojson/utils.pyi

@@ -1,8 +1,16 @@
-from _typeshed import Incomplete
 from collections.abc import Generator
+from typing import Any, Literal
 
-def coords(obj) -> Generator[Incomplete, Incomplete]: ...
-def map_coords(func, obj): ...
-def map_tuples(func, obj): ...
-def map_geometries(func, obj): ...
-def generate_random(featureType, numberVertices: int = 3, boundingBox=[-180.0, -90.0, 180.0, 90.0]): ...
+from geojson.geometry import Geometry
+
+from geojson import Feature, GeoJSON
+
+def coords(obj: Feature | Geometry | dict[str, Any]) -> Generator[tuple[Any]]: ...
+def map_coords(func, obj) -> dict[str, Any]: ...
+def map_tuples(func, obj) -> list[tuple[float]]: ...
+def map_geometries(func, obj) -> GeoJSON: ...
+def generate_random(
+    featureType: Literal["Point", "LineString", "Polygon"],
+    numberVertices: int = 3,
+    boundingBox: list[float] = [-180.0, -90.0, 180.0, 90.0],
+) -> Feature: ...