diff --git a/amiibofindr/apps/amiibo/management/commands/fetch_price.py b/amiibofindr/apps/amiibo/management/commands/fetch_price.py index 14c9ba3..b4869b7 100644 --- a/amiibofindr/apps/amiibo/management/commands/fetch_price.py +++ b/amiibofindr/apps/amiibo/management/commands/fetch_price.py @@ -14,10 +14,15 @@ from amiibofindr.apps.shop.crawlers import Crawler class Command(BaseCommand): def handle(self, *args, **kwargs): - regions = Shop.objects.all().order_by('flag_code').distinct('flag_code').values_list('flag_code', 'slug') + regions = Shop.objects.all()\ + .order_by('flag_code')\ + .distinct('flag_code')\ + .values_list('flag_code', 'slug') for region in regions: - item_codes = AmiiboShop.objects.filter(shop__flag_code=region[0]).values_list('item_id', flat=True) + item_codes = AmiiboShop.objects.filter( + shop__flag_code=region[0], + check_price=True).values_list('item_id', flat=True) amazon = Crawler(region[1]) products = amazon.fetch_batch(item_codes) for product in products: diff --git a/amiibofindr/apps/amiibo/migrations/0018_amiiboshop_check_price.py b/amiibofindr/apps/amiibo/migrations/0018_amiiboshop_check_price.py new file mode 100644 index 0000000..fe26ae8 --- /dev/null +++ b/amiibofindr/apps/amiibo/migrations/0018_amiiboshop_check_price.py @@ -0,0 +1,19 @@ +# -*- coding: utf-8 -*- +from __future__ import unicode_literals + +from django.db import models, migrations + + +class Migration(migrations.Migration): + + dependencies = [ + ('amiibo', '0017_auto_20150625_1806'), + ] + + operations = [ + migrations.AddField( + model_name='amiiboshop', + name='check_price', + field=models.BooleanField(default=True), + ), + ] diff --git a/amiibofindr/apps/amiibo/models.py b/amiibofindr/apps/amiibo/models.py index b64e28b..8ec3ffa 100644 --- a/amiibofindr/apps/amiibo/models.py +++ b/amiibofindr/apps/amiibo/models.py @@ -104,6 +104,7 @@ class AmiiboShop(models.Model): shop = models.ForeignKey('shop.Shop', related_name='amiibos_set') url = models.TextField() item_id = models.CharField(max_length=64) + check_price = models.BooleanField(default=True) class Meta: ordering = ('shop__name', ) diff --git a/amiibofindr/db.sqlite3 b/amiibofindr/db.sqlite3 new file mode 100644 index 0000000..612648f Binary files /dev/null and b/amiibofindr/db.sqlite3 differ