def grab_daily():
success = 0
for site_name in URLS:
if 'discover.com' in site_name:
grabber = XmlGrabber(site_name)
elif 'shop.upromise.com' in site_name:
grabber = ShopGrabber(site_name)
elif site_name in ['shop.amtrakguestrewards.com', 'shop.lifemiles.com']:
grabber = RetailersGrabber(site_name)
# elif 'www.bestbuy.com' in site_name:
# grabber = BestbuyGrabber(site_name)
else:
grabber = UltimateRewardsGrabber(site_name)
if grabber.grab():
success += 1
# Now it's time to check modifications and send email if changed
check_modification()
return 'OK'
def grab():
# Grab data
if request.method == 'POST':
site_name = request.form['site_name']
if 'discover.com' in site_name:
grabber = XmlGrabber(site_name)
elif 'shop.upromise.com' in site_name:
grabber = ShopGrabber(site_name)
elif 'www.bestbuy.com' in site_name:
grabber = BestbuyGrabber(site_name)
elif site_name in ['shop.amtrakguestrewards.com', 'shop.lifemiles.com']:
grabber = RetailersGrabber(site_name)
else:
grabber = UltimateRewardsGrabber(site_name)
result_id = grabber.grab()
flash(u'Successfully grabbed')
return result_id
