Added buttons for useramiibo relation

amiibofindr/apps/amiibo/context_processors.py

@@ -1,7 +1,7 @@
 # coding: utf-8
 
 # amiibo
-from .models import Collection, AmiiboPrice, AmiiboCard
+from .models import Collection, AmiiboPrice, AmiiboCard, UserAmiibo
 
 
 def collections(request):
@@ -19,3 +19,19 @@ def currencies(request):
         'CURRENCIES': AmiiboPrice.objects.all().distinct('currency')\
             .values_list('currency', flat=True)
     }
+
+
+def user_amiibo(request):
+    owned = []
+    wishlisted = []
+    trading = []
+    if request.user.is_authenticated():
+        owned  = UserAmiibo.objects.filter(own=True).values_list('id', flat=True)
+        wishlisted = UserAmiibo.objects.filter(want=True).values_list('id', flat=True)
+        trading = UserAmiibo.objects.filter(trade=True).values_list('id', flat=True)
+
+    return {
+        'USER_AMIIBO_OWNED': owned,
+        'USER_AMIIBO_WISHLIST': wishlisted,
+        'USER_AMIIBO_TRADE': trading
+    }

amiibofindr/apps/amiibo/migrations/0024_auto_20150908_2114.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', '0023_amiiboshop_image'),
+    ]
+
+    operations = [
+        migrations.RenameField(
+            model_name='useramiibo',
+            old_name='have',
+            new_name='own',
+        ),
+    ]

amiibofindr/apps/amiibo/models.py

@@ -304,7 +304,7 @@ class UserAmiibo(models.Model):
     _amiibo = models.ForeignKey(Amiibo)
 
     want = models.BooleanField(default=False)
-    have = models.BooleanField(default=False)
+    own = models.BooleanField(default=False)
     trade = models.BooleanField(default=False)
     date = models.DateTimeField(auto_now=True)
 

amiibofindr/settings/base.py

@@ -96,6 +96,7 @@ TEMPLATES = [
                 'amiibofindr.apps.core.context_processors.debug',
                 'amiibofindr.apps.core.context_processors.files',
                 'amiibofindr.apps.core.context_processors.i18n',
+                'amiibofindr.apps.amiibo.context_processors.user_amiibo',
             ],
         },
     },

amiibofindr/templates/amiibo/amiibo-figure.html

@@ -31,9 +31,31 @@ AMIIBO FIGURE
   <div class="ui page grid">
     <div class="sixteen wide column">
       <div class="ui segment">
-        <h2 class="ui header centered">
-          {{ amiibo.name }} - {{ amiibo.collection.name }}
-        </h2>
+        <div class="ui stackable grid">
+          <div class="eight wide column ui middle aligned">
+            <h2 class="ui header">
+              {{ amiibo.collection.name }}: {{ amiibo.name }}
+            </h2>
+          </div>
+          <div class="eight wide column right aligned">
+            {% if amiibo.pk in USER_AMIIBO_OWNED %}
+              <a href="#" class="ui button {% if amiibo.pk in USER_AMIIBO_TRADE %}teal{% endif %}">
+                <i class="{% if amiibo.pk in USER_AMIIBO_TRADE %}check{% else %}cancel{% endif %} icon"></i> Available for trade
+              </a>
+              <a href="#" class="ui button red">I do not have it</a>
+            {% else %}
+              <div class="ui buttons">
+                {% if amiibo.pk in USER_AMIIBO_WISHLIST %}
+                  <a class="ui button yellow">I do not want this</a>
+                {% else %}
+                  <a class="ui button">I want this</a>
+                {% endif %}
+                <div class="or" data-text="{% trans 'or' %}"></div>
+                <a class="ui positive button">I have this</a>
+            {% endif %}
+              </div>
+          </div>
+        </div>
       </div>
     </div>
   </div>
@@ -76,7 +98,7 @@ AMIIBO FIGURE
                           <i class="ui icon info circle blue"></i>
                           {% trans "Updated " %}<span data-relative="{{ shop_item.last_price.date.isoformat }}"></span><br>
                         {% else %}
-                           <i class="ui icon warning circle red"></i>
+                          <i class="ui icon warning circle red"></i>
                           {% trans "This item had never been in stock" %}
                         {% endif %}
                       </div>