Subversion Repositories SmartDukaan

'''

Created on 14-May-2010

@author: gaurav

'''

from scrapy.spider import BaseSpider

from scrapy.selector import HtmlXPathSelector

from scrapy.http import Request

from demo.items import DemoItem

from scrapy.contrib.spidermiddleware import referer

from scrapy.http.headers import Headers

from scrapy.http.request.form import FormRequest

from scrapy.log import msg

from scrapy.http.response import Response

from datastore.DataCodeAccessor import *

from datastore.DataAccessor import *

from html2text.unescaping import *

class vendor_links(BaseSpider):

    """

    Documentation for class vendor_links

    This spider collects the url for the individual vendors 

    and store them in table datastore_datadefinition_univercell_data.

    """

    def __init__(self):

        """

        Documentation for constructor

        initialize_table is called to make all the tables known in

        the scope of this class.

        Also start url needs to be feeded to the spider through start_urls.append

        Domainname is name by which this spider is known outside

        So this will be used as an argument for calling this spider 

        """

        initialize_table()

        #UNIVERCELL_DOMAINNAME = "univercell"   

        UNIVERCELL_DOMAINNAME = get_code_word("UNIVERCELL_DOMAINNAME")

        self.domain_name = UNIVERCELL_DOMAINNAME 

        #UNIVERCELL_URL = "http://www.univercell.in/mobiles/populateStore.action"

        UNIVERCELL_URL = get_code_word("UNIVERCELL_URL")

        self.start_urls.append(UNIVERCELL_URL)

    def start_requests(self):

        """

        Documentation for method start_requests

        To set various properties of the request to be made

        like referer, headers and all.

        Also suppliers entry need to be done in the table

        datastore_datadefinition_suppliers.

        @return a list of well formed requests which will be 

        crawled by spider and spider will return the response

        """

        #adding entry for the supplier i.e its name and site

        #UNIVERCELL_HOMEPAGE = "http://www.univercell.in"

        UNIVERCELL_HOMEPAGE = get_code_word("UNIVERCELL_HOMEPAGE")

        da = DataHelper()

        da.add_supplier(self.domain_name, UNIVERCELL_HOMEPAGE)

        listreq = []

        #for each request a referer has to be set

        #UNIVERCELL_REFERER = "www.google.com/search"

        UNIVERCELL_REFERER = get_code_word("UNIVERCELL_REFERER")

        for url1 in self.start_urls:

            request = Request(url = str(url1), callback=self.parse)

            request.headers.setdefault("Referer", UNIVERCELL_REFERER)

            listreq.append(request)

        return listreq

    def parse(self, response):

        """

        Documentation for method parse

        @param response of individual requests

        Using Xpaths needed information is extracted out of the response

        and added to the database

        Xpath1 = Give us section for individual vendors

        Xpath2 = Give us name for individual vendors

        Xpath3 = Give us url for individual vendors

        Url1 = To get full url for individual vendors

        var1,var2,var3 and var4 are used to get proper url

        """

        #url1 needed to get complete urls for phones

        #UNIVERCELL_URL1 = "http://www.univercell.in"

        UNIVERCELL_URL1 = get_code_word("UNIVERCELL_URL1")

        hxs = HtmlXPathSelector(response)

        #UNIVERCELL_XPATH1 = '//div[@id="mobilesTab"]/table/tr[1]/td/table/tr'

        UNIVERCELL_XPATH1 = get_code_word("UNIVERCELL_XPATH1")

        vendor_info = hxs.select(UNIVERCELL_XPATH1)

        items = []

        for i in vendor_info:

            item = {}

            #UNIVERCELL_XPATH2 = './/a/text()'

            UNIVERCELL_XPATH2 = get_code_word("UNIVERCELL_XPATH2")

            #UNIVERCELL_XPATH3 = './/a/@href' 

            UNIVERCELL_XPATH3 = get_code_word("UNIVERCELL_XPATH3")

            item['name'] = i.select(UNIVERCELL_XPATH2)[0].extract()

            temp = i.select(UNIVERCELL_XPATH3)[0].extract()

            #site having data has url containing repopulate instead of populate

            #UNIVERCELL_VAR1 = ";"

            UNIVERCELL_VAR1 = get_code_word("UNIVERCELL_VAR1")

            #UNIVERCELL_VAR2 = "?"

            UNIVERCELL_VAR2 = get_code_word("UNIVERCELL_VAR2")

            #UNIVERCELL_VAR3 = "populate"

            UNIVERCELL_VAR3 = get_code_word("UNIVERCELL_VAR3")

            #UNIVERCELL_VAR4 = "rePopulate"

            UNIVERCELL_VAR4 = get_code_word("UNIVERCELL_VAR4")

            a = str(temp).find(UNIVERCELL_VAR1)

            b = str(temp).find(UNIVERCELL_VAR2)

            temp1 = str(temp)[a:b]

            temp2 = str(temp).replace(temp1,"")

            item['site'] =  str(temp2).replace(UNIVERCELL_VAR3,UNIVERCELL_VAR4)

            items.append(item)

        da = DataHelper()

        for item in items:

            str2 = UNIVERCELL_URL1 + str(item['site'])

            da.add_univervendor( unescape(item['name'].strip()), unescape(str2))

SPIDER = vendor_links()