1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
|
"""A tool to simplify fetching robots.txt from a large number of websites.
Usage
-----
$ python fetch_robotstxt.py -l top-10000-websites.txt -d test_data
"""
from __future__ import annotations
import argparse
import sys
from pathlib import Path
from typing import TYPE_CHECKING, Any
from urllib.parse import ParseResult, urlparse, urlunparse
import scrapy
from scrapy.crawler import CrawlerProcess
if TYPE_CHECKING:
from collections.abc import Iterable
from scrapy.http import Response
from twisted.python.failure import Failure
parser = argparse.ArgumentParser(description="Download robots.txt of given websites.")
parser.add_argument(
"-l",
"--list",
action="append",
dest="websites",
help="Adds to the list of websites.",
type=Path,
)
parser.add_argument(
"-d",
"--destination",
action="store",
dest="directory",
help="Directory to save robots.txt files.",
type=Path,
)
args = parser.parse_args()
if not args.directory or not args.websites:
print("Insufficient or invalid argument(s) provided.")
sys.exit()
class RobotstxtSpider(scrapy.Spider):
name = "robotstxt_spider"
def start_requests(self) -> Iterable[scrapy.Request]:
w: Path
for w in args.websites:
if w.is_file():
with w.open() as f:
for domain in f:
yield scrapy.Request(
url=f"https://{domain.strip()}/robots.txt",
callback=self.parse,
errback=self.err_cb,
)
def parse(self, response: Response, **kwargs: Any) -> Any:
filename = urlparse(response.url).netloc
if not args.directory.exists():
args.directory.mkdir()
(args.directory / filename).write_bytes(response.body)
def err_cb(self, failure: Failure) -> Iterable[scrapy.Request]:
request = failure.request # type: ignore[attr-defined]
parts = urlparse(request.url)
parts = ParseResult(
"http", parts.netloc, parts.path, parts.params, parts.query, parts.fragment
)
url = urlunparse(parts)
yield scrapy.Request(url=url, callback=self.parse)
process = CrawlerProcess(settings={"ROBOTSTXT_OBEY": False})
process.crawl(RobotstxtSpider)
process.start()
|
