Exemple #1
0
def create(request):
    args = {}
    us = User.objects.exclude(username='******')
    args['us'] = us

    if request.method == 'POST' and request.POST.get('user'):
        args['post'] = True
        gongzi = Gongzi()
        u = User.objects.get(id=request.POST.get('user'))
        gongzi.user = u
        gongzi.year = request.POST.get('year', datetime.today().year)
        gongzi.month = request.POST.get('month', datetime.today().month)
        gongzi.xiangmu = request.POST.get('xiangmu', '')
        gongzi.yingfa = request.POST.get('yingfa', 0)
        gongzi.baoxian = request.POST.get('baoxian', 0)
        gongzi.gongjijin = request.POST.get('gongjijin', 0)
        gongzi.shuijin = request.POST.get('shuijin', 0)
        gongzi.shifa = request.POST.get('shifa', 0)
        gongzi.memo = request.POST.get('memo', '')
        gongzi.guilei = request.POST.get('guilei', '')
        gongzi.realname = u.first_name
        gongzi.idcard = u.last_name
        gongzi.dateline = datetime.now()
        gongzi.save()
        return render(request, 'gongzi/admin/create.html', args)
    return render(request, 'gongzi/admin/create.html', args)
Exemple #2
0
def upload(request):
    err = []
    this_month = datetime.now().month

    if request.method == 'POST':
        month = request.POST.get('month', None)
        year = request.POST.get('year', None)
        f = request.FILES.get('file', None)
        import os
        if month and year and f:
            des_path = os.path.abspath('.') + '/gongzi/uploads/upload.xls'
            des_f = open(des_path, "wb")
            for chunk in f.chunks():
                des_f.write(chunk)
            des_f.close()

            bk = xlrd.open_workbook(des_path)
            sheet = bk.sheet_by_index(0)
            rows_num = sheet.nrows
            #cols_num = sheet.ncols

            #title = []
            #rows = []
            gs = []
            for j in range(1, rows_num):
                try:
                    g = Gongzi()
                    g.user = User.objects.get(last_name=sheet.cell_value(j, 3))
                    g.realname = sheet.cell_value(j, 0)
                    g.idcard = sheet.cell_value(j, 2)
                    g.yingfa = smart_float(sheet.cell_value(j, 4))
                    g.baoxian = smart_float(sheet.cell_value(j, 5))
                    g.gongjijin = smart_float(sheet.cell_value(j, 6))
                    g.shuijin = smart_float(sheet.cell_value(j, 7))
                    g.shifa = smart_float(sheet.cell_value(j, 8))
                    g.xiangmu = sheet.cell_value(j, 2)
                    g.memo = sheet.cell_value(j, 9)
                    g.guilei = sheet.cell_value(j, 10)
                    g.month = month
                    g.year = year
                    g.dateline = datetime.today()
                    gs.append(g)
                except User.DoesNotExist:
                    err.append(u'第%s行中身份证号在数据库中找不到对应人员,错误数据%s' % (j + 1, sheet.cell_value(j, 2)))

            if not err:
                Gongzi.objects.bulk_create(gs)
        else:
            err.append(u'表单每项都不能为空')
        return render(request, 'gongzi/admin/upload.html', {'month': this_month, 'err': err, 'post': True})
    return render(request, 'gongzi/admin/upload.html', {'month': this_month, 'err': err})