白泽自动化运维系统.zip

#!/usr/local/baize/env/bin/python #coding:utf-8 # (c) 2016 , Tianbiao Zu <zutianbian@qq.com> # # 该文件是白泽自动化管理系统的一部分,是白泽系统的django views配置文件 ################################################################################################### import os import json import re import logging import hashlib import base64 import time import urllib import urllib2 import baize.settings as C from ftplib import FTP from urlparse import urlparse from django.http import HttpResponse from django.views.decorators.csrf import csrf_exempt from django.shortcuts import render_to_response, HttpResponseRedirect from API.API_web.config_items import Config from API.API_web.reciver import * from django.contrib.auth.hashers import check_password, make_password from django.contrib.auth.models import User from django.contrib import auth from django.contrib.auth import authenticate from django.core.context_processors import csrf from API.API_web.generate_password import fun_generate_password from django.core.mail import send_mass_mail from baize.settings import EMAIL_HOST_USER, ADMIN_EMAIL from django.utils import timezone from django.contrib.auth.decorators import login_required from pyecharts import Line from models import Asset, Detect_Role, Geo, Isp, Detect_Task, Ping_Detect, Traceroute_Detect, Curl_Detect, \ Remote_Control_Command, Remote_Control_Script, Remote_Control_Copy, Asset_Tag, Asset_Tag_Data, \ Business_Tree, Configure_Manage_Work, Configure_Manage_Work_Tag_Data, Configure_Manage_Task, \ Configure_Manage_Task_Data, Authority_Url, Authority_Bussiness, Bussiness, Bussiness_Btn, Alarm_Msg, \ Alarm_Person_Data, Alarm, Alarm_Person, Alarm_Msg_Template, Alarm_Msg_Ignore, Monitor_Item, Monitor_Screen, \ Authority_Screen, Monitor_Chart, Monitor_Screen_Data def authority_url(func): def inner(request, *args, **kwargs): try: superuser = request.user.is_superuser except Exception, e: superuser = True if not superuser: try: list_authority = Authority_Url.objects.filter(url__contains=request.path, user=request.user) except Exception, e: list_authority = [] if not list_authority: http_accept = request.META.get('HTTP_ACCEPT', '') if 'json' not in http_accept: return fun_403(request, *args, **kwargs) else: response_data = { "success": False, "msg": u"url权限不足,请联系管理员" } return HttpResponse(json.dumps(response_data), content_type="application/json") return func(request, *args, **kwargs) return inner def baize_register(username, email): C_web = Config() length = int(C_web.item(item_name='LENGTH_PASSWORD_RANDOM', group_name='PASSWORD', default=u"""8""")) _secret_method = C_web.item(item_name='SECRET_METHOD', group_name='MASTER', default=u"""pbkdf2_sha256""") _password = fun_generate_password(length=length) try: _secret_password = make_password(_password, None, _secret_method) except Exception, e: logger = logging.getLogger('log_file') logger.error(e) if C.LOG_SCREEN == 'ON': logger = logging.getLogger('log_screen') logger.error(e) _secret_password = _password message1 = (u"【Baize】注册成功", u"恭喜您,注册白泽系统成功,您的账号: %s, 密码: %s" % (username, _password), EMAIL_HOST_USER, [email, ]) message2 = (u"【Baize】注册提醒", u"%s注册成功,邮箱: %s, 密码: %s" % (username, email, _password), EMAIL_HOST_USER, [ADMIN_EMAIL, ]) try: user = User.objects.filter(username=username, email=email) except Exception, e: user = None if user: log_message = u"%s已经存在,邮箱: %s" % (username, email) logger = logging.getLogger('log_file') logger.info(log_message) if C.LOG_SCREEN == 'ON': logger = logging.getLogger('log_screen') logger.info(log_message) _re = { "success": False, "msg": u"您注册的邮箱已经存在" } else: user = User(username=username, email=email, password=_secret_password, is_superuser=True) try: _res = send_mass_mail((message1, message2), fail_silently=False) except Exception, e: logger = logging.getLogger('log_file') logger.error(e) if C.LOG_SCREEN == 'ON': logger = logging.getLogger('log_screen') logger.error(e) _res = 0 if _res >= 2: user.save() log_message = u"%s注册成功,邮箱: %s" % (username, email) logger = logging.getLogger('log_file') logger.info(log_message) if C.LOG_SCREEN == 'ON': logger = logging.getLogger('log_screen') logger.info(log_message) _re = { "success": True, "msg": u"注册成功" } elif _res == 0: log_message = u"注册失败,邮箱%s不存在" % email logger = logging.getLogger('log_file') logger.info(log_message) if C.LOG_SCREEN == 'ON': logger = logging.getLogger('log_screen') logger.info(log_message) _re = { "success": False, "msg": u"您注册的邮箱不存在" } else: log_message = u"%s注册失败,邮箱: %s" % (username, email) logger = logging.getLogger('log_file') logger.info(log_message) if C.LOG_SCREEN == 'ON': logger = logging.getLogger('log_screen') logger.info(log_message) _re = { "success": False, "msg": u"您注册的邮箱不存在" } return _re @csrf_exempt def register(request): C_web = Config() argv_local = dict() argv_local['BODY_BG'] = 'gray-bg' if request.method == 'GET': argv_local["DEFAULT"] = { "WEB_TITLE": C_web.item(item_name='WEB_TITLE', group_name='DEFAULT', default=u"Baize"), "MSG_WELCOME_REGISTER": C_web.item(item_name='MSG_WELCOME_REGISTER', group_name='DEFAULT', default=u"Register to Baize"), "MSG_REGISTER": C_web.item(item_name='MSG_REGISTER', group_name='DEFAULT', default=u"""<div class="text-normal">Create account to see it in action.</div>"""), "MSG_ACCOUNT_HAVE": C_web.item(item_name='MSG_ACCOUNT_HAVE', group_name='DEFAULT', default=u"""Already have an account?"""), "WEB_FOOTER": C_web.item(item_name='WEB_FOOTER', group_name='DEFAULT', default=u"""© 白泽自动化运维开发小组 2017"""), } return render_to_response('login/register.html', argv_local) else: username = request.POST['username'] email = request.POST['email'] _re = baize_register(username, email) if _re['msg'] == u'注册成功': argv_local["DEFAULT"] = { "WEB_TITLE": C_web.item(item_name='WEB_TITLE', group_name='DEFAULT', default=u"""Baize"""), "MSG_WELCOME": C_web.item(item_name='MSG_WELCOME', group_name='DEFAULT', default=u"""Welcome to Baize"""), "MSG_USAGE": C_web.item(item_name='MSG_USAGE', group_name='DEFAULT', default=u"""打造通用型自动化运维平台.助力传统运维向Devops转型"""), "MSG_LOGIN": C_web.item(item_name='MSG_LOGIN_SUCCESS', group_name='DEFAULT',