add waf download function.

GSA · Dec 18, 2023 · ea67f06 · ea67f06
1 parent aee51a6
commit ea67f06
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/harvester/extract/waf.py b/harvester/extract/waf.py
@@ -28,3 +28,13 @@ def traverse_waf(url, files=[], file_ext=".xml", folder="/", filters=[]):
         traverse_waf(folder, files=files, filters=filters)
 
     return files
+
+
+def download_waf(files):
+    output = []
+    for file in files:
+        res = requests.get(file)
+        if res.status_code == 200:
+            output.append(res.content)
+
+    return output