Fixed keeping trade flag when you dont have an item
This commit is contained in:
parent
0fbda0856d
commit
2ee32ae1f3
|
@ -3,6 +3,11 @@
|
||||||
from . import models
|
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):
|
def user_add_owned(user, amiibo):
|
||||||
relation, created = models.UserAmiibo.objects.get_or_create(
|
relation, created = models.UserAmiibo.objects.get_or_create(
|
||||||
_amiibo=amiibo, user=user)
|
_amiibo=amiibo, user=user)
|
||||||
|
@ -14,6 +19,7 @@ def user_remove_owned(user, amiibo):
|
||||||
relation, created = models.UserAmiibo.objects.get_or_create(
|
relation, created = models.UserAmiibo.objects.get_or_create(
|
||||||
_amiibo=amiibo, user=user)
|
_amiibo=amiibo, user=user)
|
||||||
relation.own = False
|
relation.own = False
|
||||||
|
relation.trade = False
|
||||||
relation.save()
|
relation.save()
|
||||||
|
|
||||||
def user_add_wishlist(user, amiibo):
|
def user_add_wishlist(user, amiibo):
|
||||||
|
|
|
@ -108,16 +108,19 @@ class UserAmiiboView(View):
|
||||||
# TODO: Add message
|
# TODO: Add message
|
||||||
|
|
||||||
def add_trade(self, request, amiibo):
|
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
|
# TODO: Add message
|
||||||
|
|
||||||
def remove_trade(self, request, amiibo):
|
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
|
# TODO: Add message
|
||||||
|
|
||||||
def toggle_trade(self, request, amiibo):
|
def toggle_trade(self, request, amiibo):
|
||||||
services.user_toggle_trade(request.user, amiibo)
|
if services.is_owned_by(amiibo, request.user):
|
||||||
# TODO: Add message
|
services.user_toggle_trade(request.user, amiibo)
|
||||||
|
# TODO: Add message
|
||||||
|
|
||||||
def add_owned(self, request, amiibo):
|
def add_owned(self, request, amiibo):
|
||||||
services.user_add_owned(request.user, amiibo)
|
services.user_add_owned(request.user, amiibo)
|
||||||
|
|
Loading…
Reference in New Issue