Subversion Repositories SmartDukaan

import urllib2
from BeautifulSoup import BeautifulSoup
import re
from sys import exit

class FlipkartScraper:
    def __init__(self):
        self.count_trials = 0
    
    def read(self, url):
        request = urllib2.Request(url)
        request.add_header('User-Agent', 'Mozilla/5.0 (X11; Linux i686) AppleWebKit/535.1 (KHTML, like Gecko) Chrome/13.0.782.218 Safari/535.1')
        opener = urllib2.build_opener()
        response_data = ""
        try:
            response_data = opener.open(request).read()
            print "Fetched response from flipkart for %s" %(url)
            
        except urllib2.HTTPError as e:
            print 'ERROR: ', e
            print 'Retrying'
            self.count_trials += 1
            
            if self.count_trials < 3:
                return self.read(url)
        
        self.response_data=response_data
        return self.createData(url)
    
    def createData(self,url):
        page=self.response_data.decode("utf-8")
        self.soup = BeautifulSoup(page,convertEntities=BeautifulSoup.HTML_ENTITIES)
        page = None
        self.response_data = None
        print "Soup created from flipkart data for %s" %(url)
        return self.scrape(self.soup,url)
    
    
    def scrape(self,soup,url):
        info = []
        oddSeller = soup.findAll("div" , {"class" : "line seller-item odd "})
        for data in oddSeller:
            temp={}
            try:
                businessDays = data.find('span', attrs={'class' : re.compile('fk-deliverable.*')})
                shippingTime = businessDays.find('span', attrs={'class' : re.compile('fk-bold')}).string.replace('to','').replace('business days.','').strip().replace('  ','-')
                temp['shippingTime']=shippingTime
            except:
                pass
            price = data.find('span', attrs={'class' : re.compile('pxs-final-price.*')}).string.strip('Rs.').strip()
            temp['sellingPrice']=float(price)
            for sellerInfo in data.findAll("div",{"class":re.compile(".*seller-info*")}):
                sellerName = sellerInfo.find('a').string
                temp['sellerName'] = sellerName
            for metrics in data.find("div",{"class":"fk-text-right"}):
                try:
                    metric = metrics.findAll('input', {'type': 'submit'})
                except AttributeError:
                    continue
                try:
                    inputTags = metric[0]['data-lst-buytrend']
                except TypeError:
                    continue
                dataMetrics = metric[0]['data-listing-metrics']
                try:
                    buyTrend = inputTags[0:str(inputTags).index('NWSR')].replace('_','')
                except ValueError:
                    buyTrend = inputTags[0:str(inputTags).index('WSR')].replace('_','')
                temp['buyTrend']=buyTrend
                dataMetric = dataMetrics.split(';')
                sellerCode = dataMetric[0]
                temp['sellerCode']=sellerCode
                temp['sellingPriceMetric'] = float(dataMetric[1])
                if not temp.has_key('shippingTime'):
                    print "Populating shipping time from metrics"
                    temp['shippingTime'] = dataMetric[3]
                temp['sellerScore'] = int(dataMetric[4])
                info.append(temp)
        evenSeller = soup.findAll("div" , {"class" : "line seller-item even "})
        for data in evenSeller:
            temp={}
            price = data.find('span', attrs={'class' : re.compile('pxs-final-price.*')}).string.strip('Rs.')
            try:
                businessDays = data.find('span', attrs={'class' : re.compile('fk-deliverable.*')})
                shippingTime = businessDays.find('span', attrs={'class' : re.compile('fk-bold')}).string.replace('to','').replace('business days.','').strip().replace('  ','-')
                temp['shippingTime']=shippingTime
            except:
                pass
            temp['sellingPrice']=float(price)
            for sellerInfo in data.findAll("div",{"class":re.compile(".*seller-info*")}):
                sellerName = sellerInfo.find('a').string
                temp['sellerName'] = sellerName
            for metrics in data.find("div",{"class":"fk-text-right"}):
                try:
                    metric = metrics.findAll('input', {'type': 'submit'})
                except AttributeError:
                    continue
                try:
                    inputTags = metric[0]['data-lst-buytrend']
                except TypeError:
                    continue
                dataMetrics = metric[0]['data-listing-metrics']
                try:
                    buyTrend = inputTags[0:str(inputTags).index('NWSR')].replace('_','')
                except ValueError:
                    buyTrend = inputTags[0:str(inputTags).index('WSR')].replace('_','')
                temp['buyTrend']=buyTrend
                dataMetric = dataMetrics.split(';')
                temp['sellerCode'] = dataMetric[0] 
                temp['sellingPriceMetric'] = float(dataMetric[1])
                if not temp.has_key('shippingTime'):
                    print "Populating shipping time from metrics"
                    temp['shippingTime'] = dataMetric[3]
                temp['sellerScore'] = int(dataMetric[4])
                info.append(temp)
        print "Returning Json response from flipkart for %s" %(url)
        return info

if __name__ == '__main__':
    scraper = FlipkartScraper()
    print scraper.read('http://www.flipkart.com/ps/ACCDSN84XXH5P9WG')