211 lines
9.6 KiB
Python
211 lines
9.6 KiB
Python
from django.http import HttpResponse,JsonResponse
|
|
from django.shortcuts import render
|
|
from django.shortcuts import redirect
|
|
from .models import Group,GroupUser
|
|
from .forms import GroupUserForm, UserForm
|
|
from safesite import models as s_models
|
|
from safesite.tasks import gettime
|
|
from django.forms.models import model_to_dict
|
|
from django.http import Http404
|
|
import requests
|
|
import json
|
|
import time
|
|
from datetime import datetime,date,timedelta
|
|
from django.core.serializers.json import DjangoJSONEncoder
|
|
from django.db.models import Q,F,Count,Sum
|
|
|
|
def check_login(func): # 自定义登录验证装饰器
|
|
def warpper(request,*args,**kwargs):
|
|
is_login = request.session.get('is_login', False)
|
|
if is_login:
|
|
return func(request,*args,**kwargs)
|
|
else:
|
|
return redirect('/groups/login')
|
|
return warpper
|
|
|
|
def login(request):
|
|
# 不允许重复登录
|
|
if request.session.get('is_login', None):
|
|
return redirect('/groups/index/')
|
|
if request.method == "POST":
|
|
groups_form = GroupUserForm(request.POST)
|
|
message = '请检查填写的内容！'
|
|
if groups_form.is_valid():
|
|
username = groups_form.cleaned_data.get('username')
|
|
password = groups_form.cleaned_data.get('password')
|
|
|
|
try:
|
|
user = GroupUser.objects.get(username=username)
|
|
except:
|
|
message = '用户不存在！'
|
|
return render(request, 'groups/login.html', locals())
|
|
if user.password == password:
|
|
request.session['is_login'] = True
|
|
request.session['user_id'] = user.id
|
|
request.session['user_name'] = user.username
|
|
request.session['group_id'] = user.group
|
|
return redirect('/groups/index/')
|
|
else:
|
|
return render(request, 'groups/login.html', locals())
|
|
else:
|
|
return render(request, 'groups/login.html', locals())
|
|
groups_form = GroupUserForm()
|
|
return render(request, 'groups/login.html', locals())
|
|
|
|
def logout(request):
|
|
# if not request.session.get('is_login', None):
|
|
# return redirect('/groups/login')
|
|
request.session.flush()
|
|
return redirect('/groups/login')
|
|
|
|
import json
|
|
from captcha.models import CaptchaStore
|
|
from captcha.helpers import captcha_image_url
|
|
def refresh_captcha(request):
|
|
hashkey = CaptchaStore.generate_key()
|
|
image_url = captcha_image_url(hashkey)
|
|
return HttpResponse(json.dumps({'key': hashkey, 'image_url': image_url}), content_type='application/json')
|
|
|
|
@check_login
|
|
def company(request):
|
|
companys = GroupUser.objects.get(id = request.session['user_id']).group.members.all()
|
|
for i in companys:
|
|
i.useradmin = s_models.User.objects.get(ubelongpart=i,issuper=1).username
|
|
return render(request, 'groups/company.html', locals())
|
|
|
|
def job(request):
|
|
return render(request, 'groups/job.html', locals())
|
|
|
|
def company_user(request, cid):
|
|
users = s_models.User.objects.filter(usecomp=cid)
|
|
# for user in users:
|
|
# dump(user)
|
|
return render(request, 'groups/company_user.html', locals())
|
|
|
|
def company_userdetail(request, uid):
|
|
try:
|
|
user = s_models.User.objects.get(pk=uid)
|
|
except s_models.User.DoesNotExist:
|
|
raise Http404("User does not exist")
|
|
user_form = UserForm(model_to_dict(user))
|
|
# print(type(user))
|
|
# print(user_form)
|
|
# dump(user)
|
|
return render(request, 'groups/company_userdetail.html', locals())
|
|
|
|
def groups_userupdate(request):
|
|
if request.method == 'POST':
|
|
user_form = UserForm(request.POST)
|
|
if user_form.is_valid():
|
|
model_user = s_models.User()
|
|
model_user.userid = user_form.cleaned_data.get('userid')
|
|
model_user.username = user_form.cleaned_data.get('username')
|
|
model_user.name = user_form.cleaned_data.get('name')
|
|
model_user.empid = user_form.cleaned_data.get('empid')
|
|
model_user.ubelongpart = user_form.cleaned_data.get('ubelongpart')
|
|
model_user.openid = user_form.cleaned_data.get('openid')
|
|
model_user.nickname = user_form.cleaned_data.get('nickname')
|
|
model_user.headimgurl = user_form.cleaned_data.get('headimgurl')
|
|
model_user.deletemark = user_form.cleaned_data.get('deletemark')
|
|
model_user.mpopenid = user_form.cleaned_data.get('mpopenid')
|
|
model_user.usecomp = user_form.cleaned_data.get('usecomp')
|
|
model_user.save()
|
|
return render(request, 'groups/company_userdetail.html', locals())
|
|
else:
|
|
return redirect('/groups/company/')
|
|
|
|
def dump(obj):
|
|
print('\n'.join(['%s:%s' % item for item in obj.__dict__.items()]))
|
|
|
|
@check_login
|
|
def index(req):
|
|
first_day,first_day_of_next_month=gettime()
|
|
group = GroupUser.objects.get(id = req.session['user_id']).group
|
|
companys = group.members.all()
|
|
troublenum = s_models.Trouble.objects.filter(deletemark=1,usecomp__in=companys,fxsj__range=(first_day, first_day_of_next_month)).count()
|
|
observenum = s_models.Observe.objects.filter(deletemark=1,usecomp__in=companys,looktime__range=(first_day, first_day_of_next_month)).count()
|
|
missnum = s_models.Miss.objects.filter(deletemark=1,usecomp__in=companys,misstime__range=(first_day, first_day_of_next_month)).count()
|
|
trainnum = s_models.Train.objects.filter(deletemark=1,usecomp__in=companys,starttime__range=(first_day, first_day_of_next_month)).count()
|
|
companynum = len(companys)
|
|
usernum = s_models.User.objects.filter(deletemark=1,usecomp__in=companys).count()
|
|
return render(req, 'groups/index.html', locals())
|
|
|
|
|
|
def updateposition(req):
|
|
allcompanys = s_models.Partment.objects.filter(deletemark=1,iscompany=1)
|
|
for i in allcompanys:
|
|
address = i.partname
|
|
data = getPosition(address)
|
|
if data['status'] == '1' and int(data['count'])>0:
|
|
i.position = data['geocodes'][0]
|
|
i.save()
|
|
else:
|
|
address = i.partname.replace('公司','').replace('责任','').replace('有限','')
|
|
data = getPosition(address)
|
|
if data['status'] == '1' and int(data['count'])>0:
|
|
i.position = data['geocodes'][0]
|
|
i.save()
|
|
return JsonResponse({"code":1})
|
|
|
|
def getPosition(address):
|
|
url = 'http://restapi.amap.com/v3/geocode/geo'
|
|
par = {'address': address, 'key': '86e0993305a427ac2d0b5371ad97f242'}
|
|
res = requests.get(url, par)
|
|
json_data = json.loads(res.text)
|
|
return json_data
|
|
|
|
def apipositions(req):
|
|
companys = GroupUser.objects.get(id = req.session['user_id']).group.members.all()
|
|
positions = companys.values('position','partname')
|
|
return JsonResponse({"code":1,"data":list(positions)})
|
|
def trouble(request):#隐患排查
|
|
return render(request, 'groups/troublepic.html', locals())
|
|
def observe(request):#行为观察
|
|
return render(request, 'groups/observe.html', locals())
|
|
def troubleChart(request):#隐患统计图
|
|
code = request.GET.get('code')
|
|
year = int(request.GET.get('year'))
|
|
month =int( request.GET.get('month'))
|
|
if code=='tbpic':
|
|
yhcompanys=[]
|
|
yhcompanyscount=[]
|
|
first_day,first_day_of_next_month = gettime(datetime(year,month,1))
|
|
companys = GroupUser.objects.get(id = request.session['user_id']).group.members.all()
|
|
troublelist = s_models.Trouble.objects.filter(deletemark=1,usecomp__in=companys,fxsj__range=(first_day, first_day_of_next_month)).values('usecomp__partid','usecomp__partname').annotate(number = Count('troubleid'))
|
|
return HttpResponse(json.dumps({'code':1,'data':{'troublelist':list(troublelist)}}), content_type='application/json')
|
|
def observeChart(request):#行为观察
|
|
code = request.GET.get('code')
|
|
year = int(request.GET.get('year'))
|
|
month =int( request.GET.get('month'))
|
|
if code=='tbpic':
|
|
userfy=[]#人员反映
|
|
userwz=[]#员工的位置
|
|
grfh=[]#个人防护装备
|
|
gjhsb=[]#工具和设备
|
|
cxhbz=[]#程序和标准
|
|
rtgcx=[]#人体工程学
|
|
hjzj=[]#环境整洁
|
|
first_day,first_day_of_next_month = gettime(datetime(year,month,1))
|
|
companys = GroupUser.objects.get(id = request.session['user_id']).group.members.all()
|
|
Observe = s_models.Observe.objects.filter(deletemark=1,usecomp__in=companys,looktime__range=(first_day, first_day_of_next_month)).values('usecomp__partid','usecomp__partname').annotate(number = Count('lookid'))
|
|
for j in Observe:
|
|
|
|
for i in s_models.Dicclass.objects.filter(parentid__dicid=23):
|
|
if i.dicid==24:
|
|
userfy.append(s_models.Unsafes.objects.filter(observe__in=j,unsafedicclass=i).count())
|
|
elif i.dicid==25:
|
|
userwz.append(s_models.Unsafes.objects.filter(observe__in=j,unsafedicclass=i).count())
|
|
elif i.dicid==26:
|
|
grfh.append(s_models.Unsafes.objects.filter(observe__in=j,unsafedicclass=i).count())
|
|
elif i.dicid==27:
|
|
gjhsb.append(s_models.Unsafes.objects.filter(observe__in=j,unsafedicclass=i).count())
|
|
elif i.dicid==28:
|
|
cxhbz.append(s_models.Unsafes.objects.filter(observe__in=j,unsafedicclass=i).count())
|
|
elif i.dicid==29:
|
|
rtgcx.append(s_models.Unsafes.objects.filter(observe__in=j,unsafedicclass=i).count())
|
|
elif i.dicid==30:
|
|
hjzj.append(s_models.Unsafes.objects.filter(observe__in=j,unsafedicclass=i).count())
|
|
|
|
return HttpResponse(json.dumps({'code':1,'data':{'list':list(Observe)},'userfy':userfy,'userwz':userwz,'grfh':grfh,'gjhsb':gjhsb,'cxhbz':cxhbz,'rtgcx':rtgcx,'hjzj':hjzj,}), content_type='application/json')
|
|
|