# -*- coding: utf-8 -*-

from django.shortcuts import render
from django.http import HttpResponse, HttpResponseRedirect
from django.template import RequestContext, Context, loader
from asuzr.models import Product
from asuzr.models import Attendance
from asuzr.models import Order
from asuzr.models import OrderPlan
from asuzr.models import Schedule
from asuzr.models import ProdPlan
from datetime import datetime, date, timedelta
import calendar
from django.db.models import Count, Sum
from asuzr.common import custom_date
from django.contrib.auth.decorators import login_required
from tables import *
from forms import *
from django_tables2 import RequestConfig

@login_required 
def prod_list(request):
  product_list = Product.objects.all()
  t = loader.get_template('asuzr/prod_list.html')
  c = Context({
    'product_list': product_list,
    })
  return HttpResponse(t.render(c))

@login_required 
def prod_detail(request, prod_id):
  return HttpResponse("This is %s" % prod_id)

def get_filtered_list(p_list, year, month):
  filtered_list=[]
  for a in p_list:
    a_date = a.date
    if a_date.strftime('%m/%Y').lstrip('0') == '/'.join((month,year)).lstrip('0'):
      filtered_list.append(a)
      
  return filtered_list

def get_orders_by_date(dt):
  order_list = Order.objects.filter(date=dt).order_by('id')
  return order_list

@login_required
def visit_view(request):
  curr_date = datetime.strptime(request.GET.get('date', date.today().strftime('%d.%m.%Y')), '%d.%m.%Y')
  form = DateForm(request.GET, initial = {'date': curr_date})
  
  y,m = curr_date.year, curr_date.month
  day_in_month = calendar.monthrange(y,m)[1]
  month_days = {i+1: {'date': custom_date(y,m,i+1)} for i in range(day_in_month)}
  sdate = date(y,m,1)
  edate = date(y,m,day_in_month)

  attend_list = Attendance.objects.filter(date__range = (sdate,edate))
  attend_sum = attend_list.aggregate(Sum('calls'), Sum('visits'))
  for attend in attend_list:
    month_days[attend.date.day]['attend'] = attend

  order_list = Order.objects.filter(date__range = (sdate,edate))
  order_sum = order_list.aggregate(Count('product'), Sum('price'))
  order_list = order_list.values('date')
  order_list = order_list.annotate(Count('product'), Sum('price'))

  for order in order_list:
    month_days[order['date'].day]['order'] = order

  schedule = Schedule.objects.filter(date__range = (sdate,edate))
  
  for designer in schedule:
    day = designer.date.day
    if 'designer' in month_days[day]:
      month_days[day]['designer'] = '%s, %s' % (month_days[day]['designer'], designer)
    else:
      month_days[day]['designer'] = designer
      
  table = VisitTable(month_days.values())
  RequestConfig(request, paginate={'per_page': 32}).configure(table)
  table.set_summaries({
                        'calls': attend_sum['calls__sum'],
                        'visits': attend_sum['visits__sum'],
                        'orders': order_sum['product__count'],
                        'cost': order_sum['price__sum'],
                      })
  title = 'Таблица посещаемости на %s г.' % curr_date.strftime('%B %Y')
  
  if request.method == 'POST':
    form = DateForm(request.POST)
    if form.is_valid():
      print form.cleaned_data
  else:
    form = DateForm()
    
  return render(request, 'asuzr/table.html', {'table': table, 'title': title, 'form': form})

