# coding=utf-8

import threading
import paramiko
import os
import time
import xlrd
import xlwt
import openpyxl
import logging


all_row = []
threads = []

class Logger:
    def __init__(self,loggername):

        #创建一个logger
        self.logger = logging.getLogger(loggername)
        self.logger.setLevel(logging.DEBUG)

        cwd = os.getcwd()
        print("cwd:", cwd)
        #创建一个handler，用于写入日志文件
        log_path = os.path.dirname(os.getcwd())+"excel\log" # 指定文件输出路径，注意logs是个文件夹，一定要加上/，不然会导致输出路径错误，把logs变成文件名的一部分了
        print("log_path:", log_path)
        logname = log_path + '\out.log' #指定输出的日志文件名
        print("logname: {0}",format(logname))
        fh = logging.FileHandler(logname,encoding = 'utf-8')  # 指定utf-8格式编码，避免输出的日志文本乱码
        fh.setLevel(logging.DEBUG)

        #创建一个handler，用于将日志输出到控制台
        ch = logging.StreamHandler()
        ch.setLevel(logging.DEBUG)

        # 定义handler的输出格式
        formatter = logging.Formatter('%(asctime)s-%(name)s-%(levelname)s-%(message)s')
        fh.setFormatter(formatter)
        ch.setFormatter(formatter)

        # 给logger添加handler
        self.logger.addHandler(fh)
        self.logger.addHandler(ch)


    def get_log(self):
        """定义一个函数，回调logger实例"""
        return self.logger


class read_excel(object):
    def __init__(self, num):
        #threading.Thread.__init__(self)
        #self.threadID = threadID
        self.num = num
    def read_excel(self):
        filePath = os.path.join(os.getcwd(), 'D:\浙商银行\IP.xlsx')

        wb = openpyxl.load_workbook(filePath)  # 打开excel文件
        ws = wb['Sheet1']
        res = ws.cell(row=1, column=1).value  # 获取单元格的内容
        # print(res)
        # print(wb.sheetnames)                    # 显示excel文件的所有sheet名字
        print(ws.min_row, ws.max_row, ws.max_column, ws.min_column)  # 单元格的行数、列数; max_row:最大行、max_column:最大列
        print('*******************************************************')
        #all_row = []
        for row in ws.rows:  # 按行获取单元格(Cell对象) - 生成器
            ev_row = []
            for cell in row:
                value = cell.value
                value.strip()
                print("value.strip():{0}".format(value.strip()))
                ev_row.append(value.strip())
            all_row.append(ev_row)
            #print("ev_row:", ev_row)
            #print('*******************************************************')
            #print(all_row)
            #print(len(all_row))
        #print("all_row[0][0]:", all_row[0][0])

