Update scraper.py

vicoliveres · Mar 9, 2018 · 8f4cbc8 · 8f4cbc8
1 parent 82eb8c7
commit 8f4cbc8
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/scraper.py b/scraper.py
@@ -13,10 +13,13 @@
 
 for restaurant in restaurants:
     record = {}
+    name = restaurant.cssselect("h2")
+    address = restaurant.cssselect("div.address")
+    postcode = restaurant.cssselect("div.address div")
 
-    record['Name'] = restaurant.cssselect("h2").text
-    record['Address'] = restaurant.cssselect("div.address").text
-    record['Postcode'] = restaurant.cssselect("div.address div").text
+    record['Name'] = name.text
+    record['Address'] = address.text
+    record['Postcode'] = postcode.text
     record['Company'] = "Wagamama"
 
     print record, '------------'