fix/thumbnails-proxy

PnX-SI · Jun 18, 2021 · 4fb30f4 · 4fb30f4
1 parent f69cd87
commit 4fb30f4
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/apptax/taxonomie/filemanager.py b/apptax/taxonomie/filemanager.py
@@ -78,11 +78,23 @@ def url_to_image(url):
     """
         Récupération d'une image à partir d'une url
     """
-    r = requests.get(url, stream=True)
+
+    r = requests.get(
+        url,
+        stream=True,
+        proxies={
+            "http": current_app.config.get('HTTP_PROXY'),
+            "https": current_app.config.get('HTTPS_PROXY')
+        }
+    )
+
+    if (r.status_code >=400):
+        raise Exception("Pb with request : status code {}".format(r.status_code))
     try:
         img = Image.open(io.BytesIO(r.content))
     except IOError:
         raise Exception("Media is not an image")
+
     return img
 
 

diff --git a/config.py.sample b/config.py.sample
@@ -27,3 +27,7 @@ PASS_METHOD='hash'
 # ID APPLICATION TaxHub
 # User in the authentification submodule to avoid token conflict between app on the same server
 ID_APP = 2
+
+# PROXIES
+HTTP_PROXY = None
+HTTPS_PROXY = None