Added "requests__examples/adapters.py"

Author: gil9red

requests__examples/adapters.py

@@ -0,0 +1,92 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+__author__ = 'ipetrash'
+
+
+from urllib.parse import urlparse
+from random import randint
+
+import requests
+from requests.adapters import BaseAdapter, HTTPAdapter
+
+
+class FileAdapter(BaseAdapter):
+    def send(
+        self,
+        request: requests.PreparedRequest,
+        stream: bool = False,
+        timeout: None | float | tuple[float, float] | tuple[float, None] = None,
+        verify: bool | str = True,
+        cert: None | bytes | str | tuple[bytes | str, bytes | str] = None,
+        proxies: dict[str, str] | None = None,
+    ) -> requests.Response:
+        file_name = urlparse(request.url).path
+
+        rs = requests.Response()
+        rs.status_code = 200
+        rs.raw = open(file_name, 'rb')
+        rs.request = request
+
+        return rs
+
+    def close(self) -> None:
+        pass
+
+
+class DuckDuckGoAdapter(HTTPAdapter):
+    def send(
+        self,
+        request: requests.PreparedRequest,
+        stream: bool = False,
+        timeout: None | float | tuple[float, float] | tuple[float, None] = None,
+        verify: bool | str = True,
+        cert: None | bytes | str | tuple[bytes | str, bytes | str] = None,
+        proxies: dict[str, str] | None = None,
+    ) -> requests.Response:
+        query = urlparse(request.path_url).path
+
+        url = f'https://api.duckduckgo.com/?q={query}&format=json'
+        return requests.get(url)  # TODO:
+
+
+class RandomColorAdapter(BaseAdapter):
+    def send(
+        self,
+        request: requests.PreparedRequest,
+        stream: bool = False,
+        timeout: None | float | tuple[float, float] | tuple[float, None] = None,
+        verify: bool | str = True,
+        cert: None | bytes | str | tuple[bytes | str, bytes | str] = None,
+        proxies: dict[str, str] | None = None,
+    ) -> requests.Response:
+        r, g, b = map(lambda x: hex(x)[2:].zfill(2), (randint(0, 255), randint(0, 255), randint(0, 255)))
+
+        rs = requests.Response()
+        rs.status_code = 200
+        rs._content = f'#{r}{g}{b}'.encode()
+        rs.request = request
+
+        return rs
+
+    def close(self) -> None:
+        pass
+
+
+s = requests.session()
+s.mount('file:', FileAdapter())
+s.mount('duckduckgo:', DuckDuckGoAdapter())
+s.mount('randomcolor:', RandomColorAdapter())
+
+file_name = __file__
+rs = s.get('file:' + file_name)
+print(rs.content)
+# b"#!/usr/bin/env python3\r\n# -*- coding: utf-8 -*-\r\n\r\n__author__ = 'ipetrash' ...
+
+rs = s.get('duckduckgo:' + 'Dark Souls')
+print(rs.json())
+# {'Abstract': "Dark Souls is a series of action role-playing games created by Hidet ...
+
+rs = s.get('randomcolor:')
+print(rs.content)
+# b'#535ebd'