17
\$\begingroup\$

I just finished a scraper in python using scrapy. The scraper logs in to a certain page and then scrapes a list of other pages using the authenticated session. It retrieves the title of these pages and puts it in a list of items together with the page name.

Because there are little examples to find on continuing scraping after log-in I would like to receive some feedback on if this is the best way to continue scraping.

(Also I have a weird workaround to get the pagename into the itemlist, since I already have the pagename in the pagelist, but in the action method I retrieve it from the url again. Tips on doing this in an easier way are welcome too.)

p.s. The code has been slightly edited for sharing purposes.

class ShareSpider(scrapy.Spider):
    name = "sharespider"
    start_urls = ['http://www.example.com/public/login.aspx']

    def parse(self, response):
        yield scrapy.FormRequest.from_response(
            response,
            formxpath='//form[@id="login"]',
            formdata={
                'UserName': 'UserNameHere',             
                'Password': 'PasswordHere',             
                'Action':'1',
            },
            callback=self.after_login
        )

    def after_login(self, response):
        baseurl = 'http://www.example.com/public/'
        #Specify pages to crawl here:
        pagelist = ['page1.aspx', 'page2.aspx', 'page3.aspx', 'page4.aspx']
        for page in pagelist:
            yield Request(url= baseurl + page + "?id=1",
            callback=self.action)

    def action(self, response):
        pageurl = str(response.url)
        page = re.search('public/(.*)id=1', pageurl)
        if page:
            pagename = page.group(1)
        #Get page <title> element and strip whitespace
        title = str(response.selector.xpath('//title/text()').extract_first())
        res = title.strip()

        item = PageItem()
        item['pagename'] = pagename
        item['description'] = res
        yield item
\$\endgroup\$
2
  • 1
    \$\begingroup\$ Help me understand: what effect does the last line (callback=self.after_login) have? \$\endgroup\$ Commented Jul 6, 2016 at 8:45
  • \$\begingroup\$ Good feedback, it does not seem to have any effect. No idea why i put it there... \$\endgroup\$ Commented Jul 6, 2016 at 8:50

1 Answer 1

10
+50
\$\begingroup\$

I'll start off by saying this is actually really good. There are a few minor PEP8 errors:

  • In after_login you call Request. the argument url shouldn't have a space on the right side of the equals.
  • Another thing in Request. The argument callback needs at least one more indent. However you should either aligned with opening delimiter or use hanging indents.
    And so you should change this to either of:

    yield Request(url=baseurl + page + "?id=1",
                  callback=self.action)
    
    yield Request(
        url=baseurl + page + "?id=1",
        callback=self.action)
    
  • Variables and functions should be named in snake_case. pagelist should be page_list, pageurl page_url, etc.

    Whilst not being a PEP8 error you may want to change page_list to say pages. This is as multiple pages are probably a list, and sounds more fluid.

I say the above as consistency is the best way to allow others to easily read your work. But they are quite minor.

Personally I'd move the baseurl out of the class and make it a constant. This allows you to easily change the url, say the company you're scraping change their company/domain name or their top-level domain. I'd also move other system settings, 'UserNameHere', 'PasswordHere', 'http://www.example.com'. I may even say that pagelist should be moved out too.

Now for comments on your business logic, as parse and after_login are very short 3 line functions, there's honestly not much to say about them. action however has a few problems:

  • response.url is a string and so you shouldn't cast it to one. This confused me as I it made me think response.url would be something that isn't a string.

  • You seem to be getting the response as an instance of TextResponse, which contains selector. This also contains the shortcut xpath which is equivalent to selector.xpath. Which allows you to reduce the size of that large command.

  • Unfortunately scrapy doesn't document extract_first, but looking at the source they use Parsel. The documentation on extract_first was also kinda lacking, so I read the source code again, and it seems like they always return strings or None. So you probably don't need the str around extract_first if you set default to an empty string.

  • You don't define pagename unless page is not None. But you go on to use it regardless. This is wrong, and can lead to errors. There are two ways to come at this, either silently fail. Or raise an exception, where NameError: name 'pagename' is not defined is not a good error.

  • Finally it's strange to see yield rather than return, you're only going to ever return one thing. And so unless it's a requirement imposed by scrapy I'd change it to return.

And so if I were to improve it, leaving yield and failing silently, I'd end with:

BASE_URL = 'http://www.example.com'
USER_NAME = 'UserNameHere'
PASSWORD = 'PasswordHere'
PAGES = ['page1.aspx', 'page2.aspx', 'page3.aspx', 'page4.aspx']

class ShareSpider(scrapy.Spider):
    name = "sharespider"
    start_urls = [BASE_URL + '/public/login.aspx']

    def parse(self, response):
        yield scrapy.FormRequest.from_response(
            response,
            formxpath='//form[@id="login"]',
            formdata={
                'UserName': USER_NAME,             
                'Password': PASSWORD,             
                'Action':'1',
            },
            callback=self.after_login)

    def after_login(self, response):
        base_url = BASE_URL + '/public/'
        for page in PAGES:
            yield Request(
                url=base_url + page + "?id=1",
                callback=self.action)

    def action(self, response):
        page = re.search('public/(.*)id=1', response.url)
        if page:
            page_name = page.group(1)
            title = response.xpath('//title/text()').extract_first('').strip()
            item = PageItem()
            item['pagename'] = page_name
            item['description'] = title
            yield item
\$\endgroup\$

Your Answer

By clicking “Post Your Answer”, you agree to our terms of service and acknowledge you have read our privacy policy.

Not the answer you're looking for? Browse other questions tagged or ask your own question.