Browse Source

only parse html files

Roman Hergenreder 2 years ago
parent
commit
631dade746
1 changed files with 7 additions and 1 deletions
  1. 7 1
      crawl_urls.py

+ 7 - 1
crawl_urls.py

@@ -1,3 +1,5 @@
+#!/usr/bin/env python3
+
 import argparse
 import urllib.parse
 import urllib3
@@ -55,6 +57,10 @@ class Crawler:
 
             self.visited.add(url)
             res = self.request(url)
+            content_type = res.headers.get("Content-Type", None)
+            if "text/html" not in content_type.lower().split(";"):
+                continue
+
             urls = self.collect_urls(res.text)
             for url in urls:
                 parts = urllib.parse.urlparse(url)
@@ -124,4 +130,4 @@ if __name__ == "__main__":
 
     for name, values in results.items():
         print(f"=== {name} ===")
-        print("\n".join(values))
+        print("\n".join(values))