jian

# -*- coding: utf-8 -*-
from __future__ import unicode_literals
from django.http import JsonResponse
import json
from django.shortcuts import render
from backend.models import Employee, Person
import pandas as pd
# Create your views here.
import numpy as np
import random
import string

from datetime import datetime
def index(request):
    return render(request, 'home.html')

def _map(data):
    result = []

    for index, row in data.iterrows():   # 获取每行的index、row
        tmp_row = []
        for col_name in data.columns:
            tmp_row.append(row[col_name])
        result.append(tmp_row)
        print(tmp_row)
    return result



def remove_src(li, di):
    result = li[:]
    for key in di:
        num = result.count(key)
        for i in range(num):
                # print(key, result)
            result.remove(key)
    return result

def getPerson(request):
    person_list = Person.objects.all()
    src_list = []
    for person in person_list:
        src_list.append(person.name)
    num = int(request.GET['num'])

    b = random.sample(src_list, num)
    print(type(b))
    r = list(set(b))
    tmp_list = src_list[:]
    while len(r) < num:
        tmp_list = remove_src(tmp_list, r)
        print(r, tmp_list)
        tmp_r = random.sample(tmp_list, num - len(r))
        r = list(set(r + tmp_r))
        print(r, tmp_list)
    return JsonResponse(r, safe=False)

def showNumLast(request):
    person_list = Person.objects.all()
    src_list = []
    for person in person_list:
        src_list.append(person.name)
    num = int(request.GET['num'])

    b = random.sample(src_list, num)
    print(type(b))
    r = list(set(b))
    tmp_list = src_list[:]
    while len(r) < num:
        tmp_list = remove_src(tmp_list, r)
        print(r, tmp_list)
        tmp_r = random.sample(tmp_list, num - len(r))
        r = list(set(r + tmp_r))
        print(r, tmp_list)

    for tmp in r:
        tmp_emp = Employee(name=tmp)
        tmp_emp.is_select=0
        tmp_emp.save()
    # setPesonBack()
    return JsonResponse(r, safe=False)

def reset(request):
    name_dict = {'code': 'success'}
    all_emp = Employee.objects.all()
    for tmp in all_emp:
        tmp.is_select = 1
        tmp.save()
    setPesonBack()
    return JsonResponse(name_dict)


def setPesonBack():
    Person.objects.all().delete()
    emp_list = Employee.objects.filter(is_enabled=1, is_select=1)

    for emp in emp_list:
        for i in range(emp.weight):
            tmp = Person(name=emp.name)
            tmp.save()

def setPeson(request):
    Person.objects.all().delete()
    emp_list = Employee.objects.filter(is_enabled=1, is_select=1)

    for emp in emp_list:
        for i in range(emp.weight):
            tmp = Person(name=emp.name)
            tmp.save()
    return JsonResponse({'twz': 'Love python and Django', 'zqxt': 'I am teaching Django'})

def onload(request):
    file = request.FILES.get('excel')
    df = pd.read_excel(request.FILES.get('excel'))
    print(df.items())
    print(df.columns)
    temp = _map(df)
    print(temp)

    for row in temp:
        tmp_emp = Employee(name=row[0],weight=row[1],is_trainee=row[2])
        tmp_emp.save()
    return render(request, 'excel.html')

def getEmployee(request):
    num = request.GET['num']
    print(type(num))
    # emp_list = Employee.objects.filter(is_enabled=1)
    # for emp in emp_list:
    #     emp.name
    return JsonResponse({"name": 1})



def excel(request):
    return render(request, 'excel.html')

def get_user(request):
    name_dict = {'twz': 'Love python and Django', 'zqxt': 'I am teaching Django'}
    emp = Employee(name="吴键鸿")
    emp.save()
    return JsonResponse(name_dict)