2015-12-19 09:27:12 +00:00
|
|
|
|
#!/usr/bin/python
|
|
|
|
|
# coding: utf-8
|
|
|
|
|
|
|
|
|
|
import subprocess
|
|
|
|
|
import time
|
|
|
|
|
import os
|
|
|
|
|
import sys
|
2015-12-22 09:18:53 +00:00
|
|
|
|
from smtplib import SMTP, SMTPAuthenticationError, SMTPConnectError, SMTPSenderRefused
|
2015-12-19 09:27:12 +00:00
|
|
|
|
import ConfigParser
|
|
|
|
|
import socket
|
2016-01-04 05:56:47 +00:00
|
|
|
|
import random
|
|
|
|
|
import string
|
2015-12-19 09:27:12 +00:00
|
|
|
|
|
2016-02-26 06:13:21 +00:00
|
|
|
|
import re
|
|
|
|
|
import platform
|
2016-02-27 04:01:44 +00:00
|
|
|
|
import shlex
|
2016-02-26 06:13:21 +00:00
|
|
|
|
|
2015-12-19 09:27:12 +00:00
|
|
|
|
jms_dir = os.path.dirname(os.path.abspath(os.path.dirname(__file__)))
|
|
|
|
|
sys.path.append(jms_dir)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def bash(cmd):
|
|
|
|
|
"""
|
|
|
|
|
run a bash shell command
|
|
|
|
|
执行bash命令
|
|
|
|
|
"""
|
2016-02-27 04:01:44 +00:00
|
|
|
|
return shlex.os.system(cmd)
|
2015-12-19 09:27:12 +00:00
|
|
|
|
|
|
|
|
|
|
2016-02-26 06:13:21 +00:00
|
|
|
|
def valid_ip(ip):
|
|
|
|
|
if ('255' in ip) or (ip == "0.0.0.0"):
|
|
|
|
|
return False
|
|
|
|
|
else:
|
|
|
|
|
return True
|
|
|
|
|
|
|
|
|
|
|
2015-12-19 09:27:12 +00:00
|
|
|
|
def color_print(msg, color='red', exits=False):
|
|
|
|
|
"""
|
|
|
|
|
Print colorful string.
|
|
|
|
|
颜色打印字符或者退出
|
|
|
|
|
"""
|
|
|
|
|
color_msg = {'blue': '\033[1;36m%s\033[0m',
|
|
|
|
|
'green': '\033[1;32m%s\033[0m',
|
|
|
|
|
'yellow': '\033[1;33m%s\033[0m',
|
|
|
|
|
'red': '\033[1;31m%s\033[0m',
|
|
|
|
|
'title': '\033[30;42m%s\033[0m',
|
|
|
|
|
'info': '\033[32m%s\033[0m'}
|
|
|
|
|
msg = color_msg.get(color, 'red') % msg
|
|
|
|
|
print msg
|
|
|
|
|
if exits:
|
|
|
|
|
time.sleep(2)
|
|
|
|
|
sys.exit()
|
|
|
|
|
return msg
|
|
|
|
|
|
|
|
|
|
|
2016-02-26 06:13:21 +00:00
|
|
|
|
def get_ip_addr():
|
2015-12-19 09:27:12 +00:00
|
|
|
|
try:
|
|
|
|
|
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
2016-02-26 06:13:21 +00:00
|
|
|
|
s.connect(("8.8.8.8", 80))
|
|
|
|
|
return s.getsockname()[0]
|
|
|
|
|
except Exception:
|
|
|
|
|
if_data = ''.join(os.popen("LANG=C ifconfig").readlines())
|
|
|
|
|
ips = re.findall(r'\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}', if_data, flags=re.MULTILINE)
|
|
|
|
|
ip = filter(valid_ip, ips)
|
|
|
|
|
if ip:
|
|
|
|
|
return ip[0]
|
2015-12-19 09:27:12 +00:00
|
|
|
|
return ''
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class PreSetup(object):
|
|
|
|
|
def __init__(self):
|
|
|
|
|
self.db_host = '127.0.0.1'
|
|
|
|
|
self.db_port = 3306
|
|
|
|
|
self.db_user = 'jumpserver'
|
2015-12-19 13:56:42 +00:00
|
|
|
|
self.db_pass = '5Lov@wife'
|
2015-12-19 09:27:12 +00:00
|
|
|
|
self.db = 'jumpserver'
|
|
|
|
|
self.mail_host = 'smtp.qq.com'
|
|
|
|
|
self.mail_port = 25
|
|
|
|
|
self.mail_addr = 'hello@jumpserver.org'
|
|
|
|
|
self.mail_pass = ''
|
|
|
|
|
self.ip = ''
|
2016-01-04 05:56:47 +00:00
|
|
|
|
self.key = ''.join(random.choice(string.ascii_lowercase + string.digits) \
|
|
|
|
|
for _ in range(16))
|
2016-02-26 06:13:21 +00:00
|
|
|
|
self.dist = platform.dist()[0].lower()
|
2016-02-28 07:23:14 +00:00
|
|
|
|
self.version = platform.dist()[1]
|
2016-02-26 06:13:21 +00:00
|
|
|
|
|
|
|
|
|
@property
|
|
|
|
|
def _is_redhat(self):
|
2016-02-27 04:05:34 +00:00
|
|
|
|
if self.dist == "centos" or self.dist == "redhat" or self.dist == "fedora":
|
2016-02-26 06:13:21 +00:00
|
|
|
|
return True
|
|
|
|
|
|
2016-02-28 06:40:17 +00:00
|
|
|
|
@property
|
|
|
|
|
def _is_centos7(self):
|
2016-02-28 07:23:14 +00:00
|
|
|
|
if self.dist == "centos" and self.version.startswith("7"):
|
|
|
|
|
return True
|
2016-02-28 06:40:17 +00:00
|
|
|
|
|
2016-02-26 06:13:21 +00:00
|
|
|
|
@property
|
|
|
|
|
def _is_ubuntu(self):
|
2016-02-27 04:05:34 +00:00
|
|
|
|
if self.dist == "ubuntu" or self.dist == "debian":
|
2016-02-26 06:13:21 +00:00
|
|
|
|
return True
|
2015-12-19 09:27:12 +00:00
|
|
|
|
|
2016-02-27 04:01:44 +00:00
|
|
|
|
def check_platform(self):
|
2016-02-27 04:43:59 +00:00
|
|
|
|
if not (self._is_redhat or self._is_ubuntu):
|
|
|
|
|
print(u"支持的平台: CentOS, RedHat, Fedora, Debian, Ubuntu, 暂不支持其他平台安装.")
|
|
|
|
|
exit()
|
2016-02-27 04:01:44 +00:00
|
|
|
|
|
2015-12-19 09:27:12 +00:00
|
|
|
|
def write_conf(self, conf_file=os.path.join(jms_dir, 'jumpserver.conf')):
|
|
|
|
|
color_print('开始写入配置文件', 'green')
|
|
|
|
|
conf = ConfigParser.ConfigParser()
|
|
|
|
|
conf.read(conf_file)
|
|
|
|
|
conf.set('base', 'url', 'http://%s' % self.ip)
|
2016-01-04 05:56:47 +00:00
|
|
|
|
conf.set('base', 'key', self.key)
|
2015-12-19 09:27:12 +00:00
|
|
|
|
conf.set('db', 'host', self.db_host)
|
|
|
|
|
conf.set('db', 'port', self.db_port)
|
|
|
|
|
conf.set('db', 'user', self.db_user)
|
|
|
|
|
conf.set('db', 'password', self.db_pass)
|
|
|
|
|
conf.set('db', 'database', self.db)
|
|
|
|
|
conf.set('websocket', 'web_socket_host', '%s:3000' % self.ip)
|
|
|
|
|
conf.set('mail', 'email_host', self.mail_host)
|
|
|
|
|
conf.set('mail', 'email_port', self.mail_port)
|
|
|
|
|
conf.set('mail', 'email_host_user', self.mail_addr)
|
|
|
|
|
conf.set('mail', 'email_host_password', self.mail_pass)
|
|
|
|
|
|
|
|
|
|
with open(conf_file, 'w') as f:
|
|
|
|
|
conf.write(f)
|
|
|
|
|
|
|
|
|
|
def _setup_mysql(self):
|
|
|
|
|
color_print('开始安装设置mysql (请手动设置mysql安全)', 'green')
|
2015-12-20 09:45:57 +00:00
|
|
|
|
color_print('默认用户名: %s 默认密码: %s' % (self.db_user, self.db_pass), 'green')
|
2016-02-26 06:13:21 +00:00
|
|
|
|
if self._is_redhat:
|
|
|
|
|
bash('yum -y install mysql-server')
|
|
|
|
|
bash('service mysqld start')
|
|
|
|
|
bash('chkconfig mysqld on')
|
|
|
|
|
bash('mysql -e "create database %s default charset=utf8"' % self.db)
|
|
|
|
|
bash('mysql -e "grant all on %s.* to \'%s\'@\'%s\' identified by \'%s\'"' % (self.db,
|
|
|
|
|
self.db_user,
|
|
|
|
|
self.db_host,
|
|
|
|
|
self.db_pass))
|
|
|
|
|
if self._is_ubuntu:
|
2016-02-27 05:24:19 +00:00
|
|
|
|
cmd1 = "echo mysql-server mysql-server/root_password select '' | debconf-set-selections"
|
|
|
|
|
cmd2 = "echo mysql-server mysql-server/root_password_again select '' | debconf-set-selections"
|
|
|
|
|
cmd3 = "apt-get -y install mysql-server"
|
2016-02-27 04:43:59 +00:00
|
|
|
|
bash('%s; %s; %s' % (cmd1, cmd2, cmd3))
|
2016-02-27 04:01:44 +00:00
|
|
|
|
bash('service mysql start')
|
2016-02-26 06:13:21 +00:00
|
|
|
|
bash('mysql -e "create database %s default charset=utf8"' % self.db)
|
|
|
|
|
bash('mysql -e "grant all on %s.* to \'%s\'@\'%s\' identified by \'%s\'"' % (self.db,
|
|
|
|
|
self.db_user,
|
|
|
|
|
self.db_host,
|
|
|
|
|
self.db_pass))
|
|
|
|
|
|
|
|
|
|
def _set_env(self):
|
2015-12-19 09:27:12 +00:00
|
|
|
|
color_print('开始关闭防火墙和selinux', 'green')
|
2016-02-26 06:13:21 +00:00
|
|
|
|
if self._is_redhat:
|
2016-02-28 07:01:55 +00:00
|
|
|
|
os.system("export LANG='en_US.UTF-8'")
|
2016-02-28 06:40:17 +00:00
|
|
|
|
if self._is_centos7:
|
|
|
|
|
cmd1 = "systemctl status firewalld 2> /dev/null 1> /dev/null"
|
|
|
|
|
cmd2 = "systemctl stop firewalld"
|
|
|
|
|
cmd3 = "systemctl disable firewalld"
|
|
|
|
|
bash('%s && %s && %s' % (cmd1, cmd2, cmd3))
|
2016-02-28 07:01:55 +00:00
|
|
|
|
bash('localectl set-locale LANG=en_US.UTF-8')
|
2016-02-28 06:40:17 +00:00
|
|
|
|
bash('setenforce 0')
|
|
|
|
|
else:
|
2016-02-28 07:01:55 +00:00
|
|
|
|
bash("sed -i 's/LANG=.*/LANG=en_US.UTF-8/g' /etc/sysconfig/i18n")
|
2016-02-28 06:40:17 +00:00
|
|
|
|
bash('service iptables stop && chkconfig iptables off && setenforce 0')
|
2016-02-28 07:01:55 +00:00
|
|
|
|
|
2016-02-26 06:13:21 +00:00
|
|
|
|
if self._is_ubuntu:
|
|
|
|
|
os.system("export LANG='en_US.UTF-8'")
|
2016-02-28 06:40:17 +00:00
|
|
|
|
bash("which iptables && iptables -F")
|
2016-02-26 06:13:21 +00:00
|
|
|
|
bash('which selinux && setenforce 0')
|
2015-12-19 09:27:12 +00:00
|
|
|
|
|
|
|
|
|
def _test_db_conn(self):
|
2016-02-26 06:13:21 +00:00
|
|
|
|
bash("pip install mysql-python")
|
|
|
|
|
import MySQLdb
|
2015-12-19 09:27:12 +00:00
|
|
|
|
try:
|
2015-12-29 05:09:05 +00:00
|
|
|
|
MySQLdb.connect(host=self.db_host, port=int(self.db_port),
|
2015-12-19 09:27:12 +00:00
|
|
|
|
user=self.db_user, passwd=self.db_pass, db=self.db)
|
|
|
|
|
color_print('连接数据库成功', 'green')
|
|
|
|
|
return True
|
|
|
|
|
except MySQLdb.OperationalError, e:
|
|
|
|
|
color_print('数据库连接失败 %s' % e, 'red')
|
|
|
|
|
return False
|
|
|
|
|
|
|
|
|
|
def _test_mail(self):
|
|
|
|
|
try:
|
|
|
|
|
smtp = SMTP(self.mail_host, port=self.mail_port, timeout=2)
|
|
|
|
|
smtp.login(self.mail_addr, self.mail_pass)
|
|
|
|
|
smtp.sendmail(self.mail_addr, (self.mail_addr, ),
|
|
|
|
|
'''From:%s\r\nTo:%s\r\nSubject:Jumpserver Mail Test!\r\n\r\n Mail test passed!\r\n''' %
|
|
|
|
|
(self.mail_addr, self.mail_addr))
|
|
|
|
|
smtp.quit()
|
|
|
|
|
return True
|
|
|
|
|
|
2015-12-31 08:35:17 +00:00
|
|
|
|
except Exception, e:
|
2015-12-19 09:27:12 +00:00
|
|
|
|
color_print(e, 'red')
|
2015-12-31 08:35:17 +00:00
|
|
|
|
skip = raw_input('是否跳过(y/n) [n]? : ')
|
|
|
|
|
if skip == 'y':
|
|
|
|
|
return True
|
2015-12-19 09:27:12 +00:00
|
|
|
|
return False
|
|
|
|
|
|
2016-02-26 06:13:21 +00:00
|
|
|
|
def _rpm_repo(self):
|
|
|
|
|
if self._is_redhat:
|
|
|
|
|
color_print('开始安装epel源', 'green')
|
|
|
|
|
bash('yum -y install epel-release')
|
|
|
|
|
|
|
|
|
|
def _depend_rpm(self):
|
|
|
|
|
color_print('开始安装依赖包', 'green')
|
|
|
|
|
if self._is_redhat:
|
2016-02-28 06:40:17 +00:00
|
|
|
|
bash('yum -y install git python-pip mysql-devel gcc automake autoconf python-devel vim sshpass lrzsz readline-devel')
|
2016-02-26 06:13:21 +00:00
|
|
|
|
if self._is_ubuntu:
|
2016-02-28 06:40:17 +00:00
|
|
|
|
bash("apt-get -y --force-yes install git python-pip gcc automake autoconf vim sshpass libmysqld-dev python-all-dev lrzsz libreadline-dev")
|
2015-12-19 09:27:12 +00:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@staticmethod
|
|
|
|
|
def _require_pip():
|
|
|
|
|
color_print('开始安装依赖pip包', 'green')
|
|
|
|
|
bash('pip install -r requirements.txt')
|
|
|
|
|
|
|
|
|
|
def _input_ip(self):
|
2015-12-31 03:20:41 +00:00
|
|
|
|
ip = raw_input('\n请输入您服务器的IP地址,用户浏览器可以访问 [%s]: ' % get_ip_addr()).strip()
|
2015-12-19 09:27:12 +00:00
|
|
|
|
self.ip = ip if ip else get_ip_addr()
|
|
|
|
|
|
|
|
|
|
def _input_mysql(self):
|
|
|
|
|
while True:
|
2015-12-20 09:45:57 +00:00
|
|
|
|
mysql = raw_input('是否安装新的MySQL服务器? (y/n) [y]: ')
|
|
|
|
|
if mysql != 'n':
|
2015-12-19 09:27:12 +00:00
|
|
|
|
self._setup_mysql()
|
2015-12-19 13:56:42 +00:00
|
|
|
|
else:
|
2015-12-31 03:20:41 +00:00
|
|
|
|
db_host = raw_input('请输入数据库服务器IP [127.0.0.1]: ').strip()
|
|
|
|
|
db_port = raw_input('请输入数据库服务器端口 [3306]: ').strip()
|
|
|
|
|
db_user = raw_input('请输入数据库服务器用户 [root]: ').strip()
|
|
|
|
|
db_pass = raw_input('请输入数据库服务器密码: ').strip()
|
|
|
|
|
db = raw_input('请输入使用的数据库 [jumpserver]: ').strip()
|
2015-12-19 13:56:42 +00:00
|
|
|
|
|
|
|
|
|
if db_host: self.db_host = db_host
|
|
|
|
|
if db_port: self.db_port = db_port
|
|
|
|
|
if db_user: self.db_user = db_user
|
|
|
|
|
if db_pass: self.db_pass = db_pass
|
|
|
|
|
if db: self.db = db
|
2015-12-19 09:27:12 +00:00
|
|
|
|
|
|
|
|
|
if self._test_db_conn():
|
|
|
|
|
break
|
|
|
|
|
|
|
|
|
|
print
|
|
|
|
|
|
|
|
|
|
def _input_smtp(self):
|
|
|
|
|
while True:
|
|
|
|
|
self.mail_host = raw_input('请输入SMTP地址: ').strip()
|
2015-12-20 09:45:57 +00:00
|
|
|
|
mail_port = raw_input('请输入SMTP端口 [25]: ').strip()
|
2015-12-19 09:27:12 +00:00
|
|
|
|
self.mail_addr = raw_input('请输入账户: ').strip()
|
|
|
|
|
self.mail_pass = raw_input('请输入密码: ').strip()
|
|
|
|
|
|
2015-12-20 09:45:57 +00:00
|
|
|
|
if mail_port: self.mail_port = int(mail_port)
|
2015-12-19 13:56:42 +00:00
|
|
|
|
|
2015-12-19 09:27:12 +00:00
|
|
|
|
if self._test_mail():
|
|
|
|
|
color_print('\n\t请登陆邮箱查收邮件, 然后确认是否继续安装\n', 'green')
|
|
|
|
|
smtp = raw_input('是否继续? (y/n) [y]: ')
|
|
|
|
|
if smtp == 'n':
|
|
|
|
|
continue
|
|
|
|
|
else:
|
|
|
|
|
break
|
|
|
|
|
print
|
|
|
|
|
|
|
|
|
|
def start(self):
|
2015-12-29 03:37:52 +00:00
|
|
|
|
color_print('请务必先查看wiki https://github.com/ibuler/jumpserver/wiki/Quickinstall')
|
|
|
|
|
time.sleep(3)
|
2016-02-27 04:05:34 +00:00
|
|
|
|
self.check_platform()
|
2016-02-26 06:13:21 +00:00
|
|
|
|
self._rpm_repo()
|
|
|
|
|
self._depend_rpm()
|
|
|
|
|
self._require_pip()
|
2015-12-19 09:27:12 +00:00
|
|
|
|
self._set_env()
|
|
|
|
|
self._input_ip()
|
|
|
|
|
self._input_mysql()
|
|
|
|
|
self._input_smtp()
|
|
|
|
|
self.write_conf()
|
|
|
|
|
os.system('python %s' % os.path.join(jms_dir, 'install/next.py'))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
|
pre_setup = PreSetup()
|
|
|
|
|
pre_setup.start()
|