Update scraper.py

vicoliveres · Mar 8, 2018 · 871891d · 871891d
1 parent cfb68ea
commit 871891d
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/scraper.py b/scraper.py
@@ -16,6 +16,7 @@
     names = restaurant.cssselect("h2")
     addresses = restaurant.cssselect("div.address")
     postcodes = restaurant.cssselect("div.postcode")
+    companies = "Wagamama"
 
     for name in names:
         record['Name'] = name.text
@@ -25,8 +26,9 @@
 
     for postcode in postcodes:
         record['Postcode'] = postcode.text
-
-    record['Company'] = "Wagamama"
+
+    for company in companies:
+        record['Company'] = company
 
     print record, '------------'