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

Liora Milbaum · Liora Milbaum · commit 3feb3b7ec884 · 2022-12-26T06:56:18.000+02:00
diff --git a/gitlab/client.py b/gitlab/client.py
@@ -1,5 +1,7 @@
 """Wrapper for the GitLab API."""
 
+from __future__ import annotations
+
 import os
 import re
 import time
@@ -13,6 +15,7 @@
 import gitlab.const
 import gitlab.exceptions
 from gitlab import http_backends, utils
+from gitlab.protocols.client import Client
 
 REDIRECT_MSG = (
     "python-gitlab detected a {status_code} ({reason!r}) redirection. You must update "
@@ -28,7 +31,7 @@
 )
 
 
-class Gitlab:
+class Gitlab(Client):
 
     """Represents a GitLab server connection.
 
diff --git a/gitlab/protocols/__init__.py b/gitlab/protocols/__init__.py
@@ -0,0 +1,3 @@
+"""
+Defines protocols
+"""
diff --git a/gitlab/protocols/client.py b/gitlab/protocols/client.py
@@ -0,0 +1,24 @@
+from abc import abstractmethod
+from typing import Any, Dict, Optional, Protocol, Union
+
+import requests
+
+
+class Client(Protocol):
+    @abstractmethod
+    def http_request(
+        self,
+        verb: str,
+        path: str,
+        query_data: Optional[Dict[str, Any]],
+        post_data: Optional[Union[Dict[str, Any], bytes]],
+        raw: bool,
+        streamed: bool,
+        files: Optional[Dict[str, Any]],
+        timeout: Optional[float],
+        obey_rate_limit: bool,
+        retry_transient_errors: Optional[bool],
+        max_retries: int,
+        **kwargs: Any,
+    ) -> requests.Response:
+        ...
