349 lines
12 KiB
Python
349 lines
12 KiB
Python
# coding: utf-8
|
||
|
||
from __future__ import division
|
||
import uuid
|
||
import urllib
|
||
|
||
from django.db.models import Count
|
||
from django.shortcuts import render_to_response
|
||
from django.template import RequestContext
|
||
from django.http import HttpResponseNotFound
|
||
from django.http import HttpResponse
|
||
# from jperm.models import Apply
|
||
import paramiko
|
||
from jumpserver.api import *
|
||
|
||
|
||
|
||
def getDaysByNum(num):
|
||
today = datetime.date.today()
|
||
oneday = datetime.timedelta(days=1)
|
||
li_date, li_str = [], []
|
||
for i in range(0, num):
|
||
today = today-oneday
|
||
li_date.append(today)
|
||
li_str.append(str(today)[5:10])
|
||
li_date.reverse()
|
||
li_str.reverse()
|
||
t = (li_date, li_str)
|
||
return t
|
||
|
||
|
||
def get_data(data, items, option):
|
||
dic = {}
|
||
li_date, li_str = getDaysByNum(7)
|
||
for item in items:
|
||
li = []
|
||
name = item[option]
|
||
if option == 'user':
|
||
option_data = data.filter(user=name)
|
||
elif option == 'host':
|
||
option_data = data.filter(host=name)
|
||
for t in li_date:
|
||
year, month, day = t.year, t.month, t.day
|
||
times = option_data.filter(start_time__year=year, start_time__month=month, start_time__day=day).count()
|
||
li.append(times)
|
||
dic[name] = li
|
||
return dic
|
||
|
||
|
||
@require_role(role='user')
|
||
def index_cu(request):
|
||
user_id = request.session.get('user_id')
|
||
user = get_object(User, id=user_id)
|
||
login_types = {'L': 'LDAP', 'M': 'MAP'}
|
||
username = user.username
|
||
posts = user.get_asset()
|
||
host_count = len(posts)
|
||
new_posts = []
|
||
post_five = []
|
||
for post in posts:
|
||
if len(post_five) < 5:
|
||
post_five.append(post)
|
||
else:
|
||
new_posts.append(post_five)
|
||
post_five = []
|
||
new_posts.append(post_five)
|
||
|
||
return render_to_response('index_cu.html', locals(), context_instance=RequestContext(request))
|
||
|
||
|
||
@require_role(role='user')
|
||
def index(request):
|
||
li_date, li_str = getDaysByNum(7)
|
||
today = datetime.datetime.now().day
|
||
from_week = datetime.datetime.now() - datetime.timedelta(days=7)
|
||
|
||
if is_role_request(request, 'user'):
|
||
return index_cu(request)
|
||
|
||
elif is_role_request(request, 'super'):
|
||
users = User.objects.all()
|
||
hosts = Asset.objects.all()
|
||
online = Log.objects.filter(is_finished=0)
|
||
online_host = online.values('host').distinct()
|
||
online_user = online.values('user').distinct()
|
||
active_users = User.objects.filter(is_active=1)
|
||
active_hosts = Asset.objects.filter(is_active=1)
|
||
week_data = Log.objects.filter(start_time__range=[from_week, datetime.datetime.now()])
|
||
|
||
elif is_role_request(request, 'admin'):
|
||
return index_cu(request)
|
||
# user = get_session_user_info(request)[2]
|
||
# users = User.objects.filter(dept=dept)
|
||
# hosts = Asset.objects.filter(dept=dept)
|
||
# online = Log.objects.filter(dept_name=dept_name, is_finished=0)
|
||
# online_host = online.values('host').distinct()
|
||
# online_user = online.values('user').distinct()
|
||
# active_users = users.filter(is_active=1)
|
||
# active_hosts = hosts.filter(is_active=1)
|
||
# week_data = Log.objects.filter(dept_name=dept_name, start_time__range=[from_week, datetime.datetime.now()])
|
||
|
||
# percent of dashboard
|
||
if users.count() == 0:
|
||
percent_user, percent_online_user = '0%', '0%'
|
||
else:
|
||
percent_user = format(active_users.count() / users.count(), '.0%')
|
||
percent_online_user = format(online_user.count() / users.count(), '.0%')
|
||
if hosts.count() == 0:
|
||
percent_host, percent_online_host = '0%', '0%'
|
||
else:
|
||
percent_host = format(active_hosts.count() / hosts.count(), '.0%')
|
||
percent_online_host = format(online_host.count() / hosts.count(), '.0%')
|
||
|
||
user_top_ten = week_data.values('user').annotate(times=Count('user')).order_by('-times')[:10]
|
||
host_top_ten = week_data.values('host').annotate(times=Count('host')).order_by('-times')[:10]
|
||
user_dic, host_dic = get_data(week_data, user_top_ten, 'user'), get_data(week_data, host_top_ten, 'host')
|
||
|
||
# a week data
|
||
week_users = week_data.values('user').distinct().count()
|
||
week_hosts = week_data.count()
|
||
|
||
user_top_five = week_data.values('user').annotate(times=Count('user')).order_by('-times')[:5]
|
||
color = ['label-success', 'label-info', 'label-primary', 'label-default', 'label-warnning']
|
||
|
||
# perm apply latest 10
|
||
# perm_apply_10 = Apply.objects.order_by('-date_add')[:10]
|
||
|
||
# latest 10 login
|
||
login_10 = Log.objects.order_by('-start_time')[:10]
|
||
login_more_10 = Log.objects.order_by('-start_time')[10:21]
|
||
|
||
# a week top 10
|
||
for user_info in user_top_ten:
|
||
username = user_info.get('user')
|
||
last = Log.objects.filter(user=username).latest('start_time')
|
||
user_info['last'] = last
|
||
|
||
top = {'user': '活跃用户数', 'host': '活跃主机数', 'times': '登录次数'}
|
||
top_dic = {}
|
||
for key, value in top.items():
|
||
li = []
|
||
for t in li_date:
|
||
year, month, day = t.year, t.month, t.day
|
||
if key != 'times':
|
||
times = week_data.filter(start_time__year=year, start_time__month=month, start_time__day=day).values(key).distinct().count()
|
||
else:
|
||
times = week_data.filter(start_time__year=year, start_time__month=month, start_time__day=day).count()
|
||
li.append(times)
|
||
top_dic[value] = li
|
||
return render_to_response('index.html', locals(), context_instance=RequestContext(request))
|
||
|
||
|
||
def skin_config(request):
|
||
return render_to_response('skin_config.html')
|
||
|
||
|
||
# def pages(posts, r):
|
||
# """分页公用函数"""
|
||
# contact_list = posts
|
||
# p = paginator = Paginator(contact_list, 10)
|
||
# try:
|
||
# current_page = int(r.GET.get('page', '1'))
|
||
# except ValueError:
|
||
# current_page = 1
|
||
#
|
||
# page_range = page_list_return(len(p.page_range), current_page)
|
||
#
|
||
# try:
|
||
# contacts = paginator.page(current_page)
|
||
# except (EmptyPage, InvalidPage):
|
||
# contacts = paginator.page(paginator.num_pages)
|
||
#
|
||
# if current_page >= 5:
|
||
# show_first = 1
|
||
# else:
|
||
# show_first = 0
|
||
# if current_page <= (len(p.page_range) - 3):
|
||
# show_end = 1
|
||
# else:
|
||
# show_end = 0
|
||
#
|
||
# return contact_list, p, contacts, page_range, current_page, show_first, show_end
|
||
|
||
|
||
def is_latest():
|
||
node = uuid.getnode()
|
||
jsn = uuid.UUID(int=node).hex[-12:]
|
||
with open(os.path.join(BASE_DIR, 'version')) as f:
|
||
current_version = f.read()
|
||
lastest_version = urllib.urlopen('http://www.jumpserver.org/lastest_version.html?jsn=%s' % jsn).read().strip()
|
||
|
||
if current_version != lastest_version:
|
||
pass
|
||
|
||
|
||
def login(request):
|
||
"""登录界面"""
|
||
if request.session.get('username'):
|
||
return HttpResponseRedirect('/')
|
||
if request.method == 'GET':
|
||
return render_to_response('login.html')
|
||
else:
|
||
username = request.POST.get('username')
|
||
password = request.POST.get('password')
|
||
user_filter = User.objects.filter(username=username)
|
||
if user_filter:
|
||
user = user_filter[0]
|
||
if PyCrypt.md5_crypt(password) == user.password:
|
||
request.session['user_id'] = user.id
|
||
user_filter.update(last_login=datetime.datetime.now())
|
||
if user.role == 'SU':
|
||
request.session['role_id'] = 2
|
||
elif user.role == 'GA':
|
||
request.session['role_id'] = 1
|
||
else:
|
||
request.session['role_id'] = 0
|
||
response = HttpResponseRedirect('/', )
|
||
response.set_cookie('username', username, expires=604800)
|
||
response.set_cookie('seed', PyCrypt.md5_crypt(password), expires=604800)
|
||
return response
|
||
else:
|
||
error = '密码错误，请重新输入。'
|
||
else:
|
||
error = '用户不存在。'
|
||
return render_to_response('login.html', {'error': error})
|
||
|
||
|
||
def logout(request):
|
||
request.session.delete()
|
||
return HttpResponseRedirect('/login/')
|
||
|
||
#
|
||
# def filter_ajax_api(request):
|
||
# attr = request.GET.get('attr', 'user')
|
||
# value = request.GET.get('value', '')
|
||
# if attr == 'user':
|
||
# contact_list = User.objects.filter(name__icontains=value)
|
||
# elif attr == "user_group":
|
||
# contact_list = UserGroup.objects.filter(name__icontains=value)
|
||
# elif attr == "asset":
|
||
# contact_list = Asset.objects.filter(ip__icontains=value)
|
||
# elif attr == "asset":
|
||
# contact_list = BisGroup.objects.filter(name__icontains=value)
|
||
#
|
||
# return render_to_response('filter_ajax_api.html', locals())
|
||
#
|
||
#
|
||
# def install(request):
|
||
# from juser.models import DEPT, User
|
||
# if User.objects.filter(id=5000):
|
||
# return http_error(request, 'Jumpserver已初始化，不能重复安装！')
|
||
#
|
||
# dept = DEPT(id=1, name="超管部", comment="超级管理部门")
|
||
# dept.save()
|
||
# dept2 = DEPT(id=2, name="默认", comment="默认部门")
|
||
# dept2.save()
|
||
# IDC(id=1, name="默认", comment="默认IDC").save()
|
||
# BisGroup(id=1, name="ALL", dept=dept, comment="所有主机组").save()
|
||
#
|
||
# User(id=5000, username="admin", password=PyCrypt.md5_crypt('admin'),
|
||
# name='admin', email='admin@jumpserver.org', role='SU', is_active=True, dept=dept).save()
|
||
# return http_success(request, u'Jumpserver初始化成功')
|
||
#
|
||
#
|
||
# def download(request):
|
||
# return render_to_response('download.html', locals(), context_instance=RequestContext(request))
|
||
#
|
||
#
|
||
# def transfer(sftp, filenames):
|
||
# # pool = Pool(processes=5)
|
||
# for filename, file_path in filenames.items():
|
||
# print filename, file_path
|
||
# sftp.put(file_path, '/tmp/%s' % filename)
|
||
# # pool.apply_async(transfer, (sftp, file_path, '/tmp/%s' % filename))
|
||
# sftp.close()
|
||
# # pool.close()
|
||
# # pool.join()
|
||
#
|
||
#
|
||
# def upload(request):
|
||
# pass
|
||
# # user, dept = get_session_user_dept(request)
|
||
# # if request.method == 'POST':
|
||
# # hosts = request.POST.get('hosts')
|
||
# # upload_files = request.FILES.getlist('file[]', None)
|
||
# # upload_dir = "/tmp/%s" % user.username
|
||
# # is_dir(upload_dir)
|
||
# # date_now = datetime.datetime.now().strftime("%Y%m%d%H%M%S")
|
||
# # hosts_list = hosts.split(',')
|
||
# # user_hosts = [asset.ip for asset in user.get_asset()]
|
||
# # unperm_hosts = []
|
||
# # filenames = {}
|
||
# # for ip in hosts_list:
|
||
# # if ip not in user_hosts:
|
||
# # unperm_hosts.append(ip)
|
||
# #
|
||
# # if not hosts:
|
||
# # return HttpResponseNotFound(u'地址不能为空')
|
||
# #
|
||
# # if unperm_hosts:
|
||
# # print hosts_list
|
||
# # return HttpResponseNotFound(u'%s 没有权限.' % ', '.join(unperm_hosts))
|
||
# #
|
||
# # for upload_file in upload_files:
|
||
# # file_path = '%s/%s.%s' % (upload_dir, upload_file.name, date_now)
|
||
# # filenames[upload_file.name] = file_path
|
||
# # f = open(file_path, 'w')
|
||
# # for chunk in upload_file.chunks():
|
||
# # f.write(chunk)
|
||
# # f.close()
|
||
# #
|
||
# # sftps = []
|
||
# # for host in hosts_list:
|
||
# # username, password, host, port = get_connect_item(user.username, host)
|
||
# # try:
|
||
# # t = paramiko.Transport((host, port))
|
||
# # t.connect(username=username, password=password)
|
||
# # sftp = paramiko.SFTPClient.from_transport(t)
|
||
# # sftps.append(sftp)
|
||
# # except paramiko.AuthenticationException:
|
||
# # return HttpResponseNotFound(u'%s 连接失败.' % host)
|
||
# #
|
||
# # # pool = Pool(processes=5)
|
||
# # for sftp in sftps:
|
||
# # transfer(sftp, filenames)
|
||
# # # pool.close()
|
||
# # # pool.join()
|
||
# # return HttpResponse('传送成功')
|
||
# #
|
||
# # return render_to_response('upload.html', locals(), context_instance=RequestContext(request))
|
||
#
|
||
#
|
||
# def node_auth(request):
|
||
# username = request.POST.get('username', ' ')
|
||
# seed = request.POST.get('seed', ' ')
|
||
# filename = request.POST.get('filename', ' ')
|
||
# user = User.objects.filter(username=username, password=seed)
|
||
# auth = 1
|
||
# if not user:
|
||
# auth = 0
|
||
# if not filename.startswith('/opt/jumpserver/logs/connect/'):
|
||
# auth = 0
|
||
# if auth:
|
||
# result = {'auth': {'username': username, 'result': 'success'}}
|
||
# else:
|
||
# result = {'auth': {'username': username, 'result': 'failed'}}
|
||
#
|
||
# return HttpResponse(json.dumps(result, sort_keys=True, indent=2), content_type='application/json')
|