import paramiko
import _thread
import xlrd
import time
import json

dispatcher_conf = open("dispatcher.json", encoding='utf-8')
dispatcher = json.load(dispatcher_conf)

# 编译后的可执行文件，进程名
localfile_bin = dispatcher["localfile_bin"]
# 需要分发的配套配置文件
localpath_conf = dispatcher["localpath_conf"]
position_conf = dispatcher["position_conf"]
# 机器ip信息，最好是内网地址
ipFile = dispatcher["ipFile"]
# 每台机子配置账号数量，账号起始位置为0
usernum_for_perip = dispatcher["usernum_for_perip"]

hostsuc = set()
hosterr = set()


def sendFile(hostname, username, password):
    scp = paramiko.Transport((hostname))
    scp.connect(username=username, password=password)
    sftp = paramiko.SFTPClient.from_transport(scp)
    print(hostname + "登陆成功,正在分发程序...")
    sftp.put(localfile_bin, localfile_bin)
    sftp.put(localpath_conf, localpath_conf)
    files = sftp.listdir()
    for f in files:
        if f == localfile_bin:
            print(hostname + "部署成功...")
            break
    scp.close()


def start(hostname, username, password, offset):
    ssh = paramiko.SSHClient()
    ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
    ssh.connect(hostname, username=username,
                password=password, look_for_keys=False)
    print(hostname + "登陆成功,正启动...")
    exec_command(ssh, "echo \""+str(list(range(offset, offset +
                                               usernum_for_perip)))+"\" > " + position_conf)
    exec_command(ssh, "chmod +x " + localfile_bin)
    ssh.exec_command("./" + localfile_bin + " & >/dev/null")
    time.sleep(1)
    psinfo = exec_command(ssh, "ps -e|grep " + localfile_bin)
    if (psinfo.decode('utf-8').find(localfile_bin) != -1):
        hostsuc.add(hostname)
        print(hostname + ":已启动")
    else:
        hosterr.add(hostname)
        print(hostname + ":启动失败")
    ssh.close()


def stopandclean(hostname, username, password):
    ssh = paramiko.SSHClient()
    ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
    ssh.connect(hostname, username=username,
                password=password, look_for_keys=False)
    print(hostname + "登陆成功,正在执行清理...")
    exec_command(ssh, "ps -e|grep "+localfile_bin +
                 " |awk '{print $1}' |sort -r|xargs kill -9")
    exec_command(ssh, "rm -rf logs;mkdir logs;rm -rf captchas;mkdir captchas")
    exec_command(ssh, "rm -f "+localfile_bin)
    print(hostname + "登陆成功,正在清理完成...")
    ssh.close()


def exec_command(ssh, command):
    stdin, stdout, stderr = ssh.exec_command(command)
    return stdout.read()


def dispacth(hostname, username, password, offset):
    try:
        stopandclean(hostname, username, password)
        sendFile(hostname, username, password)
        start(hostname, username, password, offset)
    except (TimeoutError, IOError) as err:
        hosterr.add(hostname)
        print(hostname + "部署异常...", err)


wb = xlrd.open_workbook(filename=ipFile)
sheet = wb.sheet_by_index(0)
for i in range(0, sheet.nrows):
    hostname = sheet.cell_value(i, 0)
    username = sheet.cell_value(i, 1)
    password = sheet.cell_value(i, 2)
    _thread.start_new_thread(
        dispacth, (hostname, username, password, i * usernum_for_perip + 1, ))

while((len(hostsuc) + len(hosterr)) != sheet.nrows):
    time.sleep(1)

if len(hosterr) > 0:
    print("部署完成，但"+str(hosterr)+"运行失败")
else:
    print("部署完成，全部正常运行")