Added context processors with a list of volume ids owned and wish listed by the user

Using that new context variables to do the needed check to complete the add/remove/wishlist UI

shelfzilla/apps/manga/context_processors.py

@@ -0,0 +1,19 @@
+from .models import UserHaveVolume, UserWishlistVolume
+
+
+def user_have_volumes(request):
+    result = []
+    if request.user:
+        result = UserHaveVolume.objects.filter(
+            user=request.user).values_list('volume', flat=True)
+
+    return {'user_have_volumes': result}
+
+
+def user_wishlisted_volumes(request):
+    result = []
+    if request.user:
+        result = UserWishlistVolume.objects.filter(
+            user=request.user).values_list('volume', flat=True)
+
+    return {'user_wishlisted_volumes': result}

shelfzilla/apps/manga/models.py

@@ -80,7 +80,7 @@ class UserHaveVolume(models.Model):
 
     def __unicode__(self):
         return "{} {} {}".format(
-            self.user.name,
+            self.user.username,
             _('have'),
             self.volume
         )
@@ -93,7 +93,7 @@ class UserWishlistVolume(models.Model):
 
     def __unicode__(self):
         return "{} {} {}".format(
-            self.user.name,
+            self.user.username,
             _('wants'),
             self.volume
         )

shelfzilla/settings/base.py

@@ -70,6 +70,8 @@ TEMPLATE_CONTEXT_PROCESSORS = (
     'django.core.context_processors.request',
 
     'shelfzilla.context_processors.pjax',
+    'shelfzilla.apps.manga.context_processors.user_have_volumes',
+    'shelfzilla.apps.manga.context_processors.user_wishlisted_volumes',
 )
 
 MIDDLEWARE_CLASSES = (

shelfzilla/themes/bootflat/templates/manga/series/includes/volume.html

@@ -1,7 +1,7 @@
 {% load i18n thumbnail %}
-<div class="well text-center volume-item">
-    <div class="badges {% if volume in user.have_volumes %}user-have-it{% endif %}">
-        {% if volume in user.have_volumes %}
+<div class="well text-center volume-item {% if volume.pk in user_have_volumes %}user-have-it{% endif %}">
+    <div class="badges">
+        {% if volume.pk in user_have_volumes %}
         <a href="{% url "volume.have_it" volume.pk %}" data-pjax="v{{ volume.pk }}" pjax-nopush pjax-messages>
             <span class="badge badge-success">
                 <i class="glyphicon glyphicon-ok"></i>
@@ -14,7 +14,7 @@
             </span>
         </a>
         <a href="{% url "volume.wishlist" volume.pk %}" data-pjax="v{{ volume.pk }}" pjax-messages pjax-nopush>
-            <span class="badge {% if volime in user.whislisted_volumes %}badge-warning{% endif %}">
+            <span class="badge {% if volume.pk in user_wishlisted_volumes %}badge-warning{% endif %}">
                 <i class="glyphicon glyphicon-star"></i>
             </span>
         </a>