[shapely] Narrower types for geom_type in BaseGeometry subclasses (#15955)

Author: hamdanal

stubs/shapely/shapely/geometry/collection.pyi

@@ -1,5 +1,5 @@
 from collections.abc import Collection
-from typing import overload
+from typing import Literal, overload
 from typing_extensions import Self
 
 from .._typing import OptGeoArrayLike
@@ -18,4 +18,6 @@ class GeometryCollection(BaseMultipartGeometry[_GeoT_co]):
 
     # more precise base overrides
     @property
+    def geom_type(self) -> Literal["GeometryCollection"]: ...
+    @property
     def boundary(self) -> None: ...

stubs/shapely/shapely/geometry/linestring.pyi

@@ -35,6 +35,8 @@ class LineString(BaseGeometry):
     ) -> LineString | MultiLineString: ...
     # more precise base overrides
     @property
+    def geom_type(self) -> Literal["LineString", "LinearRing"]: ...  # LinearRing is a subclass of LineString
+    @property
     def boundary(self) -> MultiPoint: ...
     @property
     def convex_hull(self) -> LineString: ...

stubs/shapely/shapely/geometry/multilinestring.pyi

@@ -1,4 +1,5 @@
 from collections.abc import Collection
+from typing import Literal
 from typing_extensions import Self
 
 from .base import BaseMultipartGeometry
@@ -13,4 +14,6 @@ class MultiLineString(BaseMultipartGeometry[LineString]):
     def svg(self, scale_factor: float = 1.0, stroke_color: str | None = None, opacity: float | None = None) -> str: ...  # type: ignore[override]
     # more precise base overrides
     @property
+    def geom_type(self) -> Literal["MultiLineString"]: ...
+    @property
     def boundary(self) -> MultiPoint: ...

stubs/shapely/shapely/geometry/multipoint.pyi

@@ -1,4 +1,5 @@
 from collections.abc import Collection
+from typing import Literal
 from typing_extensions import Self
 
 from .base import BaseMultipartGeometry
@@ -18,4 +19,6 @@ class MultiPoint(BaseMultipartGeometry[Point]):
     def svg(self, scale_factor: float = 1.0, fill_color: str | None = None, opacity: float | None = None) -> str: ...  # type: ignore[override]
     # more precise base overrides
     @property
+    def geom_type(self) -> Literal["MultiPoint"]: ...
+    @property
     def boundary(self) -> GeometryCollection: ...  # empty geometry collection

stubs/shapely/shapely/geometry/multipolygon.pyi

@@ -1,4 +1,5 @@
 from collections.abc import Collection
+from typing import Literal
 from typing_extensions import Self
 
 from .base import BaseMultipartGeometry
@@ -20,4 +21,6 @@ class MultiPolygon(BaseMultipartGeometry[Polygon]):
     def svg(self, scale_factor: float = 1.0, fill_color: str | None = None, opacity: float | None = None) -> str: ...  # type: ignore[override]
     # more precise base overrides
     @property
+    def geom_type(self) -> Literal["MultiPolygon"]: ...
+    @property
     def boundary(self) -> MultiLineString: ...

stubs/shapely/shapely/geometry/point.pyi

@@ -1,5 +1,5 @@
 from collections.abc import Iterable
-from typing import TypeAlias, overload
+from typing import Literal, TypeAlias, overload
 from typing_extensions import Self
 
 from .._typing import ArrayLikeSeq
@@ -33,6 +33,8 @@ class Point(BaseGeometry):
     def svg(self, scale_factor: float = 1.0, fill_color: str | None = None, opacity: float | None = None) -> str: ...  # type: ignore[override]
     # more precise base overrides
     @property
+    def geom_type(self) -> Literal["Point"]: ...
+    @property
     def boundary(self) -> GeometryCollection: ...  # empty geometry collection
     @property
     def convex_hull(self) -> Point: ...

stubs/shapely/shapely/geometry/polygon.pyi

@@ -1,5 +1,5 @@
 from collections.abc import Collection
-from typing import NoReturn, TypeAlias, overload
+from typing import Literal, NoReturn, TypeAlias, overload
 from typing_extensions import Self
 
 from .base import BaseGeometry
@@ -17,6 +17,8 @@ class LinearRing(LineString):
     def __new__(self, coordinates: _ConvertibleToLinearRing | None = None) -> Self: ...
     @property
     def is_ccw(self) -> bool: ...
+    @property
+    def geom_type(self) -> Literal["LinearRing"]: ...
 
 class InteriorRingSequence:
     def __init__(self, parent: Polygon) -> None: ...
@@ -43,6 +45,8 @@ class Polygon(BaseGeometry):
     def from_bounds(cls, xmin: float, ymin: float, xmax: float, ymax: float) -> Self: ...
     # more precise base overrides
     @property
+    def geom_type(self) -> Literal["Polygon"]: ...
+    @property
     def boundary(self) -> MultiLineString: ...
 
 def orient(polygon: Polygon, sign: float = 1.0) -> Polygon: ...