od-database/scrapy_od_database/spiders/ftp_links_spider.py
2018-06-10 14:12:55 -04:00

50 lines
1.3 KiB
Python

import json
import scrapy
import os
from scrapy_od_database.items import File
class AnonFtpRequest(scrapy.Request):
anon_meta = {
"ftp_user": "anonymous",
"ftp_password": "od-database"
}
def __init__(self, *args, **kwargs):
super(AnonFtpRequest, self).__init__(*args, **kwargs)
self.meta.update(self.anon_meta)
class FtpLinksSpider(scrapy.Spider):
"""Scrapy spider for ftp directories. Will gather all files recursively"""
name = "ftp_links"
handle_httpstatus_list = [404]
def __index__(self, **kw):
super(FtpLinksSpider, self).__init__(**kw)
self.base_url = kw.get("base_url")
def start_requests(self):
yield AnonFtpRequest(url=self.base_url, callback=self.parse)
def parse(self, response):
stripped_url = response.url[len(self.base_url) - 1:]
files = json.loads(response.body)
for file in files:
if file['filetype'] == 'd':
yield AnonFtpRequest(os.path.join(response.url, file["filename"]))
if file['filetype'] == '-':
print(file)
result = File(
name=file['filename'],
path=stripped_url.strip("/"),
size=file['size'],
mtime=file['date'])
yield result