@login_required 
def main(request, day, month, year):
  if day == None:
    day = str(date.today().day)
  if month == None:
    month = str(date.today().month)
  if year == None:
    year = str(date.today().year)
    
  d,m,y=int(day),int(month), int(year)
  
  attend_list = Attendance.objects.all().order_by('date')
  filtered_attend_list=get_filtered_list(attend_list, year, month)
  
  p_date = datetime.strptime(day+'/'+month+'/'+year, '%d/%m/%Y')
  order_list = Order.objects.filter(date=p_date).order_by('id')
  month_order_list=Order.objects.filter(date__range=(date(y,m,1),date(y,m,calendar.monthrange(y,m)[1]))).values('date').annotate(Count('product'),Sum('price'))
  
  plan = OrderPlan.objects.all()
  filtered_plan = get_filtered_list(plan, year, month)
  month_plan=0
  if len(filtered_plan) > 0:
    month_plan=filtered_plan[0].plan
    
  schedule = Schedule.objects.all().order_by('date')
  filtered_schedule = get_filtered_list(schedule, year, month)
  
  month_days={i: {'date': custom_date(int(year),int(month),i)}  for i in range(1,calendar.monthrange(int(year),int(month))[1]+1)}
  
  for l in filtered_attend_list:
    month_days[l.date.day]['attend']=l
    
  for s in filtered_schedule:
    if 'designers' in month_days[s.date.day]:
      des=', '.join((month_days[s.date.day]['designers'], ' '.join((s.designer.first_name, s.designer.last_name))))
      month_days[s.date.day]['designers'] = des
    else:
      month_days[s.date.day]['designers'] = ' '.join((s.designer.first_name, s.designer.last_name))
      
  for order in month_order_list:
    month_days[order['date'].day]['orders_count'] = order['product__count']
    month_days[order['date'].day]['orders_price'] = order['price__sum']

  month_days_values = month_days.values()
  
  sum_calls = sum(l.calls for l in filtered_attend_list)
  sum_visits = sum(l.visits for l in filtered_attend_list)
  sum_orders = sum(l['orders_count'] for l in month_days_values if 'orders_count' in l)
  sum_price = sum(l['orders_price'] for l in month_days_values if 'orders_price' in l)
  
  sum_order_price = sum(l.price for l in order_list)
  plan_balance = month_plan-sum_price
  
  d_date = p_date.strftime("%d/%m/%Y")
    
  t = loader.get_template('asuzr/attend_order.html')
  c = RequestContext(request,{
    'attend_list': month_days_values,
    'order_list': order_list,
    'sum_calls': sum_calls,
    'sum_visits': sum_visits,
    'sum_orders': sum_orders,
    'sum_price': sum_price,
    'sum_order_price': sum_order_price,
    'plan': month_plan,
    'balance': plan_balance,
    'd_date': d_date,
    })
  return HttpResponse(t.render(c))

@login_required
def sketches(request, order_id):
  curr_order = Order.objects.get(pk = order_id)
  table = SketchesTable(Sketch.objects.filter(order = curr_order))
  RequestConfig(request).configure(table)
  return render(request, 'asuzr/table.html', {'table': table, 'title': 'Эскизы заказа %s' % curr_order})

@login_required 
def orders(request, archive):
  is_archive = (archive == '1')
  Table = ArchiveOrdersTable if is_archive else OrdersTable
  table = Table(Order.objects.filter(is_done = is_archive))
  title = 'Архивная таблица заказов' if is_archive else 'Таблица выхода заказов'
  RequestConfig(request).configure(table)
  return render(request, 'asuzr/table.html', {'table': table, 'title': title})

@login_required 
def desreport(request):
  start_date = request.GET.get('sdate', date.today().strftime('%d.%m.%y'))
  sdate = datetime.strptime(start_date, '%d.%m.%y')
  end_date = request.GET.get('edate', date.today().strftime('%d.%m.%y'))
  edate = datetime.strptime(end_date, '%d.%m.%y')
  Table = DesignerTable
  table = Table(Order.objects.filter(cancelled=False, date__range=(sdate,edate)).values('designer__first_name','designer__last_name').annotate(Sum('price'),Count('designer')))
  title = u'Отчет по дизайнерам за '+' - '.join((start_date, end_date))
  RequestConfig(request).configure(table)
  return render(request, 'asuzr/table.html', {'table': table, 'title': title})

@login_required
def production_table(request, order_id):
  order_list = Order.objects.filter(is_done=False).order_by('-id')
  sel_order = Order.objects.filter(id=order_id)
  cost_items = sel_order.values('cost_items')
  t=loader.get_template('asuzr/order_costs.html')
  c=RequestContext(request,{
    'order_list' : order_list,
    'sel_order' : sel_order,
    'cost_items' : cost_items,
    })
  return HttpResponse(t.render(c))

@login_required
def prod_plan_view(request):
  curr_date = datetime.strptime(request.GET.get('date', date.today().strftime('%d.%m.%Y')), '%d.%m.%Y')
  y,m = curr_date.year, curr_date.month
  wd = curr_date.weekday()
  sdate = curr_date - timedelta(days = wd)
  edate = curr_date + timedelta(days = 6-wd)
  
  week_days = {i: {'date': custom_date(y,m,sdate.day+i)} for i in range(0,7)}
  
  prodplan_list = ProdPlan.objects.filter(start_date__range = (sdate,edate))
  
  for prodplan in prodplan_list:
    week_days[prodplan.start_date.weekday()]['prodplan'] = prodplan
  
  table = ProdPlanTable(week_days.values())
  title = u'Производственный план на %s - %s' % (sdate.strftime('%d.%m.%Y'), edate.strftime('%d.%m.%Y'))
  RequestConfig(request).configure(table)
  return render(request, 'asuzr/table.html', {'table': table, 'title': title})

def get_date(request):
  if request.method == 'POST':
    form = DateForm(request.POST)
    if form.is_valid():
      HttpResponse("AAA")
  else:
    form = DateForm()
  return render(request, 'asuzr/date_control.html', {'form': form})