only parse html files

This commit is contained in:
Roman committed 2022-01-24 20:01:49 +01:00
1 parent 8adc30d7ae
commit 631dade746
1 file changed
+7 -1
Regular → Executable
+7 -1
View File
@@ -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))