from django import template
from card_search.views import Card, Deck, Quantity
from django.shortcuts import render_to_response
register = template.Library()
@register.simple_tag
def color_id(deck):
black=blue=red=white=green = 0
queryset = Quantity.objects.filter(deck__deck_name=deck.deck_name).all()
for item in queryset:
if item.card.mana_cost:
black += item.card.mana_cost.count("{B}") * item.qty
blue += item.card.mana_cost.count("{U}") * item.qty
red += item.card.mana_cost.count("{R}") * item.qty
white += item.card.mana_cost.count("{W}") * item.qty
green += item.card.mana_cost.count("{G}") * item.qty
xdata = ["Black", "Blue", "Red", "White", "Green"]
ydata = [black, blue, red, white, green]
chartdata = {'x': xdata, 'y': ydata}
charttype = 'pieChart'
chartcontainer ='piechart_container'
data = {
'charttype': charttype,
'chartdata': chartdata,
'chartcontainer': chartcontainer,
'extra': {
'x_is_date': False,
'x_axis_format': '',
'tag_script_js': True,
'jquery_on_ready': False,
}
}
return data