using httpx instead of requests

uktrade · Aug 24, 2023 · 8a7948e · 8a7948e
1 parent 8388e94
commit 8a7948e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/prometheus/Dockerfile b/prometheus/Dockerfile
@@ -26,7 +26,7 @@ RUN \
 	chown -R prometheus:prometheus /prometheus && \
 	apk del .build-deps && \
 	pip3 install \
-		requests==2.31.0
+		httpx==0.24.1
 
 COPY prometheus.yml /etc/prometheus/
 COPY fetch_targets.py /

diff --git a/prometheus/fetch_targets.py b/prometheus/fetch_targets.py
@@ -7,6 +7,7 @@
 import sys
 import urllib.parse
 
+import httpx
 
 async def async_main(logger, target_file, url, username, password):
     while True:
@@ -17,7 +18,8 @@ async def async_main(logger, target_file, url, username, password):
                 b"Authorization": b"Basic "
                 + base64.b64encode(f"{username}:{password}".encode("ascii"))
             }
-            response = requests.get(url, headers=headers, stream=True)
+            async with httpx.AsyncClient() as client:
+                response = await client.get(url, headers=headers)
             response.raise_for_status()
             logger.debug("Code %s", response.status_code)
             raw_json = response.json()