Added better referer link generation based on field

amiibofindr/apps/amiibo/models.py

@@ -109,7 +109,9 @@ class AmiiboShop(models.Model):
         ordering = ('shop__name', )
 
     def get_url(self):
-        return amazonify(self.url, settings.AMAZON_ASSOC_TAG)
+        if self.shop.referer_id:
+            return amazonify(self.url, self.shop.referer_id)
+        return self.url
 
     def update_price(self, price, currency):
         price_obj, is_new = AmiiboPrice.objects.get_or_create(

amiibofindr/apps/shop/migrations/0004_shop_visible.py

@@ -0,0 +1,19 @@
+# -*- coding: utf-8 -*-
+from __future__ import unicode_literals
+
+from django.db import models, migrations
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('shop', '0003_auto_20150624_0049'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='shop',
+            name='visible',
+            field=models.BooleanField(default=True),
+        ),
+    ]

amiibofindr/apps/shop/models.py

@@ -13,6 +13,7 @@ class Shop(models.Model):
     url = models.CharField(max_length=64)
     flag_code = models.CharField(max_length=2, blank=True, null=True)
     referer_id = models.CharField(max_length=64, blank=True, null=True)
+    visible = models.BooleanField(default=True)
 
     class Meta:
         ordering = ('name', )