#-*- coding:utf-8 -*-
'''
python一键启动服务器--所有部署的项目
'''
import paramiko
import configparser
import time
import ast
import threadpool
##设置日志打印级别
#logging.basicConfig(format='%(asctime)s - %(pathname)s[line:%(lineno)d] - %(levelname)s: %(message)s',level=logging.DEBUG)
#配置文件读取
cf = configparser.ConfigParser()
cf.read(".config.ini") # 读取配置文件，如果写文件的绝对路径，就可以不用os模块
missHosts=["data4","name2","name4"]#todo 排除host
pool = threadpool.ThreadPool(4)
'''
执行操作
'''
def operation():
secs = cf.sections() # 获取文件中所有的section(一个配置文件中可以有多个配置，如数据库相关的配置，邮箱相关的配置，每个section由[]包裹，即[section])，并以列表的形式返回
secs=[sec for sec in secs if sec not in missHosts]
print("{}==>开始初始化所有数据库服务。。。。。。。。".format(time.strftime("%Y-%m-%d %H:%M:%S",time.localtime())))
db_success = {}
#先初始化数据库操作
for sec in secs:
host = cf.get(sec, "host")
port = cf.get(sec, "port")
user = cf.get(sec, "user")
pwd = cf.get(sec, "pwd")
ssh = getConnection(host,port,user,pwd)
if ssh:
db_status = initDatabse(sec,ssh)
db_success.setdefault(sec,db_status)
else:
print("【error】{}主机连接失败！".format(sec))
closeConnection(ssh)
#如果数据库都初始化成功，则开始执行启动指令
if False not in db_success.values():
print("{}==>所有数据库服务初始化完成！！！".format(time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())))
for sec in secs:
host = cf.get(sec, "host")
port = cf.get(sec, "port")
user = cf.get(sec, "user")
pwd = cf.get(sec, "pwd")
ssh = getConnection(host, port, user, pwd)
if ssh:
startProject(sec,ssh)
else:
print("【error】{}主机连接失败！".format(sec))
closeConnection(ssh)
else:
print("{}==>数据库没有初始化成功,请检查后在尝试！",format("|".join([k for k,v in db_success.items() if v==False])))
#启动项目
def startProject(secname,ssh):
flag = {}
if assertKey(secname,"command"):
print("启动项目操作：【{}】".format(secname))
dbs = ast.literal_eval(cf.get(secname,"command"))
for k,v in dbs.items():
success = assertServiceStatus(ssh,k)
if success:
print("=====[{}]项目服务已经启动,指令=[{}]".format(secname,k))
flag.setdefault(k,True)
else:
print("=====[{}]项目服务正在启动,指令=[{}]".format(secname, v))
executeService(ssh,v)
#再次验证是否都已启动
time.sleep(1)
for k,v in dbs.items():
success = assertServiceStatus(ssh,k)
if success:
flag.setdefault(k,True)
else:
print("=====[{}]主机中[{}]启动失败！".format(secname, v))
#尝试重启一次，返回True 或False ，并会打印执行信息
flag.setdefault(k,restartProject(ssh,k,v))
if False not in flag.values():
print("=====[{}]所有项目完成！".format(secname))
return True
else:
print("【error】=====[{}]项目启动失败！".format(secname))
return False
#初始化所有数据库
def initDatabse(secname,ssh):
flag = {}
if assertKey(secname,"database"):
print("数据库操作：【{}】:".format(secname))
dbs = ast.literal_eval(cf.get(secname,"database"))
for k,v in dbs.items():
success = assertServiceStatus(ssh,k)
if success:
print("=====[{}]数据库服务已经启动,指令=[{}]".format(secname,k))
flag.setdefault(k,True)
else:
print("=====[{}]数据库服务正在启动,指令=[{}]".format(secname, v))
executeService(ssh,v)
#再次验证程序是否都已启动
time.sleep(2)
for k,v in dbs.items():
success = assertServiceStatus(ssh,k)
if success:
flag.setdefault(k,True)
else:
print("=====[{}]主机中[{}]数据库服务初始化失败！".format(secname, v))
flag.setdefault(k,False)
if False not in flag.values():
print("=====[{}]初始化数据库完成！".format(secname))
return True
else:
print("【error】=====[{}]初始化数据库失败！".format(secname))
return False
#针对启动失败的程序，tomcat启动失败，但是ps查询还是会有进程
def restartProject(ssh,k,v):
pids=[]
stdin, stdout, stderr = ssh.exec_command(k,get_pty=True)#查询进程
lines = stdout.readlines()
for line in lines:
pid = line[9:14]
pids.append(pid)
print("进程=[{}],info=【{}】".format(pid,line))
if pids:
ssh.exec_command("kill -9 {}".format(" ".join(pids)))
#杀完进程后尝试重新启动
stdin, stdout, stderr = ssh.exec_command(v,get_pty=True)
for out in stdout:
print("执行返回信息:【{}】".format(out))
for err in stderr:
print("执行错误信息:【{}】".format(err))
time.sleep(3)
success = assertServiceStatus(ssh,k)
return success
#判断服务是否启动
def assertServiceStatus(ssh,command):
stdin, stdout, stderr = ssh.exec_command(command,get_pty=True)
res = [line for line in stdout]
if len(res) > 0:
return True
else:
return False
#判断服务是否启动
def executeService(ssh,command):
try:
ssh.exec_command("{}".format(command),timeout=2)#不等待执行结果
except:
print("{}==>执行完毕！")
def assertKey(secname,key):
has =cf.has_option(secname,key)
return has
def getConnection(host,port,user,pwd):
'''
重新获取链接
:param host:
:param port:
:param user:
:param pwd:
:return:
'''
ssh = paramiko.SSHClient() # 创建SSH对象
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy()) # 允许连接不在know_hosts文件中的主机
try:
ssh.connect(hostname=host, port=port, username=user, password=pwd,timeout=5) # 连接服务器
return ssh
except:
print("【error】{}连接失败".format(host))
return None
def closeConnection(ssh):
'''
关闭链接
:param ssh:
:return:
'''
try:
ssh.close()
except:
print("【error】{}:链接关闭异常".format(time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())))
if __name__ == '__main__':
start = time.time()
operation()
end = time.time()
print("所有项目启动完成，共耗时{}秒".format((end-start)))
# sec="data2"
# host = cf.get(sec, "host")
# port = cf.get(sec, "port")
# user = cf.get(sec, "user")
# pwd = cf.get(sec, "pwd")
# ssh = getConnection(host, port, user, pwd)
# k = "service mysqld status"
# v = "cd /var/neo4j-community-3.5.1/bin;./neo4j start"
# #restartProject(ssh,k,v)
# stdin, stdout, stderr = ssh.exec_command(k, get_pty=True) # 查询进程
#
# if str(stdout.readlines()[-1]).__contains__("Started MySQL Server."):
# print("服务已经启动！")