Properly mark overridden methods with @OverRide

- Imported from typing_extensions for compatibility

Author: datalogics-kam

src/pdfrest/client.py

@@ -43,6 +43,7 @@
     ValidationError,
     field_validator,
 )
+from typing_extensions import override
 
 from .exceptions import (
     PdfRestApiError,
@@ -1980,10 +1981,12 @@ def __init__(
         )
         self._files_client = _FilesClient(self)
 
+    @override
     def __enter__(self) -> PdfRestClient:
         _ = super().__enter__()
         return self
 
+    @override
     def __exit__(self, exc_type: Any, exc: Any, traceback: Any) -> None:
         super().__exit__(exc_type, exc, traceback)
 
@@ -2409,10 +2412,12 @@ def __init__(
         )
         self._files_client = _AsyncFilesClient(self)
 
+    @override
     async def __aenter__(self) -> AsyncPdfRestClient:
         _ = await super().__aenter__()
         return self
 
+    @override
     async def __aexit__(self, exc_type: Any, exc: Any, traceback: Any) -> None:
         await super().__aexit__(exc_type, exc, traceback)
 

src/pdfrest/exceptions.py

@@ -5,6 +5,7 @@
 from typing import Any
 
 import httpx
+from typing_extensions import override
 
 __all__ = (
     "PdfRestApiError",
@@ -55,6 +56,7 @@ def __init__(
         detail = message or f"pdfRest API returned status code {status_code}"
         super().__init__(detail)
 
+    @override
     def __str__(self) -> str:  # pragma: no cover - mirrors Exception.__str__
         base = super().__str__()
         if self.response_content is None:

src/pdfrest/models/public.py

@@ -16,6 +16,7 @@
     HttpUrl,
 )
 from pydantic_core import CoreSchema
+from typing_extensions import override
 
 __all__ = ("PdfRestErrorResponse", "PdfRestFile", "PdfRestFileID", "UpResponse")
 
@@ -55,9 +56,11 @@ def __new__(cls, value: str) -> PdfRestFileID:
             raise ValueError(msg)
         return str.__new__(cls, value.lower())
 
+    @override
     def __str__(self) -> str:
         return str.__str__(self)
 
+    @override
     def __repr__(self) -> str:
         return f"{self.__class__.__name__}({super().__repr__()})"
 

tests/test_client.py

@@ -7,6 +7,7 @@
 
 import httpx
 import pytest
+from typing_extensions import override
 
 from pdfrest import (
     AsyncPdfRestClient,
@@ -51,10 +52,12 @@ class NonSeekableByteStream(BytesIO):
     def __init__(self, payload: bytes) -> None:
         super().__init__(payload)
 
+    @override
     def seek(self, *args: Any, **kwargs: Any) -> int:
         msg = "non-seekable"
         raise UnsupportedOperation(msg)
 
+    @override
     def tell(self) -> int:
         msg = "non-seekable"
         raise UnsupportedOperation(msg)

tests/test_files.py

@@ -12,6 +12,7 @@
 import httpx
 import pytest
 import pytest_asyncio
+from typing_extensions import override
 
 from pdfrest import AsyncPdfRestClient, PdfRestClient
 from pdfrest.models import PdfRestFile, PdfRestFileID
@@ -26,12 +27,14 @@ def __init__(self, payload: bytes) -> None:
         self._payload = payload
         self._consumed = False
 
+    @override
     def __iter__(self) -> Iterator[bytes]:
         if self._consumed:
             return iter(())
         self._consumed = True
         return iter((self._payload,))
 
+    @override
     def close(self) -> None:  # pragma: no cover - trivial
         ...
 
@@ -41,11 +44,13 @@ def __init__(self, payload: bytes) -> None:
         self._payload = payload
         self._consumed = False
 
+    @override
     async def __aiter__(self):
         if not self._consumed:
             self._consumed = True
             yield self._payload
 
+    @override
     async def aclose(self) -> None:  # pragma: no cover - trivial
         ...