ops
/
jump-admin
8
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity
Browse Source

cmd args\import logging

master
牛批的一批 5 years ago
parent
80128e4a2f
commit
a0a4921def
1 changed files with 76 additions and 79 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 145
      server/jumpserver.py

145
server/jumpserver.py
View File

@ -1,9 +1,10 @@
# -*- coding: UTF-8 -*-
import argparse
import sys
import getopt
import os
import time
import logging
from threading import Timer
import subprocess
import sqlite3
@ -16,16 +17,15 @@ import requests
from flask import request, Flask, redirect, session, render_template, g
from flask_cors import CORS
# 启动flask
app = Flask(__name__,
static_folder="/home/daniel/vue-admin/dist/static",
template_folder="/home/daniel/vue-admin/dist")
app.config['SECRET_KEY'] = os.urandom(24)
CORS(app)
# http bind
gHost = ""
gPort = 0
gDebug = 0
# 命令行参数
gCmdArgs = {}
# 数据库文件
gSqlite3File = "/usr/local/jumpserver/jumpserver.db"
@ -62,32 +62,28 @@ SSO_EXPIRE_TIMEOUT = 24*60*60
# 解析命令行
def parse_cmd():
''' 解析命令行参数 '''
global gHost, gPort, gDebug
try:
opts, _ = getopt.getopt(sys.argv[1:], "Hh:p:l:d:", [
"ip=", "port=", "debug="])
except getopt.GetoptError:
print('python3 %s -h <host> -p <port> -d <debug>' % (sys.argv[0]))
sys.exit(0)
for opt, arg in opts:
if opt == '-H':
print('python3 %s -h <host> -p <port> -d <debug>' % (sys.argv[0]))
sys.exit(0)
elif opt in ("-h", "--host"):
gHost = arg
elif opt in ("-p", "--port"):
gPort = int(arg)
elif opt in ("-l", "--debug"):
gDebug = int(arg)
if len(gHost) == 0 or gPort == 0:
print('python3 %s -h <host> -p <port> -d <debug>' % (sys.argv[0]))
sys.exit(0)
global gCmdArgs
# 显示默认值
default_value_fmt = " (default: %(default)s)"
# 参数
parser = argparse.ArgumentParser()
parser.add_argument("-host", type=str, help="bind host"+default_value_fmt, default="0.0.0.0")
parser.add_argument("-port", type=int, help="bind port"+default_value_fmt, default=8080)
parser.add_argument("-debug", type=bool, help="enable debug"+default_value_fmt, default=False)
# 解析
gCmdArgs = parser.parse_args()
logging.debug(gCmdArgs)
# 简单校验
if len(gCmdArgs.host) == 0 or gCmdArgs.port <= 0:
parser.print_help()
sys.exit(1)
# 主页url
global gUrl
gUrl = "http://%s:%d/" % (gHost, gPort)
gUrl = "http://%s:%d/" % (gCmdArgs.host, gCmdArgs.port)
# 只有执行结果
@ -105,7 +101,7 @@ def sync_remote_control_file(host, port):
status, output = exec_command_output(
"scp -P %d manager_user.sh %s@%s:.manager_user.sh" % (port, gDefaultSSHAdmin, host))
if status != 0:
print("sync_remote_control_file error %s" % output)
logging.error("sync_remote_control_file error %s", output)
return -1
return 0
@ -181,7 +177,7 @@ def checkCookie(request):
# if rets == None:
# return False
# if rets["ret"] == -200:
# print(u"不信任的主机，请添加白名单")
# logging.error(u"不信任的主机，请添加白名单")
# return False
# if rets["ret"] != 1:
# return False
@ -304,7 +300,7 @@ def do_login(request):
username = request.form.get("username") or "admin"
password = request.form.get("password") or "123456"
print("login username:%s password:%s" % (username, password))
logging.info("login username:%s password:%s", username, password)
resp = {}
resp["msg"] = "ok"
resp["code"] = 200
@ -343,6 +339,7 @@ def do_userlist(request):
resp.append(res)
return json.dumps(resp)
# 添加用户
def do_add_user(request):
if request.method == "GET":
@ -359,7 +356,7 @@ def do_add_user(request):
status, output = exec_command_output("sudo sh %s add %s \"%s\"" % (
gManagerUserShellFile, name, gDefaultInitPassword))
if status != 0:
print("output=%s" % output)
logging.error("output=%s", output)
return "error %s" % output
# 新增用户 sql
@ -381,7 +378,7 @@ def do_del_user(request):
status, output = exec_command_output(
"sudo sh %s del %s" % (gManagerUserShellFile, username))
if status != 0:
print("output=%s" % output)
logging.error("output=%s", output)
return "error %s" % output
g.db.execute("delete from users where name='%s'" % username)
@ -412,7 +409,7 @@ def do_modify_user(request):
# 检查
users = g.db.execute("select sudo,desc from users where name='%s'" % username).fetchall()
if len(users) == 0:
print("user(%s) not exitst" % username)
logging.error("user(%s) not exitst", username)
return "user(%s) not exitst" % username
user = users[0]
@ -430,8 +427,7 @@ def do_modify_user(request):
status, output = exec_command_output(
"sudo sh manager_user.sh %s %s" % (opParam, username))
if status != 0:
print("%s user user(%s) failed! => output=%s" %
(opParam, username, output))
logging.error("%s user user(%s) failed! => output=%s", opParam, username, output)
return "error: %s user user(%s) failed! => output=%s" % (opParam, username, output)
if len(password) > 0:
@ -440,8 +436,7 @@ def do_modify_user(request):
status, output = exec_command_output(
"sudo sh manager_user.sh %s %s" % (opParam, username))
if status != 0:
print("%s user user(%s) failed! => output=%s" %
(opParam, username, output))
logging.error("%s user user(%s) failed! => output=%s", opParam, username, output)
return "error: %s user user(%s) failed! => output=%s" % (opParam, username, output)
if desc != user_desc:
@ -449,13 +444,11 @@ def do_modify_user(request):
# 记录在数据库中
if change:
print("update users set sudo=%d,desc=\"%s\" where name='%s'" %
(sudo, desc, username))
g.db.execute("update users set sudo=%d,desc=\"%s\" where name='%s'" % (
sudo, desc, username))
logging.debug("update users set sudo=%d,desc=\"%s\" where name='%s'", sudo, desc, username)
g.db.execute("update users set sudo=%d,desc=\"%s\" where name='%s'" % (sudo, desc, username))
g.db.commit()
print("modify user:%s successful [output: %s]" % (username, output))
logging.info("modify user:%s successful [output: %s]", username, output)
return "modify user:%s successful [output: %s]" % (username, output)
@ -519,13 +512,13 @@ def do_host_adduser(request):
ret = g.db.execute("select count(1) from hostuser where hostname='%s' and username='%s'" % (
hostname, username)).fetchone()
if (len(ret) > 0 and ret[0]) >= 1:
print("user(%s) exitst on host(%s)" % (username, hostname))
logging.error("user(%s) exitst on host(%s)", username, hostname)
return "user(%s) exitst on host(%s)" % (username, hostname)
# 检查
hostips = g.db.execute("select ip,port from hosts where name='%s'" % (hostname)).fetchone()
if hostips == None:
print("host(%s) not exitst on hosts" % hostname)
logging.error("host(%s) not exitst on hosts", hostname)
return "host(%s) not exitst on hosts" % hostname
hostip = hostips[0]
hostport = int(hostips[1])
@ -537,17 +530,15 @@ def do_host_adduser(request):
status, publicSshRsaKey = exec_command_output(
"sudo cat /home/%s/.ssh/id_rsa.pub" % username)
if status != 0:
print("cat user(%s) id_rsa.pub failed!", username)
logging.error("cat user(%s) id_rsa.pub failed!", username)
return "error %s" % publicSshRsaKey
print("ssh %s@%s -p%d sudo sh .manager_user.sh add %s %s \"\"%s\"\"" %
(gDefaultSSHAdmin, hostip, hostport, username, gDefaultInitPassword, publicSshRsaKey))
logging.debug("ssh %s@%s -p%d sudo sh .manager_user.sh add %s %s \"\"%s\"\"", gDefaultSSHAdmin, hostip, hostport, username, gDefaultInitPassword, publicSshRsaKey)
# 主机上新建用户
status, output = exec_command_output("ssh %s@%s -p%d sudo sh .manager_user.sh add %s %s \"\"%s\"\"" % (
gDefaultSSHAdmin, hostip, hostport, username, gDefaultInitPassword, publicSshRsaKey))
if status != 0:
print("remote add user host(%s) user(%s) failed! => output=%s" %
(hostname, username, output))
logging.error("remote add user host(%s) user(%s) failed! => output=%s", hostname, username, output)
return "error: remote add user host(%s) user(%s) failed! => output=%s" % (hostname, username, output)
# 记录在数据库中
@ -555,8 +546,7 @@ def do_host_adduser(request):
hostname, username)).fetchone()
g.db.commit()
print("host remote =>> add user:%s to host:%s successful [output: %s]" % (
username, hostname, output))
logging.info("host remote =>> add user:%s to host:%s successful [output: %s]", username, hostname, output)
return "host remote =>> add user:%s to host:%s successful" % (username, hostname)
@ -575,13 +565,13 @@ def do_host_deluser(request):
ret = g.db.execute("select count(1) from hostuser where hostname='%s' and username='%s'" % (
hostname, username)).fetchone()
if (len(ret) > 0 and ret[0]) == 0:
print("user(%s) not exitst on host(%s)" % (username, hostname))
logging.error("user(%s) not exitst on host(%s)", username, hostname)
return "user(%s) not exitst on host(%s)" % (username, hostname)
# 检查
hostips = g.db.execute("select ip,port from hosts where name='%s'" % (hostname)).fetchone()
if hostips == None:
print("host(%s) not exitst on hosts" % hostname)
logging.error("host(%s) not exitst on hosts", hostname)
return "host(%s) not exitst on hosts" % hostname
hostip = hostips[0]
hostport = int(hostips[1])
@ -593,8 +583,7 @@ def do_host_deluser(request):
status, output = exec_command_output(
"ssh %s@%s -p%d sudo sh .manager_user.sh del %s" % (gDefaultSSHAdmin, hostip, hostport, username))
if status != 0:
print("remote del user host(%s) user(%s) failed! => output=%s" %
(hostname, username, output))
logging.error("remote del user host(%s) user(%s) failed! => output=%s", hostname, username, output)
return "error: remote del user host(%s) user(%s) failed! => output=%s" % (hostname, username, output)
# 记录在数据库中
@ -602,10 +591,10 @@ def do_host_deluser(request):
hostname, username))
g.db.commit()
print("host remote =>> del user:%s from host:%s successful [output: %s]" % (
username, hostname, output))
logging.info("host remote =>> del user:%s from host:%s successful [output: %s]", username, hostname, output)
return "host remote =>> del user:%s from host:%s successful" % (username, hostname)
# 主机上修改用户信息
def do_host_modifyuser(request):
if request.method == "GET":
@ -627,7 +616,7 @@ def do_host_modifyuser(request):
hostusers = g.db.execute("select sudo,desc from hostuser where hostname='%s' and username='%s' and isdelete=0" % (
hostname, username)).fetchone()
if len(hostusers) > 0 and hostusers[0] != None:
print("user(%s) not exitst on host(%s)" % (username, hostname))
logging.error("user(%s) not exitst on host(%s)", username, hostname)
return "user(%s) not exitst on host(%s)" % (username, hostname)
hostuser = hostusers[0]
@ -641,7 +630,7 @@ def do_host_modifyuser(request):
# 检查
hostips = g.db.execute("select ip,port from hosts where name='%s'" % (hostname)).fetchone()
if hostips == None:
print("host(%s) not exitst on hosts" % hostname)
logging.error("host(%s) not exitst on hosts", hostname)
return "host(%s) not exitst on hosts" % hostname
hostip = hostips[0]
hostport = int(hostips[1])
@ -653,19 +642,20 @@ def do_host_modifyuser(request):
status, output = exec_command_output(
"ssh %s@%s -p%d sudo sh .manager_user.sh %s %s" % (gDefaultSSHAdmin, hostip, hostport, opParam, username))
if status != 0:
print("remote %s user host(%s) user(%s) failed! => output=%s" %
(opParam, hostname, username, output))
return "error: remote %s user host(%s) user(%s) failed! => output=%s" % (opParam, hostname, username, output)
logging.error("remote %s user host(%s) user(%s) failed! => output=%s", opParam, hostname, username, output)
return "error: remote %s user host(%s) user(%s) failed! => output=%s" % (
opParam, hostname, username, output)
if len(password) > 0:
# 主机上修改密码
opParam = "passwd"
status, output = exec_command_output(
"ssh %s@%s -p%d sudo sh .manager_user.sh %s %s %s" % (gDefaultSSHAdmin, hostip, hostport, opParam, username, password))
"ssh %s@%s -p%d sudo sh .manager_user.sh %s %s %s" % (
gDefaultSSHAdmin, hostip, hostport, opParam, username, password))
if status != 0:
print("remote %s user host(%s) user(%s) failed! => output=%s" %
(opParam, hostname, username, output))
return "error: remote %s user host(%s) user(%s) failed! => output=%s" % (opParam, hostname, username, output)
logging.error("remote %s user host(%s) user(%s) failed! => output=%s", opParam, hostname, username, output)
return "error: remote %s user host(%s) user(%s) failed! => output=%s" % (
opParam, hostname, username, output)
if desc != user_desc:
change = True
@ -676,8 +666,7 @@ def do_host_modifyuser(request):
sudo, desc, hostname, username))
g.db.commit()
print("host remote =>> modify user:%s from host:%s successful [output: %s]" % (
username, hostname, output))
logging.info("host remote =>> modify user:%s from host:%s successful [output: %s]", username, hostname, output)
return "host remote =>> modify user:%s from host:%s successful [output: %s]" % (username, hostname, output)
@ -690,7 +679,8 @@ def do_userhostlist(request):
else:
return "invalid request for getting user host list"
hosts = g.db.execute("select id,name,ip,port,desc,date from hosts where isdelete=0 and name in (select hostname from hostuser where username='%s')" % username).fetchall()
hosts = g.db.execute(
"select id,name,ip,port,desc,date from hosts where isdelete=0 and name in (select hostname from hostuser where username='%s')" % username).fetchall()
resp = []
for host in hosts:
res = {}
@ -779,8 +769,16 @@ def sso_timeout():
# 循环定时器
Timer(SSO_TIMER_PERIOD, sso_timeout).start()
def init_log():
# 设置日志
LOG_FORMAT = "[%(asctime)s %(levelname)s %(filename)s:%(funcName)s:%(lineno)d] => %(message)s"
logging.basicConfig(level=logging.DEBUG, format=LOG_FORMAT)
def main():
# 初始化日志
init_log()
# 解析命令行
parse_cmd()
@ -792,12 +790,11 @@ def main():
# 启动HTTP服务
app.run(
host=gHost,
port=gPort,
debug=gDebug
host=gCmdArgs.host,
port=gCmdArgs.port,
debug=gCmdArgs.debug
)
# 入口
if __name__ == '__main__':
main()
Write Preview
Loading…
Cancel
Save