class acquisition_of_information(read_excel, threading.Thread):

    def __init__(self, num, threadID, name):
        self.logger = Logger(__name__)
        #super().__init__(self, num)
        threading.Thread.__init__(self)
        self.threadID = threadID
        self.num = num
        print("threadID:{0} name:{1}".format(threadID, name))
    def run(self):
        # 创建SSHClient实例对象
        ssh = paramiko.SSHClient()
        # 调用方法，标识没有远程机器的公钥，允许访问
        ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
        len1 = len(all_row)
        print("len1:", len1)
        # 连接远程机器 地址端口用户名密码
        for num in range(0, len1):
            fo = open(all_row[num][0] + ".log", "wb+")
            fo.truncate()  # 清空文件内容
            all_row[num][0].strip()   #  去除excel表格内容头和尾的空格
            print("all_row{0}[0]:{1}".format(num, all_row[num][0].strip()))
            all_row[num][1].strip()   #  去除excel表格内容头和尾的空格
            print("all_row{0}[1]:{1}".format(num, all_row[num][1].strip()))
            all_row[num][2].strip()   #  去除excel表格内容头和尾的空格
            print("all_row{0}[2]:{1}".format(num, all_row[num][2].strip()))
            try:
                ssh.connect(all_row[num][0].strip(), 22, all_row[num][1].strip(), all_row[num][2].strip())
            except IOError:
                #print   ("连接服务器失败,IP：{0}".format(all_row[num][0]))
                self.logger.get_log().debug("连接服务器失败,IP：{0}".format(all_row[num][0]))
            else:
                #print   ("连接服务器成功，IP：{0}".format(all_row[num][0]))
                self.logger.get_log().debug("连接服务器成功，IP：{0}".format(all_row[num][0]))
            print("num:", num)
            # 创建目录
            stdin, stdout, stderr = ssh.exec_command("ip a | grep 192 | awk '{print $2}'")
            fo.write(stdout.read())
            str = "***************************" \
                  "***************************\n"
            str = str.encode()
            fo.write(str)
            stdin, stdout, stderr = ssh.exec_command("uname -a | grep -i 'linux' | wc -l")
            res = stdout.read()
            print("res:", res)
            if res == b'1\n':
                str = "df -Ph\n"
                str = str.encode()
                fo.write(str)
                stdin, stdout, stderr = ssh.exec_command("df -Ph")
                fo.write(stdout.read())
                str = "***************************" \
                      "***************************\n"
                str = str.encode()
                fo.write(str)
            else:
                str = "df -Pg\n"
                str = str.encode()
                fo.write(str)
                stdin, stdout, stderr = ssh.exec_command("df -Pg")
                fo.write(stdout.read())
                str = "***************************" \
                      "***************************\n"
                str = str.encode()
                fo.write(str)
            str = "cat .profile\n"
            str = str.encode()
            fo.write(str)
            stdin, stdout, stderr = ssh.exec_command("cd /root && cat .profile")
            res, err = stdout.read(), stderr.read()
            fo.write(res)
            str = "***************************" \
                  "***************************\n"
            str = str.encode()
            fo.write(str)
            str = "arp -a\n"
            str = str.encode()
            fo.write(str)
            stdin, stdout, stderr = ssh.exec_command("arp -a")
            res, err = stdout.read(), stderr.read()
            fo.write(res)
            str = "***************************" \
                  "***************************\n"
            str = str.encode()
            fo.write(str)

            str = "netstat -rn\n"
            str = str.encode()
            fo.write(str)
            stdin, stdout, stderr = ssh.exec_command("netstat -rn")
            res, err = stdout.read(), stderr.read()
            fo.write(res)
            str = "***************************" \
                  "***************************\n"
            str = str.encode()
            fo.write(str)

            str = "cat /etc/rc.local\n"
            str = str.encode()
            fo.write(str)
            stdin, stdout, stderr = ssh.exec_command("cat /etc/rc.local")
            res, err = stdout.read(), stderr.read()
            fo.write(res)
            str = "***************************" \
                  "***************************\n"
            str = str.encode()
            fo.write(str)

            str = "cat /etc/hosts\n"
            str = str.encode()
            fo.write(str)
            stdin, stdout, stderr = ssh.exec_command("cat /etc/hosts")
            res, err = stdout.read(), stderr.read()
            fo.write(res)
            str = "***************************" \
                  "***************************\n"
            str = str.encode()
            fo.write(str)

            str = "netstat -an|grep LISTEN*\n"
            str = str.encode()
            fo.write(str)
            stdin, stdout, stderr = ssh.exec_command("netstat -an|grep LISTEN*")
            res, err = stdout.read(), stderr.read()
            fo.write(res)
            str = "***************************" \
                  "***************************\n"
            str = str.encode()
            fo.write(str)

            str = "ps -ef | grep java\n"
            str = str.encode()
            fo.write(str)
            stdin, stdout, stderr = ssh.exec_command("ps -ef | grep java")
            res, err = stdout.read(), stderr.read()
            fo.write(res)
            str = "***************************" \
                  "***************************\n"
            str = str.encode()
            fo.write(str)

            str = "ps -ef | grep zabbix\n"
            str = str.encode()
            fo.write(str)
            stdin, stdout, stderr = ssh.exec_command("ps -ef | grep zabbix")
            res, err = stdout.read(), stderr.read()
            fo.write(res)
            str = "***************************" \
                  "***************************\n"
            str = str.encode()
            fo.write(str)

            str = "ps -ef | grep ctm\n"
            str = str.encode()
            fo.write(str)
            stdin, stdout, stderr = ssh.exec_command("ps -ef | grep ctm")
            res, err = stdout.read(), stderr.read()
            fo.write(res)
            str = "***************************" \
                  "***************************\n"
            str = str.encode()
            fo.write(str)
            str = "crontab -l\n"
            str = str.encode()
            fo.write(str)
            stdin, stdout, stderr = ssh.exec_command("crontab -l")
            res, err = stdout.read(), stderr.read()
            fo.write(res)
            str = "--------------------------" \
                  "***************************" \
                  "***************************" \
                  "---------------------------\n"
            str = str.encode()
            fo.write(str)
            ssh.close()
            fo.close()
            self.logger.get_log().debug("服务器信息获取完成,IP：{0}".format(all_row[num][0]))
            num = num + 2

    def __del___(self, num):
        self.num = num


class print_all_row(read_excel, threading.Thread):
    def __init__(self, num, threadID, name):
        threading.Thread.__init__(self)
        self.threadID = threadID
        self.num = num
        print("print_all_row name:", name)

    def run(self):
        print("all_row[0][0]:", all_row[0][0])
        print("all_row[0][1]:", all_row[0][1])
        print("all_row[0][2]:", all_row[0][2])

def main():
    t = Logger("hmk").get_log().debug("User %s is loging" % 'jeck')
    start = time.time()
    excel = read_excel(0)
    excel.read_excel()
    thread1 = acquisition_of_information(0, "Thread-2", "zhouhaiwu")
    thread2 = acquisition_of_information(1, "Thread-3", "zhouhaiwu")
    thread3 = print_all_row(0, "Thread-4", "zhouhaiwu")
    # 开启新线程
    #thread1.start()
    thread1.start()
    thread2.start()
    thread3.start()
    # 添加线程到线程列表
    #threads.append(thread1)
    threads.append(thread1)
    threads.append(thread2)
    threads.append(thread3)
    # 等待所有线程完成
    for t in threads:
        t.join()
    print   ("Exiting Main Thread")
    end = time.time()
    print (str(end))

if __name__ == '__main__':
    main()