312 lines
12 KiB
Python
Executable File
312 lines
12 KiB
Python
Executable File
# coding: utf-8
|
||
from datetime import datetime
|
||
from datetime import timedelta
|
||
from itertools import groupby
|
||
import json
|
||
import time
|
||
|
||
from django.conf import settings
|
||
from django.contrib.auth import login, logout as auth_logout
|
||
from django.contrib.auth.decorators import login_required
|
||
from django.contrib.auth.forms import AuthenticationForm
|
||
from django.core.urlresolvers import reverse
|
||
from django.http import HttpResponse, HttpResponseRedirect
|
||
from django.shortcuts import render_to_response, render, get_object_or_404, redirect
|
||
from django.template import RequestContext
|
||
from django.forms.formsets import formset_factory
|
||
from django.contrib import messages
|
||
from django.db.models import Count, Sum, Max, Min
|
||
|
||
from models import *
|
||
from forms import *
|
||
|
||
import utils
|
||
|
||
def home(request):
|
||
if request.method == "POST":
|
||
form = AuthenticationForm(data=request.POST)
|
||
if form.is_valid():
|
||
# Log the user in.
|
||
login(request, form.get_user())
|
||
else:
|
||
if request.user.is_authenticated():
|
||
# Redirect to the register
|
||
return HttpResponseRedirect( reverse('main.views.register') )
|
||
else:
|
||
form = AuthenticationForm(request)
|
||
return render_to_response('home.html', locals(), context_instance=RequestContext(request))
|
||
|
||
def register(request):
|
||
products = Product.objects.filter(active=True)
|
||
cheapest_product_price = products.aggregate(Min('sale_price_int')).values()[0] or 0
|
||
half_a_year_ago = datetime.now() - timedelta(days=180)
|
||
users = User.objects.filter(userprofile__balance__gt=cheapest_product_price)
|
||
# sort users after user's last purchase time
|
||
users = sorted(users, key=lambda u: u.get_profile().last_purchase_date(), reverse=True)
|
||
users_js = users_format_js(users)
|
||
|
||
# multiple orderlines (one per product)
|
||
OrderLineFormSet = formset_factory(OrderLineForm)
|
||
if request.method == "POST":
|
||
orderform = OrderForm(request.POST)
|
||
formset = OrderLineFormSet(request.POST)
|
||
|
||
if formset.is_valid() and orderform.is_valid():
|
||
order_sum = sum([ol['amount'] * ol['unit_price'] for ol in formset.cleaned_data])
|
||
order = orderform.save(commit=False)
|
||
# can he afford it?
|
||
profile = order.customer.get_profile()
|
||
if order_sum > profile.balance:
|
||
messages.error(request, '{0} {1} har ikke råd, mangler {2} kr.'.format(
|
||
order.customer.first_name,
|
||
order.customer.last_name,
|
||
int(order_sum - profile.balance)))
|
||
return HttpResponseRedirect( reverse('main.views.register') )
|
||
# empty order?
|
||
if order_sum == 0:
|
||
messages.error(request, 'Du har ikke valgt hva du skal kjøpe...')
|
||
return HttpResponseRedirect( reverse('main.views.register') )
|
||
order.order_sum = order_sum
|
||
order.save()
|
||
# substract order from balance
|
||
profile.balance -= order.order_sum
|
||
profile.save()
|
||
|
||
orderlines = []
|
||
for ol in formset.cleaned_data:
|
||
if ol['amount'] > 0:
|
||
product = Product.objects.get(pk=ol['product'])
|
||
OrderLine.objects.create(
|
||
order=order,
|
||
product=product,
|
||
amount=ol['amount'],
|
||
unit_price=ol['unit_price'])
|
||
|
||
orderlines.append(str(ol['amount']) + " " + str(product))
|
||
|
||
messages.success(request, '{0} {1} kjøpte {2}.'.format(
|
||
order.customer.first_name,
|
||
order.customer.last_name,
|
||
", ".join(orderlines)
|
||
))
|
||
return HttpResponseRedirect( reverse('main.views.register') )
|
||
else:
|
||
# TODO specify error(s)
|
||
messages.error(request, 'Skjemaet er ikke gyldig.')
|
||
orderform = OrderForm(request.POST)
|
||
formset = OrderLineFormSet(request.POST)
|
||
else:
|
||
orderform = OrderForm()
|
||
formset = OrderLineFormSet()
|
||
return render_to_response('register.html', locals(), context_instance=RequestContext(request))
|
||
|
||
def deposit(request):
|
||
users = User.objects.all().order_by('first_name','last_name')
|
||
users_js = users_format_js(users)
|
||
allowed_users = utils.users_with_perm('add_transaction')
|
||
|
||
# FIXME: dont need this
|
||
try:
|
||
limit_deposists = settings.BBS_LIMIT_DEPOSITS
|
||
except e:
|
||
pass
|
||
|
||
if request.method == "POST":
|
||
form = DepositForm(request.POST)
|
||
if form.is_valid():
|
||
amount = form.cleaned_data['amount']
|
||
user = form.cleaned_data['user']
|
||
if amount + user.get_profile().balance > settings.BBS_SALDO_MAX:
|
||
messages.error(request, '{0} {1} kan ikke sette inn {2} kr, det overskrider maks saldo ({3} kr) med {4} kr'.format(
|
||
user.first_name,
|
||
user.last_name,
|
||
amount,
|
||
settings.BBS_SALDO_MAX,
|
||
amount + user.get_profile().balance - settings.BBS_SALDO_MAX
|
||
))
|
||
return HttpResponseRedirect( reverse('main.views.deposit') )
|
||
|
||
transaction = form.save()
|
||
profile = transaction.user.get_profile()
|
||
profile.balance += transaction.amount
|
||
profile.save()
|
||
|
||
messages.success(request, '{0} {1} satte inn {2} kr. Ny saldo er {3}'.format(
|
||
transaction.user.first_name,
|
||
transaction.user.last_name,
|
||
transaction.amount,
|
||
profile.balance))
|
||
return HttpResponseRedirect( reverse('main.views.deposit') )
|
||
else:
|
||
messages.error(request, 'Skjemaet er ikke gyldig.')
|
||
form = DepositForm(request.POST)
|
||
else:
|
||
form = DepositForm()
|
||
|
||
return render_to_response('deposit.html', locals(), context_instance=RequestContext(request))
|
||
|
||
def log(request, limit=datetime.now()-timedelta(days=2)):
|
||
if limit:
|
||
# display last two days of orders and transactions
|
||
orders = Order.objects.filter(created__gte=limit).order_by('-created')
|
||
transactions = Transaction.objects.filter(created__gte=limit).order_by('-created')
|
||
|
||
# no action in the last two days?
|
||
# ...then display the last 5 items
|
||
if len(orders) == 0:
|
||
orders = Order.objects.all().order_by('-created')[:5]
|
||
|
||
if len(transactions) == 0:
|
||
transactions = Transaction.objects.all().order_by('-created')[:5]
|
||
else:
|
||
# display everything
|
||
orders = Order.objects.all().order_by('-created')
|
||
transactions = Transaction.objects.all().order_by('-created')
|
||
|
||
return render_to_response('log.html', locals(), context_instance=RequestContext(request))
|
||
|
||
def stats(request):
|
||
products = Product.objects.filter(active=True)
|
||
|
||
return render_to_response('stats.html', locals(), context_instance=RequestContext(request))
|
||
|
||
def serialize_product(product):
|
||
f_product = {}
|
||
for attr in Product._meta.get_all_field_names():
|
||
if hasattr(product, attr) and product.__getattribute__(attr):
|
||
if type(product.__getattribute__(attr)) is datetime:
|
||
f_product[attr] = str(product.__getattribute__(attr))
|
||
else:
|
||
f_product[attr] = product.__getattribute__(attr)
|
||
return f_product
|
||
|
||
def products(request, product_id=None):
|
||
if product_id:
|
||
#single
|
||
products = get_object_or_404(Product, pk=product_id)
|
||
else:
|
||
#many
|
||
products = Product.objects.filter(active=True)
|
||
|
||
if product_id:
|
||
f_products = serialize_product(products)
|
||
else:
|
||
f_products = [serialize_product(product) for product in products]
|
||
|
||
return HttpResponse(json.dumps(f_products), content_type='application/javascript; charset=utf8')
|
||
|
||
def stats_product(product):
|
||
sales = OrderLine.objects.filter(product=product)
|
||
# reverse lookup via order, customer
|
||
per_user = sales.values('order__customer__first_name','order__customer__last_name').annotate(num=Sum('amount')).order_by('num').reverse()
|
||
f_per_user = []
|
||
for row in per_user:
|
||
who = row['order__customer__first_name'] + " " + row['order__customer__last_name']
|
||
num = row['num']
|
||
f_per_user.append([who, num])
|
||
response = {
|
||
'product' : serialize_product(Product.objects.get(pk=product)) if type(product) is unicode else serialize_product(product),
|
||
'counts' : f_per_user,
|
||
'total_counts' : sum([row['num'] for row in per_user]),
|
||
}
|
||
return response
|
||
|
||
def stats_products(request, product=None):
|
||
if product:
|
||
#single
|
||
f_products = [stats_product(product)]
|
||
sales = OrderLine.objects.filter(product=product)
|
||
else:
|
||
#all
|
||
sales = OrderLine.objects.filter(product__active=True)
|
||
f_products = [stats_product(p) for p in Product.objects.filter(active=True)]
|
||
|
||
totals = float(sum([sale.amount for sale in sales]))
|
||
response = {
|
||
'request' : 'all' if not product else product,
|
||
'response' : {
|
||
'products' : f_products,
|
||
'total_count' : totals,
|
||
}
|
||
}
|
||
return HttpResponse(json.dumps(response), content_type='application/javascript; charset=utf8')
|
||
|
||
def stats_orders(request):
|
||
total = float(Order.objects.count())
|
||
per_user = Order.objects.values('customer__first_name','customer__last_name').annotate(num=Count('customer')).order_by('num').reverse()
|
||
f_per_user = []
|
||
for row in per_user:
|
||
who = row['customer__first_name'] + " " + row['customer__last_name']
|
||
num = row['num']
|
||
f_per_user.append([who, num])
|
||
return HttpResponse(json.dumps(f_per_user), content_type='application/javascript; charset=utf8')
|
||
|
||
def stats_orders_hourly(request):
|
||
orders = Order.objects.all()
|
||
f_hourly = dict([(key,0) for key in range(0,24)]) # init
|
||
# group by hour
|
||
for key, values in groupby(orders, key=lambda row: row.created.hour):
|
||
count = len(list(values))
|
||
f_hourly[key] = f_hourly[key] + count
|
||
|
||
if len(orders) == 0:
|
||
return HttpResponse(json.dumps({}), content_type='application/javascript; charset=utf8')
|
||
response = {
|
||
'start': str(orders[0].created),
|
||
'hourly' : [[k,v] for k,v in f_hourly.iteritems()],
|
||
'total' : sum(f_hourly.values()),
|
||
}
|
||
return HttpResponse(json.dumps(response), content_type='application/javascript; charset=utf8')
|
||
|
||
def stats_products_realtime(request):
|
||
end_time = datetime.now()
|
||
start_time = datetime.now() - timedelta(hours=24)
|
||
|
||
order_lines = OrderLine.objects.filter(order__created__range=(start_time, end_time)).order_by("order__created")
|
||
products = {}
|
||
for order_line in order_lines:
|
||
order_time_in_milliseconds = int(time.mktime(order_line.order.created.timetuple()) * 1000)
|
||
if order_line.product not in products:
|
||
products[order_line.product] = [[order_time_in_milliseconds, order_line.amount]]
|
||
else:
|
||
prev_entry = products[order_line.product][-1]
|
||
products[order_line.product].append([order_time_in_milliseconds, prev_entry[1] + order_line.amount])
|
||
|
||
serialized_products = {}
|
||
for product, value in products.iteritems():
|
||
serialized_products[product.name] = value
|
||
|
||
response = {
|
||
'products': serialized_products
|
||
}
|
||
|
||
return HttpResponse(json.dumps(response), content_type='application/javascript; charset=utf8')
|
||
|
||
|
||
def logout(request):
|
||
auth_logout(request)
|
||
return render_to_response('registration/logout.html', locals(), context_instance=RequestContext(request))
|
||
|
||
def users_format_js(users):
|
||
users_js = []
|
||
for user in users:
|
||
users_js.append('"' + user.first_name + ' ' + user.last_name[:1] + '"')
|
||
users_js = u"[{0}]".format(u",".join(users_js))
|
||
return users_js
|
||
|
||
def create_user(request):
|
||
if request.method == "POST":
|
||
form = SimpleCreateUserForm(data=request.POST)
|
||
if form.is_valid():
|
||
user = form.save()
|
||
messages.success(request, "Hurra! {0} {1} er lagt til".format(user.first_name, user.last_name))
|
||
return redirect('create-user')
|
||
|
||
else:
|
||
messages.error(request, 'Feil med skjemaet, se under.')
|
||
form = SimpleCreateUserForm(data=request.POST)
|
||
else:
|
||
form = SimpleCreateUserForm()
|
||
|
||
return render(request, 'registration/create_user.html', locals())
|