fix: add timeouts to requests.get

Author: alvarolopez

imgsync/distros/__init__.py

@@ -116,7 +116,7 @@ def _download_one(self, url, checksum):
         """
         with tempfile.NamedTemporaryFile(suffix=".imgsync", delete=False) as location:
             try:
-                response = requests.get(url, stream=True)
+                response = requests.get(url, stream=True, timeout=10)
             except Exception as e:
                 os.remove(location.name)
                 LOG.error(e)

imgsync/distros/centos.py

@@ -67,7 +67,7 @@ def __init__(self):
 
     def _get_index(self):
         url = self.url + "image-index"
-        r = requests.get(url)
+        r = requests.get(url, timeout=10)
         if r.ok:
             aux = r.text
         else:

imgsync/distros/debian.py

@@ -43,7 +43,7 @@ def _sync_latest(self):
         base_url = self.url
 
         checksum_file = base_url + "SHA512SUMS"
-        checksum_file = requests.get(checksum_file)
+        checksum_file = requests.get(checksum_file, timeout=10)
         if checksum_file.status_code != 200:
             LOG.error("Could not get checksums file %s" % checksum_file.url)
             return

imgsync/distros/ubuntu.py

@@ -49,7 +49,7 @@ def _sync_latest(self):
         LOG.info("Downloading %s", filename)
         base_url = self.url + "current/"
         checksum_file = base_url + "SHA256SUMS"
-        checksum_file = requests.get(checksum_file)
+        checksum_file = requests.get(checksum_file, timeout=10)
         if checksum_file.status_code != 200:
             LOG.error("Could not get checksums file %s" % checksum_file.url)
             return