fix content type filtering

CybercentreCanada · Jun 6, 2022 · aacc08e · aacc08e
1 parent f81b53d
commit aacc08e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/urldownloader/urldownloader.py b/urldownloader/urldownloader.py
@@ -18,7 +18,8 @@ def fetch_uri(self, uri: str, apply_filter: bool = True) -> str:
             resp = requests.get(uri, allow_redirects=True)
             # Only concerned with gathering responses of interest
             if resp.ok:
-                if apply_filter and resp.headers.get('Content-Type') in self.content_type_filter:
+                if apply_filter and any(content_type in resp.headers.get('Content-Type')
+                                        for content_type in self.content_type_filter):
                     return
                 resp_fh = NamedTemporaryFile(delete=False)
                 resp_fh.write(resp.content)