Django Wars [Parte 15]: Compra e venda de itens na loja

O Django Wars foi idealizado e desenvolvido no Django 1.4. Algumas funcionalidades aqui descritas podem não funcionar ou funcionar de forma diferente em outras versões do Django

No artigo anterior, criamos a página da loja, onde o usuário consegue ver uma listagem dos itens que a loja possui com a opção de comprar ou vender itens.
Agora, precisamos programar as ações da loja.
Ou seja, precisamos implementar a compra e a venda de itens.

No arquivo loja.html, no artigo anterior, definimos os links de compra e de venda de itens.

<div class="container">
   <p>Armaduras:</p>
   {% for armadura in armaduras %}
       <img src="{{ armadura.imagem.url }}" />
       {{ armadura.nome }}
 
       {% if armadura in player.armaduras.all %}
           <a href="vender/armadura/{{ armadura.pk }}">vender por {{ armadura.venda }}</a>
       {% else %}
           <a href="comprar/armadura/{{ armadura.pk }}">comprar por {{ armadura.compra }}</a>
       {% endif %}
       <br />
 
   {%  endfor %}
 
   <hr />
 
   <p>Armas:</p>
   {% for arma in armas %}
       <img src="{{ arma.imagem.url }}" />
       {{ arma.nome }}
 
       {% if arma in player.armas.all %}
           <a href="vender/arma/{{ arma.pk }}">vender por {{ arma.venda }}</a>
       {% else %}
           <a href="comprar/arma/{{ arma.pk }}">comprar por {{ arma.compra }}</a>
       {% endif %}
       <br />
   {%  endfor %}
 
</div> <!-- /container -->

Isso já é meio caminho andado.
Agora só precisamos definir as urls no urls.py e escrever as funções no views.py.

from django.conf.urls import patterns, include, url

import djangowars.views

# Uncomment the next two lines to enable the admin:
from django.contrib import admin
admin.autodiscover()

urlpatterns = patterns('',
                       url(r'^$', djangowars.views.index),
                       url(r'^registrar/$', djangowars.views.registrar), # pagina de cadastro
                       url(r'^login/$', djangowars.views.logar), # pagina de login

                       url(r'^crimes/$', djangowars.views.crimes),

                       url(r'^loja/$', djangowars.views.loja), # pagina de loja
                       url(r'^loja/comprar/armadura/(\d+)/$', djangowars.views.comprar_armadura), # pagina de loja
                       url(r'^loja/vender/armadura/(\d+)/$', djangowars.views.vender_armadura), # pagina de loja
                       #url(r'^compra/arma/(\d+)/$', djangowars.views.compra_arma), # pagina de loja
                       #url(r'^venda/arma/(\d+)/$', djangowars.views.venda_arma), # pagina de loja
                       
                       # Uncomment the next line to enable the admin:
                       url(r'^admin/', include(admin.site.urls)),
)

Reparem as linhas 16~20, onde definimos todos os links referentes a loja, e os associamos as suas respectivas funções.
Como no Django os links são definidos através de Expressões Regulares, utilizamos o (d+) para informar ao Django que esperamos que seja passado um ou mais dígitos [0-1].
Essa parte montada na url será passada para a view como uma variável da chamada da função.
Fica mais fácil de compreender isso descrevendo a própria função:

#pagina de compra de armaduras
def comprar_armadura(request, item):
    if not request.user.is_authenticated(): # Se o usuario nao esta logado
        return redirect(logar)

    armadura = get_object_or_404(Armadura, pk=item) #procura a armadura no banco de dados, se nao encontrar retorna uma pagina de 404
    player = request.user.get_profile() 

   #verifica se o usuario tem dinheiro para comprar o item
    if player.carteira >= armadura.compra:
        player.carteira = player.carteira - armadura.compra #tira o valor da armadura na carteira
        #coloca a armadura na lista de armaduras que o usuario possui
        player.armaduras.add(armadura) #https://docs.djangoproject.com/en/1.4/ref/models/relations/
        player.save() #salva no banco de dados as alteracoes no usuario

    return redirect(loja)

Notem como na definição do método comprar_armadura, além da variável que já é passada por padrão (request), temos uma nova variável (item), que foi passada a partir do urls.py.
O método para vender uma armadura é análogo ao de comprar uma.
A diferença encontra-se no fato que, ao invés de subtrair o valor do item na carteira, adicionamos este valor e, ao invés de adicionar a armadura na lista de armaduras do usuários, nós a removemos.

#pagina de venda de armaduras
def vender_armadura(request, item):
    if not request.user.is_authenticated():
        return redirect(logar)

    armadura = get_object_or_404(Armadura, pk=item)
    player = request.user.get_profile()

    if armadura in player.armaduras.all():
        player.carteira = player.carteira + armadura.venda
        player.armaduras.remove(armadura) #https://docs.djangoproject.com/en/1.4/ref/models/relations/
        player.save()

    return redirect(loja)

Definimos os links, e as views, para compra e venda de armaduras.
Para comprar e vender armas é a mesma coisa.
Fica então como exercício.