随机分配岗位的一个小程序

# -*- coding: utf-8 -*- from django.http import HttpResponse from django.shortcuts import render, redirect from django.utils.translation import ugettext as _ from django.contrib.auth.models import User from models import Job, UserJob, UserProfile import json import random from django.contrib import auth import datetime # Create your views here. # 登录 def login(request): if request.method == 'GET': return render(request, 'login.html') else: username = request.POST['username'] password = request.POST['password'] # user = User.objects.filter(username=username).first() user = auth.authenticate(username=username, password=password) if user is not None and user.is_active: auth.login(request, user) else: error_msg = {"error_msg": "用户名或密码错误"} return render(request,'login.html', error_msg) # return redirect('/test/master/login') # 保存用户档案 if not user.user_profile.all(): user_profile = UserProfile() user_profile.name = user.username else: user_profile = user.user_profile.all()[0] user_profile.user = user user_profile.save() # 将用户档案保存到会话 # request.session["user_profile"] = user_profile request.session["user_profile_name"] = user_profile.name # return render(request, 'test2/arrange_job.html') return redirect('/master/arrange') def logout(request): auth.logout(request) # Redirect to a success page. return redirect("/master/login/") # 安排岗位 def arrange(request): # 获取当天当前安排的人员岗位 result_dict = _get_user_job_current_day() # profile = request.user.user_profile.all()[0] # current_user = {} # current_user["user_profile_name"] = profile.name return render(request, 'master/arrange_job.html', result_dict) # 获取所有用户 def get_users(request): try: users = User.objects.all() result_dict = {} user_list = [] for user in users: user_dict = {} # user_list.append(user.username) if 'root' == user.username or 'admin' == user.username: continue user_dict["id"] = user.id # 保存用户档案 if not user.user_profile.all(): user_profile = UserProfile() user_profile.name = user.name else: user_profile = user.user_profile.all()[0] user_profile.user = user user_profile.save() profile = user.user_profile.all()[0] user_dict["name"] = profile.name user_list.append(user_dict) result_dict["user_list"] = user_list result_dict["status"] = 0 except Exception as e: result_dict["status"] = -1 print e.message return HttpResponse(json.dumps(result_dict)) # 获取所有岗位 def get_jobs(request): try: jobs = Job.objects.all() result_dict = {} job_list = [] for job in jobs: job_dict = {} job_dict["id"] = job.id job_dict["jobname"] = job.job_name job_list.append(job_dict) result_dict["status"] = 0 result_dict["job_list"] = job_list except Exception as e: result_dict["status"] = -1 print e.message return HttpResponse(json.dumps(result_dict)) # 随机分配岗位 def random_arrange_job(request): # 避免F5刷新 # _aviodF5Refresh(request) if 'submitTime' in request.session: timestamp_session = request.session['submitTime'] timestamp_request = request.POST['avoidF5Submit'] if (timestamp_request == timestamp_session): # error_msg_dict = {"error_msg": _(u'请勿按F5刷新！')} result_dict = _get_user_job_current_day() result_dict["error_msg"] =u'请勿按F5刷新！' return render(request, 'master/arrange_job.html', result_dict) else: request.session['submitTime'] = request.POST['avoidF5Submit'] else: request.session['submitTime'] = request.POST['avoidF5Submit'] # 勾选的人的数量和编辑框中选的工作岗位的数量必须是相等的 # 将勾选的所有人放到一个list里面 body_content = request.POST # print body_content user_id_list = [] job_id_list = [] job_count = 0 for key in body_content.keys(): if key.startswith("user"): user_id_list.append(key.split('_')[1]) elif key.startswith("job"): job_id_list.append(key.split('_')[1]) if '' != body_content.get(key): job_count += int(body_content.get(key)) # job_count += int(key.split('_')[1]) # job_id_count_list = [] # print user_id_list # print job_id_list user_len = len(user_id_list) # 保存勾选用户的id的list并使这个list里边的id顺序是随机的 random_user_id_list = [] random_user_id_list_repeat = [] random_user_id_list = _get_random_user_id_list(user_id_list, job_count, user_len) print random_user_id_list # while (len(random_user_id_list) != user_len): # random_num = random.choice(user_id_list) # if random_num not in random_user_id_list: # random_user_id_list.append(random_num) # if job_count > user_len: # repeat_count = job_count - user_len # while (len(random_user_id_list_repeat) != repeat_count): # random_num = random.choice(user_id_list) # if random_num not in random_user_id_list_repeat: # random_user_id_list_repeat.append(random_num) # random_user_id_list.extend(random_user_id_list_repeat) _delete_current_day_user_job() # 删除当天重复安排的工作 即，一天只能安排一次 后一次的安排会把前面安排的覆盖掉 # user_job_exists_list = UserJob.objects.filter(arrange_time__contains=datetime.datetime.now().strftime("%Y-%m-%d")) # for user_job_item in user_job_exists_list: # user_job_item.delete() # 保存安排的用户岗位到UserJob表 _save_user_job(body_content, random_user_id_list) # 获取当天当前安排的人员岗位 result_dict = _get_user_job_current_day() return render(request, 'master/arrange_job.html', context=result_dict) # return redirect('/test/arrange') # 获取随机存放的user_id列表 def _get_random_user_id_list(user_id_list, job_count, user_len): if job_count > user_len*2: raise Exception random_user_id_list_repeat = [] random_user_id_list = _get_user_ids(user_id_list, user_len) if job_count > user_len: repeat_count = job_count - user_len random_user_id_list_repeat = _get_user_ids(user_id_list, repeat_count) # while (len(random_user_id_list) != user_len): # random_num = random.choice(user_id_list) # if random_num not in random_user_id_list: # random_user_id_list.append(random_num) # if job_count > user_len: # repeat_count = job_count - user_len # while (len(random_user_id_list_repeat) != repeat_count): # random_num = random.choice(user_id_list) # if random_num not in random_user_id_list_repeat: # random_user_id_list_repeat.append(random_num) random_user_id_list.extend(random_user_id_list_repeat) return random_user_id_list def _get_user_ids(user_id_list, _len): random_user_id_list = [] while (len(random_user_id_list) != _len): random_num = random.choice(user_id_list) if random_num not in random_user_id_list: random_user_id_list.append(random_num) return random_user_id_list # 删除当天重复安排的岗位 def _delete_current_day_user_job(): user_job_exists_list = UserJob.objects.filter(arrange_time__contains=dateti