zoukankan      html  css  js  c++  java
  • Civic Hacking with Python – Part 2

    Civic Hacking with Python – Part 2 « syslogd.net

    Civic Hacking with Python – Part 2

    I apologize for the time it took before writing this part, but you know what they say: Better late then never.

    In this part, I’ll explain how I extracted (scraped) the data from the Transports Quebec database mentioned in Part 1 using Python, Scrapy and a few other tools.

    This post is not a full-fledged tutorial on using Scrapy, but it should give you a place to start if you’d like to do something similar.

    In order to jump directly to the good stuff, I’ll skip the Scrapy installation and assume you are already familiar with Python. If you are not familiar with Python, you’re missing a lot. Get started here, then use some Google-Fu and start hacking on your own project to keep learning.

    DISCLAIMER:

    • This small project was a quick and dirty drive-by experiment. This code is not: a complete solution, the best solution, production-ready, [insert your own similar statement here]. I’m sure you can do better :-) (Feel free to post your comments and improvements, however, keep in mind that I don’t plan to maintain or improve this code)
    • This code follows my late-night style guide.
    • For non-French speaking readers: This project extracts data from a French (Quebec, Canada) website, however, you can continue to read, the actual content is not really important and you should be able to follow without problem. You may also learn some French words. ;-)
    • For French-speaking readers: If you are reading this, you probably won’t mind the English comments in my code. Unless required to do otherwise, I always write my code/comments in English. It’s easier to share this way.
    • I’m sure you’ll find typos.
    • Everything below was done using Python 2.7 and Scrapy v0.13. I have not tested with Scrapy 0.14+ or Python 3+ (UPDATE 2011/12/12: As Pablo Hoffman (none other than the lead Scrapy developer) noted in the comments, Scrapy does not support Python 3 and there are currently no plans to support it. Sorry about that!)

    That said, let’s get started.

    Our target: The Transports Quebec infrastructure database (Well, they call it “Ponts et routes: Information aux citoyens”.)

    If you visit the website, you’ll notice that the first page you get is an empty form like this one:

    So, to get to the actual data, you need to submit the form by clicking on “Afficher les résultats“. Luckily, the default form values will display all entries. After submitting the form, you’ll get results like these:

    This is what our main scraper will have to deal with. Unfortunately, not all the fields we want to extract are available in this table. The rest is in what I call the “details” page. You get there by clicking on the “Fiche No”  (first column) link (e.g. 00002). The details page look like this:

    So for each entry found, our scraper will have to follow the “Fiche No” link and scrape the details page too.

    In addition, each page of the report only contains 15 items. This means we’ll have to navigate the pages too.

    Tools You’ll Need

    In order to identify HTML elements to extract, you’ll need these:

    NOTE: I used these tools. You may have your own tools that you prefer. Feel free to let me know of other good ones in the comments. (No need to point out that Safari has a web inspector, I know, but I did not find anything equivalent to Firefinder.)

    Getting Started

    Switch to a work directory then create your base Scrapy project (I called mine mtqinfra):

    $ scrapy startproject mtqinfra
    $ find .
    .
    ./mtqinfra
    ./mtqinfra/__init__.py
    ./mtqinfra/items.py
    ./mtqinfra/pipelines.py
    ./mtqinfra/settings.py
    ./mtqinfra/spiders
    ./mtqinfra/spiders/__init__.py
    ./scrapy.cfg

    At this point we have a “skeleton” project. Now let’s create a very simple spider just to see if we can get this to work. (NOTE: There’s a “scrapy   genspider” command but I won’t use it here.) Create the file spiders/mtqinfra_spider.py:

    01#!/usr/bin/env python
    02# encoding=utf-8
    03 
    04from scrapy.spider import BaseSpider
    05from scrapy.http import Request
    06from scrapy.http import FormRequest
    07from scrapy.selector import HtmlXPathSelector
    08from scrapy import log
    09import sys
    10### Kludge to set default encoding to utf-8
    11reload(sys)
    12sys.setdefaultencoding('utf-8')
    13 
    14class MTQInfraSpider(BaseSpider):
    15    name = "mtqinfra"
    16    allowed_domains = ["www.mtq.gouv.qc.ca"]
    17    start_urls = [
    19    ]
    20 
    21    def parse(self, response):
    22        pass

    Now, let’s see if this appears to work:

    $ scrapy crawl mtqinfra
    2011-11-22 00:52:09-0500 [scrapy] INFO: Scrapy 0.13.0 started (bot: mtqinfra)
    [... more here ...]
    2011-11-22 00:52:10-0500 [mtqinfra] DEBUG: Redirecting (302) to <GET http://www.mtq.gouv.qc.ca/pls/apex/f?p=102:56:2747914247598050::NO:RP::> from <POST http://www.mtq.gouv.qc.ca/pls/apex/wwv_flow.accept>
    2011-11-22 00:52:10-0500 [mtqinfra] DEBUG: Crawled (200) <GET http://www.mtq.gouv.qc.ca/pls/apex/f?p=102:56:2747914247598050::NO:RP::> (referer: None)
    2011-11-22 00:52:10-0500 [mtqinfra] INFO: Closing spider (finished)
    [... more here ...]

    As you can see, the bot did request our page, got redirected (Status 302) because the initial URL did not include a session ID (2747914247598050) and finally downloaded (Status 200) the page.

    Posting the Form

    OK, the next step if to get our bot to submit the form for us. Let’s tweak the parse method a bit and add another method to handle the real parsing like this:

    01def parse(self, response):
    02    return [FormRequest.from_response(response,
    03                                      callback=self.parse_main_list)]
    04 
    05def parse_main_list(self, response):
    06    self.log("After submitting form.", level=log.INFO)
    07    with open("results.html", "w") as f:
    08        f.write(response.body)
    09    import os
    10    os.system("open results.html")

    Now, the parse method return a “FormRequest” object that will instruct the bot to submit the form then call “self.parse_main_list” with the response.

    $ scrapy crawl mtqinfra

    Oops. The response contains no results. Whats wrong? Well, if you search a bit (I sniffed the network to compare what is sent by Firefox when the form button is pressed versus what is sent by Scrapy), you’ll find that the “p_request” form field is not set to “RECHR” by Scrapy as when sent by the browser. This is due to the fact that it is empty by default and set by a Javascript function. Let’s fix that:

    01def parse(self, response):
    02    return [FormRequest.from_response(response,
    03                                      formdata={ "p_request": "RECHR" },
    04                                      callback=self.parse_main_list)]
    05 
    06def parse_main_list(self, response):
    07    self.log("After submitting form.", level=log.INFO)
    08    with open("results.html", "w") as f:
    09        f.write(response.body)
    10    import os
    11    os.system("open results.html")

    Ahh, much better. Now we can begin our parser.

    Identifying HTML Elements and Their Corresponding XPath

    Scrapy uses XPath to select and extract elements from a web page. Well, technically speaking you could parse the response body any way you want (e.g. using regular expressions), but XPath is very powerful so I suggest you give it a try.

    I won’t write an XPath tutorial here, but simply put, XPath is a query language that allows you to select elements from HTML like you would do with SQL to extract fields from a table. Although XPath queries can appear intimidating at first, the XPath syntax itself is pretty simple.

    Here are some tips to understand, learn and use XPath quickly and identify elements you want to extract.

    Use Firebug to identify absolute XPath expressions

    In Firebug, the absolute XPath expression to select an HTML element is displayed in a tooltip:

    You can also right-click an element and select “Copy XPath” to copy it to the clipboard:

    Use Firefinder to test XPath expressions

    In Firebug, select the Firefinder tab  and enter your XPath expression (or query or filter or selector, whatever you want to call it) then click “Filter”. The matching element(s) will be listed below and highlighted on the page. Because we’ll need to loop on each result table row, try it with this expression:

    //table[@id="R10432126941777590"]//table[@summary="Report"]/tbody/tr

    This expression will select each row of the result table.

    Use “scrapy shell” to test XPath expressions in Scrapy

    Scrapy has a very handy “shell” mode to help you test stuff. In order to bypass the form submission process and get directly on the result page, submit the form with your browser and then copy the URL that includes your session ID. If you do a “GET” on this URL, you’ll get the page you were viewing in your browser (as long as the session ID is still valid). Let’s try it:

    01$ scrapy shell http://www.mtq.gouv.qc.ca/pls/apex/f?p=102:56:482485043431341::NO:RP::
    02[... MORE HERE ...]
    03>>> hxs.select('//table[@id="R10432126941777590"]//table[@summary="Report"]/tbody/tr[2]')
    04[]
    05>>> hxs.select('//table[@id="R10432126941777590"]//table[@summary="Report"]/tr[2]')
    06[<HtmlXPathSelector xpath='//table[@id="R10432126941777590"]//table[@summary="Report"]/tr[2]' data=u'<tr onmouseover="row_mouse_over104321269'>]
    07>>> row2 = hxs.select('//table[@id="R10432126941777590"]//table[@summary="Report"]/tr[2]')
    08>>> row2_cells = row2.select('td')
    09>>> len(row2_cells)
    1011
    11>>> row2_cells[0]
    12<HtmlXPathSelector xpath='td' data=u'<td class="t3data" align="center"><a hre'>
    13>>> row2_cells[0].extract()
    14u'
    15<a href="f?p=102:53:482485043431341::NO:53:P53_IDE_STRCT_0001:211033"><img title="Fiche de la structure: 00002" src="wwv_flow_file_mgr.get_file?p_security_group_id=1848625384920754&p_fname=detail.gif" alt="Fiche de la structure: 00002" />
    1600002</a>
    17'
    18>>> row2_cells[0].select('a/text()').extract()[0]
    19u'00002'
    20>>> row2_cells[0].select('a/@href').extract()[0]
    21u'f?p=102:53:482485043431341::NO:53:P53_IDE_STRCT_0001:211033'

    IMPORTANT: Take note of lines 3 and 4. I am unsure why, but while Firefinder takes the “tbody” tag into account in XPath expressions, Scrapy does not want them. Thus, our previously working XPath returns nothing. If you remove the “tbody” tag (line 5), the expression will work and return the second row of the result table.

    Line 8 shows the power of XPath and the Scrapy HtmlXPathSelector object. To extract an array of cells for row #2, on the HtmlXPathSelector for the row we simply call “select(‘td’)”.

    The rest of the lines shows how to use the extract() method to extract HTML, text and attribute values.

    Create XPath expressions that are general and specific at the same time

    Although this does not appear to make much sense, here’s what I mean:

    Take this XPath (Firefinder format, remove tbody for Scrapy):

    /html/body/form/table/tbody/tr[2]/td/table[4]/tbody/tr/td/table[3]/tbody/tr[2]/td/table

    It is a very specific and absolute XPath to the results table. Should the web page change just a bit (e.g., an extra row in the first table of the form or a new table to hold new information), your XPath will become invalid or point to the wrong table. Now, in the page generated by the underlying reporting logic (Oracle Application Express (APEX) in this case), we noticed that the parent table of the results table has the “id” attribute set to “R10432126941777590″ and that the actual results table has the attribute “summary” set to “Report”. We can then use the following XPath to get to the same table:

    //table[@id="R10432126941777590"]//table[@summary="Report"]

    It is more “general” as it skips over everything but two tables. It simply says: “Get me the tables that have their “summary” attribute set to “Report” that are also “under” (in) tables that have their “id” attribute set to “R10432126941777590″. However, because the “id” is very specific (only match one table) and the “summary” is also (somewhat) specific because it only match one table inside that other table, we are unlikely to match anything else. Thats what I mean by general and specific at the same time.

    Now, I don’t know Oracle APEX enough to be certain the “id” used above won’t change if the report HTML format is modified, so maybe my solution could break later in this case, however, the principle in general is still good.

    Use relative XPath expressions and HtmlXPathSelector objects

    Don’t use absolute XPath expressions (as mentioned above) or repeat expressions in your code. Instead use the powerful HtmlXPathSelector objects to navigate in the HTML structure using relative XPath expressions. For example this code gets you columns 1-3 of row 2 or the results table but it sucks:

    hxs = HtmlXPathSelector(response)
    row2_cell1 = hxs.select('/html/body/form/table/tr[2]/td/table[4]/tr/td/table[3]/tr[2]/td/table/tr[2]/td[1]')
    row2_cell2 = hxs.select('/html/body/form/table/tr[2]/td/table[4]/tr/td/table[3]/tr[2]/td/table/tr[2]/td[2]')
    row2_cell3 = hxs.select('/html/body/form/table/tr[2]/td/table[4]/tr/td/table[3]/tr[2]/td/table/tr[2]/td[3]')

    This code does the same, but does not suck:

    hxs = HtmlXPathSelector(response)
    rows = hxs.select('//table[@id="R10432126941777590"]//table[@summary="Report"]/tr')
    row2 = rows[1] # NOTE: rows is a Python array, indexing starts at 0
    cells = row.select('td')
    row2_cell1 = cells[0]
    row2_cell2 = cells[1]
    row2_cell3 = cells[2]

    Why? Because in the first case, if anything changes in the HTML, you’ll need to modify 3 XPath expressions. In the second case, you’ll probably need to modify only one (if necessary at all). Of course, this example is simplified a bit to show you the concept (you’ll probably want to loop over rows and cells in your code as we’ll do later), but I hope you get the idea. Unfortunately, sometimes there is no (safe) way to get to an element other than by using an (almost) absolute XPath. Just try to minimize their use in your project.

    The MTQInfraItem Object

    The parsers that will handle responses as the website is crawled return “Item” objects (or Request objects to instruct the crawler to request more pages). “Item” objects are more or less a “model” of the data you will be scraping. It is a container for the structured data you will be extracting from the HTML pages. Edit the items.py file and replace the existing “MtqinfraItem” with this one:

    01class MTQInfraItem(Item):
    02    # From main table
    03    record_no = Field()
    04    record_href = Field()
    05    structure_id = Field()
    06    structure_name = Field()
    07    structure_type = Field()
    08    structure_type_img_href = Field()
    09    territorial_direction = Field()
    10    rcm = Field()
    11    municipality = Field()
    12    road = Field()
    13    obstacle = Field()
    14    gci = Field()
    15    ai_desc = Field()
    16    ai_img_href = Field()
    17    ai_code = Field()
    18    location_href = Field()
    19    planned_intervention = Field()
    20    # From details
    21    road_class = Field()
    22    latitude = Field()
    23    longitude = Field()
    24    construction_year = Field()
    25    picture_href = Field()
    26    last_general_inspection_date = Field()
    27    next_general_inspection_date = Field()
    28    average_daily_flow_of_vehicles = Field()
    29    percent_trucks = Field()
    30    num_lanes = Field()
    31    fusion_marker = Field()

    As you can see, to create your own MTQInfraItem type, you simply subclass the Item class and add a bunch of fields that you later plan to populate and save in your output.

    Scraping the Main List

    Scraping the main page requires us to do the following:

    1. Process each row of the results table and for each one:
      1. Extract all the data we want to keep
      2. Create a new MTQInfraItem object for the data
      3. Save the item in a buffer because it is still incomplete as we need to scrape the “details” page to extract the rest of the fields
      4. Return a “Request” object to the crawler to inform it that the “details” page needs to be requested
    2. Check if there is another page containing results and if so, return a ”Request” object to the crawler to inform it that another page of results needs to be requested.

    The final parser for the main list looks like this:

    001def parse_main_list(self, response):
    002    try:
    003        # Parse the main table
    004        hxs = HtmlXPathSelector(response)
    005        rows = hxs.select('//table[@id="R10432126941777590"]//table[@summary="Report"]/tr')
    006        if not rows:
    007            self.log("Failed to extract results table from response for URL '{:s}'. Has 'id' changed?".format(response.request.url), level=log.ERROR)
    008            return
    009        for row in rows:
    010            cells = row.select('td')
    011            # Skip header
    012            if not cells:
    013                continue
    014            # Check if this is the last row. It contains only one cell and we must dig in to get page info
    015            if len(cells) == 1:
    016                total_num_records = int(hxs.select('//table[@id="R19176911384131822"]/tr[2]/td/table/tr[8]/td[2]/text()').extract()[0])
    017                first_record_on_page = int(cells[0].select('//span[@class="fielddata"]/text()').extract()[0].split('-')[0].strip())
    018                last_record_on_page = int(cells[0].select('//span[@class="fielddata"]/text()').extract()[0].split('-')[1].strip())
    019                self.log("Scraping details for records {:d} to {:d} of {:d} [{:.2f}% done].".format(first_record_on_page,
    020                            last_record_on_page, total_num_records, float(last_record_on_page)/float(total_num_records)*100), level=log.INFO)
    021                # DEBUG: Switch check if you only want to process a certain number of records (e.g. 45)
    022                #if last_record_on_page < 45:
    023                if last_record_on_page < total_num_records:
    024                    page_links = cells[0].select('//a[@class="fielddata"]/@href').extract()
    025                    if len(page_links) == 1:
    026                        # On first page
    027                        next_page_href = page_links[0]
    028                    else:
    029                        next_page_href = page_links[1]
    030                    # Request to scrape next page
    031                    yield Request(url=response.request.url.split('?')[0]+'?'+next_page_href.split('?')[1], callback=self.parse_main_list)
    032                    continue
    033                else:
    034                    # Nothing more to do
    035                    break
    036 
    037            # Cell 1: Record # + Record HREF
    038            record_no = cells[0].select('a/text()').extract()[0].strip()
    039            record_relative_href = cells[0].select('a/@href').extract()[0]
    040            record_href = response.request.url.split('?')[0]+'?'+record_relative_href.split('?')[1]
    041            structure_id = re.sub(ur"^.+:([0-9]+)$", ur'\1', record_href)
    042            # Cell 2: Name
    043            structure_name = "".join(cells[1].select('.//text()').extract()).strip()
    044            # Cell 3: Structure Type Image
    045            structure_type = cells[2].select('img/@alt').extract()[0]
    046            structure_type_img_relative_href = cells[2].select('img/@src').extract()[0]
    047

                structure_type_img_href = re.sub(r'/[^/]*

    , r'/', response.request.url) + structure_type_img_relative_href

    048            # Cell 4: Combined Territorial Direction + Municipality
    049            territorial_direction = "".join(cells[3].select('b//text()').extract()).strip()
    050            # NOTE: Municipality taken from details page as it was easier to parse.
    051            # Cell 5: Road
    052            road = "".join(cells[4].select('.//text()').extract()).strip()
    053            # Cell 6: Obstacle
    054            obstacle = "".join(cells[5].select('.//text()').extract()).strip()
    055            # Cell 7: GCI (General Condition Index)
    056            gci = cells[6].select('nobr/text()').extract()[0].strip()
    057            # Cell 8: AI (Accessibility Index)
    058            # Defaults to "no_restriction" as most records will have this code.
    059            ai_code = 'no_restriction'
    060            if cells[7].select('nobr/img/@alt'):
    061                ai_desc = cells[7].select('nobr/img/@alt').extract()[0]
    062                ai_img_relative_href = cells[7].select('nobr/img/@src').extract()[0]
    063

                    ai_img_href = re.sub(r'/[^/]*

    , r'/', response.request.url) + ai_img_relative_href

    064            else:
    065                # If no image found for AI, then code = not available
    066                ai_code = 'na'
    067                if cells[7].select('nobr/text()'):
    068                    # Some text was available, use it
    069                    ai_desc = cells[7].select('nobr/text()').extract()[0]
    070                else:
    071                    ai_desc = "N/D"
    072                # Use our own Gray trafic light hosted on CloudApp
    073                ai_img_href = "http://cl.ly/2r2A060b1g0N0l3f1y3L/feugris.png"
    074            # Set ai_code according to description if applicable
    075            if re.search(ur'certaines', ai_desc, re.I):
    076                ai_code = 'restricted'
    077            elif re.search(ur'fermée', ai_desc, re.I):
    078                ai_code = 'closed'
    079            # Cell 9: Location HREF
    080            onclick = cells[8].select('a/@onclick').extract()[0]
    081            location_href = re.sub(ur"^javascript:pop_url\('(.+)'\);$", ur'\1', onclick)
    082            # Cell 10: Planned Intervention
    083            planned_intervention = "".join(cells[9].select('.//text()').extract()).strip()
    084            # Cell 11: Report (yes/no image only) (SKIP)
    085 
    086            item = MTQInfraItem()
    087            item['record_no'] = record_no
    088            item['record_href'] = record_href
    089            item['structure_id'] = structure_id
    090            item['structure_name'] = structure_name
    091            item['structure_type'] = structure_type
    092            item['structure_type_img_href'] = structure_type_img_href
    093            item['territorial_direction'] = territorial_direction
    094            item['road'] = road
    095            item['obstacle'] = obstacle
    096            item['gci'] = gci
    097            item['ai_desc'] = ai_desc
    098            item['ai_img_href'] = ai_img_href
    099            item['ai_code'] = ai_code
    100            item['location_href'] = location_href
    101            item['planned_intervention'] = planned_intervention
    102            self.items_buffer[structure_id] = item
    103            # Request to scrape details
    104            yield Request(url=record_href, callback=self.parse_details)
    105    except Exception as e:
    106        # Something went wrong parsing this page. Log URL so we can determine which one.
    107        self.log("Parsing failed for URL '{:s}'".format(response.request.url), level=log.ERROR)
    108        raise # Re-raise exception

    More details for each lines:

    Lines 2,105-108: We wrap our code in a try/except block to log any parsing error with our own message.

    Lines 11-13: This is where we skip the header. The logic works because the table header cells are “th” tags, not “td”, so cells is None.

    Lines 14-35: This is where we check if we’ve reached the last page or not. If not, we create the “Request” object for the next page.

    Lines 21-22: Note the commented “if last_record_on_page < 45:” line. We’ll refer to it in the “Testing It” section below.

    Lines 37-84: This is where we extract our data.

    Lines 86-101: Here, we create our MTQInfraItem and set the fields we just extracted.

    Line 102: Here we save our MTQInfraItem to our internal buffer so we can use it later when we parse the corresponding “details” page.

    Line 104: Finally we return a “Request” object so the crawler will request the corresponding “details” page and call our “parse_details” method with the response.

    Scraping the Details

    I won’t post the code to scrape the “details” page here as it is mostly code similar to lines 37-84 of the previous parser. The only thing to note is that in parse_details(), we actually return the final MTQInfraItem object to the crawler so it can be sent down the pipeline.

    Testing It

    Before you run this puppy for the first time, you should limit the crawling to a small number of records. I used 45 records because each page has 15. This give us a reasonable sample to validate most of our code. This is where line 22 in parse_main_list() comes handy. Simply uncomment it and comment line 23 to stop processing after 45 records.

    If you try to run the crawler as-is on the Transports Quebec website, you’ll probably get errors. At least I did. Apparently, the website does not process concurrent requests using the same session ID. You get an error page when you attempt to do so. By default, Scrapy will attempt to crawl websites more quickly by executing requests concurrently. To disable this completely, add the following lines to settings.py:

    CONCURRENT_REQUESTS_PER_DOMAIN = 1
    CONCURRENT_SPIDERS = 1

    REMEMBER: Be polite. Try to minimize the impact of your scraping on the web server. Do your testing on a small number of pages until your are satisfied with your output. Don’t scrape thousand of pages, add a new field and then scrape thousand of pages again. This is particularly true if you test this project. Don’t hammer the Transports Quebec website just for fun, they will simply raise my taxes to buy a bigger server :-)

    By default if you simply run “scrapy crawl mtqinfra” Scrapy will print each item on stdout. If you want to save the output in a usable format, you can use the “-o output_file” and “–output-format=format” options. e.g.:

    scrapy crawl mtqinfra --output-format=csv -o output.csv

    NOTE: If you attempt to save to XML at this point, you’ll only get a bunch of exceptions because the default XML exporter only handles strings fields and our items have floats. Read on for the solution.

    Generating Output Files in Different Formats

    OK, you tested the crawler and you are satisfied but you want to save the output in different formats, in a format of your own or in a database. This is where pipelines and exporters come into play.

    A pipeline is simply a Python object with a “process_item” method. Once added to our settings.py file, the pipeline object will be instantiated by the crawler and its “process_item” method will be called for each MTQInfraItem. You can then save the item, change it or discard it so other pipelines won’t process it.

    Exporters are objects with predefined methods that can be used to persist data in a specific format. Scrapy comes with predefined exporters for CSV, JSON, LineJSON, XML, Pickle (Python) and Pretty Print. You can easily subclass these to modify some of their behavior or subclass the BaseItemExporter class to create your own exporter. On our case, we’ll do both.

    Here’s what our exporters.py file looks like:

    01#!/usr/bin/env python
    02# encoding=utf-8
    03 
    04from scrapy.contrib.exporter import CsvItemExporter
    05from scrapy.contrib.exporter import JsonItemExporter
    06from scrapy.contrib.exporter import JsonLinesItemExporter
    07from scrapy.contrib.exporter import XmlItemExporter
    08from scrapy.contrib.exporter import BaseItemExporter
    09 
    10import json
    11import simplekml
    12 
    13class MTQInfraXmlItemExporter(XmlItemExporter):
    14    def serialize_field(self, field, name, value):
    15        # Base XML exporter expects strings only. Convert any float or int to string.
    16        value = str(value)
    17        return super(MTQInfraXmlItemExporter, self).serialize_field(field, name, value)
    18 
    19class MTQInfraJsonItemExporter(JsonItemExporter):
    20    def __init__(self, file, **kwargs):
    21        # Base JSON exporter does not use dont_fail=True and I want to pass JSONEncoder args.
    22        self._configure(kwargs, dont_fail=True)
    23        self.file = file
    24        self.encoder = json.JSONEncoder(**kwargs)
    25        self.first_item = True
    26 
    27class MTQInfraKmlItemExporter(BaseItemExporter):
    28    def __init__(self, filename, **kwargs):
    29        self._configure(kwargs, dont_fail=True)
    30        self.filename = filename
    31        self.kml = simplekml.Kml()
    32        self.icon_styles = {}
    33 
    34    def _escape(self, str_value):
    35        # For now, we only deal with ampersand, the rest is properly escaped.
    36        return str_value.replace('&', '&')
    37 
    38    def start_exporting(self):
    39        pass
    40 
    41    def export_item(self, item):
    42        # ACTUAL CODE REMOVED FOR BLOG. PLEASE CHECK GITHUB REPO FOR SOURCE.
    43 
    44    def finish_exporting(self):
    45        # NOTE: The KML file is over 40Mb in size. The XML serializing will take a while and will
    46        #       probably get your laptop fan to start :-)
    47        self.kml.save(self.filename)

    The MTQInfraXmlItemExporter and MTQInfraJsonItemExporter are simply customized versions of their equivalent base Scrapy exporters. The MTQInfraKmlItemExporter is a custom exporter to save output in KML format. It uses the simplekml module. Almost all the work is done in export_item(), which is the method called for each MTQInfraItem created by our parsers. start_exporting/finish_exporting are, as their name imply, called at the start and finish and can be used to setup your exporter or finalize the export process respectively.

    Our pipelines.py file contains the following:

    01#!/usr/bin/env python
    02# encoding=utf-8
    03 
    04from scrapy.xlib.pydispatch import dispatcher
    05from scrapy import signals
    06from scrapy.exceptions import DropItem
    07from scrapy.contrib.exporter import CsvItemExporter
    08from scrapy.contrib.exporter import JsonLinesItemExporter
    09# Custom exporters
    10from exporters import MTQInfraJsonItemExporter
    11from exporters import MTQInfraXmlItemExporter
    12from exporters import MTQInfraKmlItemExporter
    13import csv
    14 
    15class MTQInfraPipeline(object):
    16    def __init__(self):
    17        self.fields_to_export = [
    18            'latitude',
    19            'longitude',
    20            'record_no',
    21            # MORE FIELDS IN THE REAL FILE. REMOVED FOR BLOG.
    22            'record_href',
    23            'location_href',
    24            'structure_type_img_href'
    25        ]
    26        dispatcher.connect(self.spider_opened, signals.spider_opened)
    27        dispatcher.connect(self.spider_closed, signals.spider_closed)
    28 
    29    def spider_opened(self, spider):
    30        self.csv_exporter = CsvItemExporter(open(spider.name+".csv", "w"),
    31                                            fields_to_export=self.fields_to_export, quoting=csv.QUOTE_ALL)
    32        self.json_exporter = MTQInfraJsonItemExporter(open(spider.name+".json", "w"),
    33                                                      fields_to_export=self.fields_to_export,
    34                                                      sort_keys=True, indent=4)
    35        self.jsonlines_exporter = JsonLinesItemExporter(open(spider.name+".linejson", "w"),
    36                                                        fields_to_export=self.fields_to_export)
    37 
    38        self.xml_exporter = MTQInfraXmlItemExporter(open(spider.name+".xml", "w"),
    39                                                    fields_to_export=self.fields_to_export,
    40                                                    root_element="structures", item_element="structure")
    41        # Make a quick copy of the list
    42        kml_fields = self.fields_to_export[:]
    43        kml_fields.append('fusion_marker')
    44        self.kml_exporter = MTQInfraKmlItemExporter(spider.name+".kml", fields_to_export=kml_fields)
    45        self.csv_exporter.start_exporting()
    46        self.json_exporter.start_exporting()
    47        self.jsonlines_exporter.start_exporting()
    48        self.xml_exporter.start_exporting()
    49        self.kml_exporter.start_exporting()
    50 
    51    def process_item(self, item, spider):
    52        self.csv_exporter.export_item(item)
    53        self.json_exporter.export_item(item)
    54        self.jsonlines_exporter.export_item(item)
    55        self.xml_exporter.export_item(item)
    56        # Add fusion_marker to KML for use in Google Fusion Table
    57        if item['ai_code'] == "no_restriction":
    58            item['fusion_marker'] = "small_green"
    59        elif item['ai_code'] == "restricted":
    60            item['fusion_marker'] = "small_yellow"
    61        elif item['ai_code'] == "closed":
    62            item['fusion_marker'] = "small_red"
    63        else:
    64            item['fusion_marker'] = "small_blue"
    65        self.kml_exporter.export_item(item)
    66        return item
    67 
    68    def spider_closed(self, spider):
    69        self.csv_exporter.finish_exporting()
    70        self.json_exporter.finish_exporting()
    71        self.jsonlines_exporter.finish_exporting()
    72        self.xml_exporter.finish_exporting()
    73        self.kml_exporter.finish_exporting()

    Some notes on the code:

    Line 20 and other fields_to_export-related lines: This is used to export fields in a certain order and to exclude the fusion_marker field from all but the KML output.

    Lines 29-30: These lines connect Scrapy events to our pipeline. In this case, the spider_opened and spider_closed methods will be called on “start/stop” of the spider, allowing us to setup our exporters and call their start_exporting/finish_exporting methods.

    Lines 54-69: This method, as mentioned above, is called for each item created by our parsers. In turns, it calls the export_item method of each exporter.

    In order for Scrapy to use our pipeline, we need to add the following lines to settings.py:

    # Our do-it-all pipeline
    ITEM_PIPELINES = [
        'mtqinfra.pipelines.MTQInfraPipeline'
    ]

    Running It For Real

    When ready to run your scraper on thousand of pages, I suggest you add the following to settings.py:

    LOG_FILE = 'mtqinfra.log'

    Or use the –logfile option when running “scrapy crawl”. This will save the Scrapy output to the specified log file. If you still want to see things flowing on your terminal, do a “tail -f” on the log on another terminal, this way you get the best of both worlds.

    As mentioned in “Testing It”, be polite. Try to make sure your code generate the proper output with a limited number of pages first. You don’t want to run your scraper for hours (this project does not take hours to crawl but this is an example) and then find out you forgot to include a field and need to reprocess each page.

    Also, try to scrape the website during the night, when your traffic has probably less impact.

    Finally, if you do run it and then realize your output has errors or needs to be changed, consider “reprocessing” your own results instead of scraping the website again (if possible). For this reason, I strongly suggest you always save your data in LineJSON format as it is super easy to reprocess. See next section for an example.

    Reprocessing Results if Needed

    If after scraping thousand of pages you realized you had a typo in a generated field (e.g. our KML popup), don’t rescrape the whole website again. Instead, consider reprocessing your own data. Of course, this can only be done if everything you need is already in your previously scraped data. If a field is missing completely and cannot be generated/computed, you’re out of luck.

    Here’s an example of how you could reprocess previously saved LineJSON data:

    01###
    02### ADD NECESSARY MODULE IMPORTS AND/OR MODIFIED EXPORTERS/PIPELINES HERE
    03### SEE SAMPLE reprocess_json.py FILE IN GITHUB PROJECT FOR MORE DETAILS.
    04###
    05 
    06### Create a fake spider object with any fields/methods needed by your exporters.
    07 
    08class FakeSpider(object):
    09    # Set spider name
    10    # NOTE: Make sure you don't use the same one as the original spider because you'll
    11    #       overwrite the previous data (and with this implementation, script will fail too).
    12    name = "mtqinfra-reprocessed"
    13 
    14### MAIN
    15 
    16# This is the previously scraped data
    17input_file = open("mtqinfra.linejson")
    18 
    19pipeline = MTQInfraPipeline()
    20pipeline.spider_opened(FakeSpider)
    21 
    22for line in input_file:
    23    item = MTQInfraItem(json.loads(line))
    24    pipeline.process_item(item, FakeSpider)
    25 
    26pipeline.spider_closed(FakeSpider)
    27input_file.close()

    The Source Code

    You can download the complete source code for the scraper on Github.

    Final Note

    Scrapy is a very powerful scraping framework. It does much more than what I use in this project. Have a look at the documentation to learn more.

    I’ll stop here as writing this post actually took more time than coding the project itself. Yes, I’m serious. This either shows you how powerful Python+Scrapy are, or how much I suck at writing blog posts :-)

    I hope someone will find this useful. Feel free to share in the comments section.

  • 相关阅读:
    【leetcode】92. 反转链表 II
    【leetcode】91. 解码方法
    【leetcode】89. 格雷编码
    【leetcode】86. 分隔链表
    【leetcode】82. 删除排序链表中的重复元素 II
    为什么选择react
    React 全家桶实现后台管理界面
    前后端同构
    浅谈React前后端同构防止重复渲染
    由React引发的前后端分离架构的思考
  • 原文地址:https://www.cnblogs.com/lexus/p/2402802.html
Copyright © 2011-2022 走看看