-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathtrustbasket_spider.py
130 lines (114 loc) · 6.42 KB
/
trustbasket_spider.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
import scrapy
import csv
import json
import re
from html.parser import HTMLParser
def remove_non_ascii(text):
return ''.join([i if ord(i) < 128 else '' for i in text])
class MLStripper(HTMLParser):
def __init__(self):
self.reset()
self.strict = False
self.convert_charrefs= True
self.fed = []
def handle_data(self, d):
self.fed.append(d)
def get_data(self):
return '|'.join(self.fed)
def strip_tags(html):
s = MLStripper()
s.feed(html)
return s.get_data()
class HCVID285_trustbasket_com_Spider(scrapy.Spider):
name = "HCVID285.trustbasket.com"
products_seen = []
jsonData = []
def closed(self, reason):
toWrite = {'data' : HCVID285_trustbasket_com_Spider.jsonData}
with open('./json/HCVID285.trustbasket.com.json', 'w+') as outfile:
json.dump(toWrite, outfile, ensure_ascii=False)
def start_requests(self):
url = "http://www.trustbasket.com/"
yield scrapy.Request(url=url, callback=self.parse)
def parse(self, response):
x = response.css('ul#accessibleNav > li')
for columns in x:
head_category = columns.css('a::text').extract_first().strip()
if head_category != 'Home' and head_category != 'Composter' and head_category != 'Terracotta' and head_category != 'Plants' and head_category != 'More':
list_items = columns.css('ul > li')
for item in list_items:
page = item.css('a::attr(href)').extract_first()
sub_category = item.css('a::text').extract_first().strip()
if page is not None:
print(page)
request = scrapy.Request(url= 'http://www.trustbasket.com' + page + '?page=1', callback = self.parsePage)
request.meta['pagenum'] = 1
request.meta['category'] = str(head_category) + '|' + str(sub_category)
yield request
page = 'http://www.trustbasket.com/pages/indoor-bokashi-composter'
request = scrapy.Request(url=page+'?page=1', callback = self.parsePage)
request.meta['pagenum'] = 1
request.meta['category'] = 'Composter'
yield request
page = 'http://www.trustbasket.com/collections/terracotta-clay-planters-pots'
request = scrapy.Request(url=page+'?page=1', callback = self.parsePage)
request.meta['pagenum'] = 1
request.meta['category'] = 'Terracotta'
yield request
page = 'http://www.trustbasket.com/collections/plants'
request = scrapy.Request(url=page+'?page=1', callback = self.parsePage)
request.meta['pagenum'] = 1
request.meta['category'] = 'Plants'
yield request
def parsePage(self, response):
products = response.css('div.grid.grid-border div.grid-uniform > div.grid-item')
print(len(products),response.request.url)
if len(products) == 1 and 'Sorry' in products.css('p::text').extract_first().strip():
return
else:
for product in products:
product_link = 'http://www.trustbasket.com' + product.css('a::attr(href)').extract_first()
product_name = product.css('p::text').extract_first()
product_image = product.css('img::attr(src)').extract_first()
request = scrapy.Request(url=product_link, callback=self.parseProduct)
request.meta['product_link'] = product_link
request.meta['product_image'] = product_image
request.meta['product_name'] = product_name
request.meta['product_category'] = response.meta['category']
yield request
nextpage = response.meta['pagenum'] + 1
product_link = response.request.url[:response.request.url.index('?page=')] + '?page=' + str(nextpage)
request = scrapy.Request(url=product_link, callback=self.parsePage)
request.meta['category'] = response.meta['category']
request.meta['pagenum'] = nextpage
yield request
def parseProduct(self, response):
productLink = response.meta['product_link']
productCategory = response.meta['product_category']
productImage = response.meta['product_image']
productName = response.meta['product_name']
productPrice = response.css('meta[itemprop="price"]::attr(content)').extract_first().replace('Rs.','').strip()
productDesc = response.css('div.product-description').extract_first()
if productDesc:
productDesc = strip_tags(productDesc)
content = response.css('script[type=text\/javascript]').extract_first()
data = json.loads(re.search('var meta = (.*?);', content).group(1))
productSku = data["product"]["variants"][0]["sku"]
productCategory = productCategory.replace('"','``')
productDesc = re.sub('\n' ,'|',productDesc)
productDesc = re.sub('\t','',productDesc)
productDesc = re.sub('\r','',productDesc)
productDesc = re.sub('[ ]+',' ',productDesc)
productDesc = re.sub('(\| )+','|',productDesc)
productDesc = re.sub('\|+','|',productDesc)
productDesc = productDesc.strip('|')
productDesc = productDesc.replace('"','``')
productName = productName.replace('"','``')
productDesc = remove_non_ascii(productDesc)
productCategory = remove_non_ascii(productCategory)
productName = remove_non_ascii(productName)
with open('HCVID285.trustbasket.com.csv', 'a', newline='') as csvfile:
fieldnames = ['Title', 'Category', 'URL', 'IMG_SRC', 'Merchant', 'Price', 'SKU', 'Description']
writer = csv.DictWriter(csvfile, fieldnames=fieldnames)
writer.writerow({'Title': '"'+str(productName)+'"', 'IMG_SRC': '"'+str(productImage)+'"', 'Description': '"'+str(productDesc)+'"','Price': '"'+str(productPrice)+'"', 'URL': '"'+str(productLink)+'"', 'Category': '"'+str(productCategory)+'"', 'SKU': '"'+str(productSku)+'"', 'Merchant': '"NA"'})
HCVID285_trustbasket_com_Spider.jsonData.append({'Title': productName, 'Description': productDesc, 'IMG_SRC': productImage, 'Price': productPrice, 'SKU': productSku, 'Category': productCategory, 'URL': productLink, 'Merchant': 'NA'})