diff --git a/install/install.py b/install/install.py index 317c8ee80..1f99ee9a1 100644 --- a/install/install.py +++ b/install/install.py @@ -5,6 +5,16 @@ import subprocess import time import os import sys +import smtplib +import MySQLdb +from smtplib import SMTP, SMTPAuthenticationError, SMTPConnectError +import ConfigParser +import socket +import fcntl +import struct + +jms_dir = os.path.dirname(os.path.abspath(os.path.dirname(__file__))) +sys.path.append(jms_dir) def bash(cmd): @@ -34,7 +44,91 @@ def color_print(msg, color='red', exits=False): return msg +def get_ip_addr(ifname='eth0'): + try: + s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) + return socket.inet_ntoa(fcntl.ioctl( + s.fileno(), + 0x8915, + struct.pack('256s', ifname[:15]) + )[20:24]) + except: + ips = os.popen("LANG=C ifconfig | grep \"inet addr\" | grep -v \"127.0.0.1\" | awk -F \":\" '{print $2}' | awk '{print $1}'").readlines() + if len(ips) > 0: + return ips[0] + return '' + + class PreSetup(object): + def __init__(self): + self.db_host = '127.0.0.1' + self.db_port = 3306 + self.db_user = 'jumpserver' + self.db_pass = 'mysql234' + self.db = 'jumpserver' + self.mail_host = 'smtp.qq.com' + self.mail_port = 25 + self.mail_addr = 'hello@jumpserver.org' + self.mail_pass = '' + self.ip = '' + + 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) + conf.set('db', 'host', self.db_host) + conf.set('db', 'port', self.db_port) + conf.set('db', 'user', self.db_user) + conf.set('db', 'pass', 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') + bash('yum -y install mysql-server') + bash('service mysqld start') + 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)) + + @staticmethod + def _set_env(): + color_print('开始关闭防火墙和selinux', 'green') + bash('service iptables stop && chkconfig iptables off && setenforce 0') + + def _test_db_conn(self): + try: + MySQLdb.connect(host=self.db_host, port=self.db_port, + 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 + + except (SMTPAuthenticationError, socket.timeout), e: + color_print(e, 'red') + return False @staticmethod def _rpm_repo(): @@ -51,6 +145,58 @@ class PreSetup(object): color_print('开始安装依赖pip包', 'green') bash('pip install -r requirements.txt') + def _input_ip(self): + ip = raw_input('\n请输入您服务器的IP地址，用户浏览器可以访问 [%s]: ' % get_ip_addr()) + self.ip = ip if ip else get_ip_addr() + + def _input_mysql(self): + while True: + db_host = raw_input('请输入数据库服务器IP [127.0.0.1]: ') + db_port = raw_input('请输入数据库服务器端口 [3306]: ') + db_user = raw_input('请输入数据库服务器用户 [root]: ') + db_pass = raw_input('请输入数据库服务器密码: ') + db = raw_input('请输入使用的数据库 [jumpserver]: ') + + 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 + + mysql = raw_input('是否使用已经存在的数据库服务器? (y/n) [n]: ') + + if mysql != 'y': + self._setup_mysql() + + if self._test_db_conn(): + break + + print + + def _input_smtp(self): + while True: + self.mail_host = raw_input('请输入SMTP地址: ').strip() + self.mail_port = int(raw_input('请输入SMTP端口: ').strip()) + self.mail_addr = raw_input('请输入账户: ').strip() + self.mail_pass = raw_input('请输入密码: ').strip() + + if self._test_mail(): + color_print('\n\t请登陆邮箱查收邮件, 然后确认是否继续安装\n', 'green') + smtp = raw_input('是否继续? (y/n) [y]: ') + if smtp == 'n': + continue + else: + break + print + + @staticmethod + def _pull(): + color_print('开始更新jumpserver', 'green') + bash('git pull') + os.chdir(jms_dir) + os.chmod('logs', 0777) + os.chmod('keys', 0777) + def start(self): self._rpm_repo() self._depend_rpm() diff --git a/install/next.py b/install/next.py index 7837b17a5..5999b7b41 100644 --- a/install/next.py +++ b/install/next.py @@ -3,15 +3,9 @@ import sys import os -import MySQLdb -import smtplib -import ConfigParser import django from django.core.management import execute_from_command_line -import socket -from smtplib import SMTP, SMTPAuthenticationError, SMTPConnectError -import fcntl -import struct +import shutil jms_dir = os.path.dirname(os.path.abspath(os.path.dirname(__file__))) sys.path.append(jms_dir) @@ -20,153 +14,18 @@ os.environ['DJANGO_SETTINGS_MODULE'] = 'jumpserver.settings' if django.get_version() != '1.6': setup = django.setup() -from jumpserver.api import chown, bash, PyCrypt, ServerError, get_object, mkdir from juser.user_api import db_add_user, server_add_user from connect import color_print -def get_ip_addr(ifname='eth0'): - try: - s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) - return socket.inet_ntoa(fcntl.ioctl( - s.fileno(), - 0x8915, - struct.pack('256s', ifname[:15]) - )[20:24]) - except: - ips = os.popen("LANG=C ifconfig | grep \"inet addr\" | grep -v \"127.0.0.1\" | awk -F \":\" '{print $2}' | awk '{print $1}'").readlines() - if len(ips) > 0: - return ips[0] - return '' - - class Setup(object): """ 安装jumpserver向导 """ + def __init__(self): - self.db_host = '127.0.0.1' - self.db_port = 3306 - self.db_user = 'jumpserver' - self.db_pass = 'mysql234' - self.db = 'jumpserver' - self.mail_host = 'smtp.qq.com' - self.mail_port = 25 - self.mail_addr = 'hello@jumpserver.org' - self.mail_pass = '' - self.ip = '' - self.admin_user = 'admin' - self.admin_pass = 'Lov@jms' - - 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) - conf.set('db', 'host', self.db_host) - conf.set('db', 'port', self.db_port) - conf.set('db', 'user', self.db_user) - conf.set('db', 'pass', 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') - bash('yum -y install mysql-server') - bash('service mysqld start') - 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)) - - @staticmethod - def _pull(): - color_print('开始更新jumpserver', 'green') - bash('git pull') - os.chdir(jms_dir) - mkdir('logs', mode=0777) - mkdir('keys', mode=0777) - - @staticmethod - def _set_env(): - color_print('开始关闭防火墙和selinux', 'green') - bash('service iptables stop && chkconfig iptables off && setenforce 0') - - def _test_db_conn(self): - try: - MySQLdb.connect(host=self.db_host, port=self.db_port, - 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 - - except (SMTPAuthenticationError, socket.timeout), e: - color_print(e, 'red') - return False - - def _input_ip(self): - ip = raw_input('\n请输入您服务器的IP地址，用户浏览器可以访问 [%s]: ' % get_ip_addr()) - self.ip = ip if ip else get_ip_addr() - - def _input_mysql(self): - while True: - db_host = raw_input('请输入数据库服务器IP [127.0.0.1]: ') - db_port = raw_input('请输入数据库服务器端口 [3306]: ') - db_user = raw_input('请输入数据库服务器用户 [root]: ') - db_pass = raw_input('请输入数据库服务器密码: ') - db = raw_input('请输入使用的数据库 [jumpserver]: ') - - 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 - - mysql = raw_input('是否使用已经存在的数据库服务器? (y/n) [n]: ') - - if mysql != 'y': - self._setup_mysql() - - if self._test_db_conn(): - break - - print - - def _input_smtp(self): - while True: - self.mail_host = raw_input('请输入SMTP地址: ').strip() - self.mail_port = int(raw_input('请输入SMTP端口: ').strip()) - self.mail_addr = raw_input('请输入账户: ').strip() - self.mail_pass = raw_input('请输入密码: ').strip() - - if self._test_mail(): - color_print('\n\t请登陆邮箱查收邮件, 然后确认是否继续安装\n', 'green') - smtp = raw_input('是否继续? (y/n) [y]: ') - if smtp == 'n': - continue - else: - break - print + self.admin_user = '' + self.admin_pass = '' def _input_admin(self): while True: @@ -190,15 +49,14 @@ class Setup(object): admin_groups='', email='admin@jumpserver.org', uuid='MayBeYouAreTheFirstUser', is_active=True) server_add_user(self.admin_user, self.admin_user, ssh_key_login_need=False) + @staticmethod + def _cp_zzsh(): + os.chdir(jms_dir) + shutil.copy('zzjumpserver.sh', '/etc/profile.d/') + def start(self): print "开始安装Jumpserver, 要求环境为 CentOS 6.5 x86_64" - self._pull() - self._input_ip() - self._input_mysql() - self._input_smtp() self._sync_db() - self.write_conf() - self._input_admin() self._create_admin()