fmartingr
/
amiibofindr
mirror of https://github.com/fmartingr/amiibofindr.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Fixed keeping trade flag when you dont have an item

This commit is contained in:
Felipe Martín 2015-09-11 18:46:36 +02:00
parent 0fbda0856d
commit 2ee32ae1f3
2 changed files with 13 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
amiibofindr/apps/amiibo/services.py
View File

@ -3,6 +3,11 @@
from . import models
def is_owned_by(amiibo, user):
relation, created = models.UserAmiibo.objects.get_or_create(
_amiibo=amiibo, user=user)
return relation.owned
def user_add_owned(user, amiibo):
relation, created = models.UserAmiibo.objects.get_or_create(
_amiibo=amiibo, user=user)
@ -14,6 +19,7 @@ def user_remove_owned(user, amiibo):
relation, created = models.UserAmiibo.objects.get_or_create(
_amiibo=amiibo, user=user)
relation.own = False
relation.trade = False
relation.save()
def user_add_wishlist(user, amiibo):

11
amiibofindr/apps/amiibo/views.py
View File

@ -108,16 +108,19 @@ class UserAmiiboView(View):
# TODO: Add message
def add_trade(self, request, amiibo):
services.user_add_trade(request.user, amiibo)
if services.is_owned_by(amiibo, request.user):
services.user_add_trade(request.user, amiibo)
# TODO: Add message
def remove_trade(self, request, amiibo):
services.user_remove_trade(request.user, amiibo)
if services.is_owned_by(amiibo, request.user):
services.user_remove_trade(request.user, amiibo)
# TODO: Add message
def toggle_trade(self, request, amiibo):
services.user_toggle_trade(request.user, amiibo)
# TODO: Add message
if services.is_owned_by(amiibo, request.user):
services.user_toggle_trade(request.user, amiibo)
# TODO: Add message
def add_owned(self, request, amiibo):
services.user_add_owned(request.user, amiibo)