36 lines
No EOL
1.5 KiB
Python
36 lines
No EOL
1.5 KiB
Python
from django.shortcuts import render, redirect
|
|
from .models import *
|
|
from users.models import UserInventory
|
|
from django.contrib.auth.decorators import login_required
|
|
from django.contrib import messages
|
|
|
|
# Create your views here.
|
|
def home(request):
|
|
items = Item.objects.filter(is_active=True)
|
|
print(items)
|
|
return render(request, 'shop/home.html', {'items': items})
|
|
|
|
@login_required
|
|
def buy(request, item_id):
|
|
item = Item.objects.get(id=item_id)
|
|
# ON vérifie si l'utilisateur a assez d'argent pour acheter l'item
|
|
if request.user.money >= item.price:
|
|
# On vérifie que l'utilisateur n'a pas déjà acheté l'item
|
|
if not request.user.inventory.filter(item=item).exists():
|
|
# On déduit le prix de l'item de l'argent de l'utilisateur
|
|
user_inventory = UserInventory.objects.get(user=request.user, item__name='Or')
|
|
user_inventory.quantity -= item.price
|
|
user_inventory.save()
|
|
|
|
# On ajoute l'item dans l'inventaire de l'utilisateur
|
|
user_inventory = UserInventory.objects.create(user=request.user, item=item)
|
|
user_inventory.quantity = 1
|
|
user_inventory.save()
|
|
messages.success(request, f"Vous avez acheté {item.name} pour {item.price} pièces d'or.")
|
|
else:
|
|
messages.error(request, f"Vous avez déjà acheté {item.name}.")
|
|
else:
|
|
messages.error(request, f"Vous n'avez pas assez d'argent pour acheter {item.name}.")
|
|
return redirect('shop_home')
|
|
|
|
|