Switch to urllib3

Author: m-aciek

build_status.py

@@ -8,16 +8,15 @@
 import tomllib
 from collections.abc import Iterator
 
-from requests import Session
+from urllib3 import PoolManager
 
 
-def get_languages(requests: Session) -> Iterator[tuple[str, bool]]:
-    data = requests.get(
-        'https://raw.githubusercontent.com/'
-        'python/docsbuild-scripts/refs/heads/main/config.toml',
-        timeout=10,
-    ).text
-    config = tomllib.loads(data)
+def get_languages(http: PoolManager) -> Iterator[tuple[str, bool]]:
+    data = http.request(
+        'GET',
+        'https://raw.githubusercontent.com/python/docsbuild-scripts/refs/heads/main/config.toml',
+    ).data
+    config = tomllib.loads(data.decode())
     for code, language in config['languages'].items():
         language_code = code.lower().replace('_', '-')
         in_switcher = language.get('in_prod', config['defaults']['in_prod'])
@@ -26,7 +25,7 @@ def get_languages(requests: Session) -> Iterator[tuple[str, bool]]:
 
 def main() -> None:
     languages = {
-        language: in_switcher for language, in_switcher in get_languages(Session())
+        language: in_switcher for language, in_switcher in get_languages(PoolManager())
     }
     print(languages)
     for code in ('en', 'pl', 'ar', 'zh-cn', 'id'):

generate.py

@@ -16,7 +16,7 @@
 from pathlib import Path
 from tempfile import TemporaryDirectory
 
-from requests import Session
+import urllib3
 from git import Repo
 from jinja2 import Template
 
@@ -45,13 +45,14 @@ def get_completion_progress() -> Iterator['LanguageProjectData']:
         )
         subprocess.run(['make', '-C', cpython_dir / 'Doc', 'venv'], check=True)
         subprocess.run(['make', '-C', cpython_dir / 'Doc', 'gettext'], check=True)
-        languages_built = dict(build_status.get_languages(session := Session()))
+        http = urllib3.PoolManager()
+        languages_built = dict(build_status.get_languages(http))
         for language, repo in get_languages_and_repos(devguide_dir):
             built = language.code in languages_built
             if repo:
                 completion, translators_data = get_completion(clones_dir, repo)
                 visitors_num = (
-                    get_number_of_visitors(language.code, session) if built else 0
+                    get_number_of_visitors(language.code, http) if built else 0
                 )
             else:
                 completion = 0.0

visitors.py

@@ -4,20 +4,22 @@
 import urllib.parse
 import zipfile
 
-from requests import Session
+from urllib3 import PoolManager
 
 
-def get_number_of_visitors(language: str, requests: Session) -> int:
+def get_number_of_visitors(language: str, http: PoolManager) -> int:
     params = urllib.parse.urlencode(
         {'filters': f'[["contains","event:page",["/{language}/"]]]', 'period': 'all'}
     )
-    r = requests.get(
-        f'https://plausible.io/docs.python.org/export?{params}', timeout=20
-    )
-    logging.info(f'Plausible export responded with {r.status_code=}')
+    r = http.request('GET', f'https://plausible.io/docs.python.org/export?{params}')
+    logging.info(f'Plausible export responded with {r.status=}')
     with (
-        zipfile.ZipFile(io.BytesIO(r.content), 'r') as z,
+        zipfile.ZipFile(io.BytesIO(r.data), 'r') as z,
         z.open('visitors.csv') as csv_file,
     ):
         csv_reader = csv.DictReader(io.TextIOWrapper(csv_file))
         return sum(int(row['visitors']) for row in csv_reader)
+
+
+if __name__ == '__main__':
+    print(get_number_of_visitors('pl', PoolManager()))