feat: PEP 544 – Protocols: Structural subtyping (static duck typing)

Liora Milbaum · Liora Milbaum · commit 0fde7ca981d7 · 2023-01-24T06:59:33.000+02:00
diff --git a/.github/workflows/protocol.yml b/.github/workflows/protocol.yml
@@ -0,0 +1,20 @@
+name: 'Protocaol'
+
+on:
+  push:
+    branches:
+      - main
+    paths:
+      - gitlab/protocols/**
+  pull_request:
+    branches:
+      - main
+    paths:
+      - gitlab/protocols/**
+
+jobs:
+  protocol:
+    runs-on: ubuntu-latest
+    steps:
+    - name: Fail
+      run: /bin/false
diff --git a/gitlab/http_backends/requests_backend.py b/gitlab/http_backends/requests_backend.py
@@ -6,8 +6,11 @@
 from requests.structures import CaseInsensitiveDict
 from requests_toolbelt.multipart.encoder import MultipartEncoder  # type: ignore
 
+from gitlab.protocols.http_backend import HttpBackend
+from gitlab.protocols.http_backend_response import HttpBackendResponse
 
-class RequestsResponse:
+
+class RequestsResponse(HttpBackendResponse):
     def __init__(self, response: requests.Response) -> None:
         self._response: requests.Response = response
 
@@ -35,7 +38,7 @@ def json(self) -> Any:
         return self._response.json()
 
 
-class RequestsBackend:
+class RequestsBackend(HttpBackend):
     def __init__(self, session: Optional[requests.Session] = None) -> None:
         self._client: requests.Session = session or requests.Session()
 
diff --git a/gitlab/protocols/__init__.py b/gitlab/protocols/__init__.py
@@ -0,0 +1,3 @@
+"""
+Defines protocols
+"""
diff --git a/gitlab/protocols/http_backend.py b/gitlab/protocols/http_backend.py
@@ -0,0 +1,29 @@
+import sys
+from abc import abstractmethod
+from typing import Any, Dict, Optional, Union
+
+from requests_toolbelt.multipart.encoder import MultipartEncoder  # type: ignore
+
+from gitlab.protocols.http_backend_response import HttpBackendResponse
+
+if sys.version_info >= (3, 8):
+    from typing import Protocol
+else:
+    from typing_extensions import Protocol
+
+
+class HttpBackend(Protocol):
+    @abstractmethod
+    def http_request(
+        self,
+        method: str,
+        url: str,
+        json: Optional[Union[Dict[str, Any], bytes]],
+        data: Optional[Union[Dict[str, Any], MultipartEncoder]],
+        params: Optional[Any],
+        timeout: Optional[float],
+        verify: Optional[Union[bool, str]],
+        stream: Optional[bool],
+        **kwargs: Any,
+    ) -> HttpBackendResponse:
+        ...
diff --git a/gitlab/protocols/http_backend_response.py b/gitlab/protocols/http_backend_response.py
@@ -0,0 +1,15 @@
+import sys
+from abc import abstractmethod
+
+import requests
+
+if sys.version_info >= (3, 8):
+    from typing import Protocol
+else:
+    from typing_extensions import Protocol
+
+
+class HttpBackendResponse(Protocol):
+    @abstractmethod
+    def __init__(self, response: requests.Response) -> None:
+        ...
