如何从一个功能访问另一个功能的数据

时间:2019-05-30 17:58:15

标签: python scrapy splash scrapy-splash

我有一个Web抓取程序,在我的Spider中,我需要访问process_csv函数中的信息,但是我需要从start_requests函数中访问它。在我的process_csv函数中,我有一个变量“ ip”,该变量从CSV文件中获取ips,并在开头添加http://,在末尾添加端口号,然后函数将它们与需要的用户代理和URL配对被刮掉。无论如何,我都需要在start_requests函数中访问这些ip,因此我需要从start_requests函数中调用变量“ ip”。我该怎么做?

SplashSpider.py

import csv
from scrapy.spiders import Spider
from scrapy_splash import SplashRequest
from ..items import GameItem

# process the csv file so the url + ip address + useragent pairs are the same as defined in the file # returns a list of dictionaries, example:
# [ {'url': 'http://www.starcitygames.com/catalog/category/Rivals%20of%20Ixalan',
#    'ip': 'http://204.152.114.244:8050',
#    'ua': "Mozilla/5.0 (BlackBerry; U; BlackBerry 9320; en-GB) AppleWebKit/534.11"},
#    ...
# ]
def process_csv(csv_file):
    data = []
    reader = csv.reader(csv_file)
    next(reader)
    for fields in reader:
        if fields[0] != "":
            url = fields[0]
        else:
            continue # skip the whole row if the url column is empty
        if fields[1] != "":
            ip = "http://" + fields[1] + ":8050" # adding http and port because this is the needed scheme
        if fields[2] != "":
            useragent = fields[2]
        data.append({"url": url, "ip": ip, "ua": useragent})
    return data


class MySpider(Spider):
    name = 'splash_spider'  # Name of Spider

    # notice that we don't need to define start_urls
    # just make sure to get all the urls you want to scrape inside start_requests function

    # getting all the url + ip address + useragent pairs then request them
    def start_requests(self):

        # get the file path of the csv file that contains the pairs from the settings.py
        with open(self.settings["PROXY_CSV_FILE"], mode="r") as csv_file:
            # requests is a list of dictionaries like this -> {url: str, ua: str, ip: str}
            requests = process_csv(csv_file)

        for req in requests:
            # no need to create custom middlewares
            # just pass useragent using the headers param, and pass proxy using the meta param
            yield SplashRequest(url=req["url"], callback=self.parse, args={"wait": 3},
                                splash_url = ip,
                                headers={"User-Agent": req["ua"]},
                                meta={
                                    "proxy": (req["ip"]),
                                  })

    # Scraping
#    def parse(self, response):
#        item = GameItem()
#        for game in response.css("tr"):
#            # Card Name
#            item["card_name"] = game.css("a.card_popup::text").extract_first()
#            yield item   

1 个答案:

答案 0 :(得分:1)

在您的process_csv函数中,您要在data后面附加一个名为{"url": url, "ip": ip, "ua": useragent}的列表。因此,您可以从函数中调用每个记录,这将返回一个字典,然后可以调用ip。示例:

request_ip = process_csv(csv_file)[0]["ip"]-将返回列表中第一条记录中的ip。您可以运行循环以获取所有ip并将其存储在列表中。

希望这会有所帮助。