#!/usr/bin/python3
from netmiko.huawei.huawei import HuaweiSSH
import os
from datetime import datetime
import smtplib
from email.mime.text import MIMEText
from email.header import Header
from email.mime.multipart import MIMEMultipart

CMD_STR = 'cmd_str'
CMD_NAME = 'name'
IP = 'ip'
USER_NAME = 'username'
PASSWORD = 'password'

# email config
smtp_host = '你的郵箱發(fā)送付服務地址'
smtp_port = 25
smtp_email = '你的發(fā)送郵箱號'
# 用戶名，默認為發(fā)件人郵箱前綴
smtp_user ='你的發(fā)送郵箱號'
# 密碼（注意，某些郵箱需要為SMTP服務單獨設置授權碼，詳情查看相關幫助）
smtp_password = '你的發(fā)送郵箱密碼'

class ServerMaintenance:
    net_connect = None
    curr_dir = os.getcwd()
    fo = None
    file_name = None
    email_files = []

    # 建立服務器連接
    def Connect_HuaweiSSH(self, _ip, _user_name, _pw):
        try:
            S5720 = {
                'device_type': 'huawei',
                'ip': _ip,
                'username': _user_name,
                'password': _pw,
            }
            # 實例化HuaweiSSH
            net_connect = HuaweiSSH(**S5720)
            self.net_connect = net_connect
        except:
            print("連接" + _ip + "錯誤")

    # 運行指令
    def execute(self, cmd_str, cmd_name):
        result = self.net_connect.send_command(cmd_str)
        print(result)
        self.printFile('\n-------------------------任務【' + cmd_name + '】開始巡查 -------------------------\n')
        self.printFile(result)
        self.printFile('\n-------------------------任務【' + cmd_name + '】結束巡查 -------------------------\n')
        return result

    # 關閉連接
    def disconnect(self):
        self.net_connect.disconnect()
        print("連接斷開" + self.net_connect)

    # 文件寫入
    def printFile(self, txt):
        self.fo.write(txt)
    # 發(fā)送郵箱
    def sendEmail(self):
        msg = MIMEMultipart()
        msg['From'] = Header("master", 'utf-8')
        msg['To'] = Header('13693567027@163.com', 'utf-8')
        subject = 'Test Email'
        msg['Subject'] = Header(subject, 'utf-8')
        # 郵件正文內(nèi)容
        content = ''


        for filename in self.email_files:
            content+=filename+'\n'
            # 郵件附件
            att = MIMEText(open(filename+'.txt', 'rb').read(), 'base64', 'utf-8')
            att['Content-Type'] = 'application/octet-stream'  # 指定文件格式
            att["Content-Disposition"] = 'attachment; filename="服務器巡檢報告.txt"'
            msg.attach(att)

        msg.attach(MIMEText(content, 'plain', 'utf-8'))
        try:
            # # 創(chuàng)建安全連接，使用SMTP_SSL連接SMTP服務器
            smtp_client = smtplib.SMTP(smtp_host)  # 實例化對象
            smtp_client.connect(smtp_host, smtp_port)  # 連接163郵箱服務器，端口號為25
            # # 向服務器標識用戶身份
            smtp_client.login(smtp_user, smtp_password)
            # 發(fā)送郵件
            smtp_client.sendmail(smtp_user, '13693567027@163.com', msg.as_string())
            print("郵件發(fā)送成功")
        except smtplib.SMTPException as e:
            print("Error: 無法發(fā)送郵件 ", e)

        # 關閉SMTP連接
        smtp_client.quit()

    def run(self,commands,hosts):
        for host in hosts:
            self.file_name = "服務器巡查_" + host[IP] + '_' + datetime.now().strftime(format="%Y年%m月%d日%H時%M分%S秒")
            self.email_files.append(self.file_name)
            self.fo = open(self.file_name + ".txt", 'w+', encoding='utf-8')
            self.Connect_HuaweiSSH(host[IP], host[USER_NAME], host[PASSWORD])
            self.printFile(
                '\n================================主機開始巡檢： ' + host[IP] + '================================\n')
            for command in commands:
                self.execute(command[CMD_STR], command[CMD_NAME])
            self.printFile(
                '\n================================主機巡檢結束： ' + host[IP] + '================================\n')



commands = [
    {CMD_STR: "ps -ef |grep java", CMD_NAME: "運行中的應用"},
    {CMD_STR: "docker ps", CMD_NAME: "運行中的容器"},
    {CMD_STR: "df -TH", CMD_NAME: "磁盤空間"},
    {CMD_STR: "free -h", CMD_NAME: "內(nèi)存"},
    {CMD_STR: "ps -ef", CMD_NAME: "進程"},
    {CMD_STR: "crontab  -l", CMD_NAME: "定時任務"},
    {CMD_STR: "cat /var/log/messages|grep ERROR && dmesg |grep ERROR", CMD_NAME: "操作系統(tǒng)日志"},
    {CMD_STR: "date", CMD_NAME: "操作系統(tǒng)時間"},
    {CMD_STR: "firewall-cmd --list-all", CMD_NAME: "防火墻"},
    {CMD_STR: "dmidecode -t system", CMD_NAME: "服務器型號"},
    {CMD_STR: "uname -a && uname -r", CMD_NAME: "操作系統(tǒng)與內(nèi)核"},
    {CMD_STR: "last", CMD_NAME: "用戶登錄日志"},
]

hosts = [
    {IP: '你的服務器ip', USER_NAME: '你的服務器賬號', PASSWORD: '你的服務器密碼'}
]

serverMain = ServerMaintenance()
serverMain.run(commands,hosts)
serverMain.sendEmail()

# -*- coding: utf-8 -*-
"""
linux 自動化腳本
# @Time: 2022/11/4 10:20
# @Author: lln
# @File: linuxOpsStartUp.py
"""
import json
import os
import platform
import time


def runCommand(command):
    """
    執(zhí)行命令，將所有讀到的數(shù)據(jù)去除空行
    :param command: 命令
    :return: 去除空行后的命令
    """
    lines = os.popen(command).readlines()
    res = []
    for line in lines:
        res.append(line.replace('\n', ''))
    return res


def getSystemInfo():
    """
    使用內(nèi)置庫獲取系統(tǒng)信息
    """
    res = {
        "操作系統(tǒng)名稱及版本號": platform.platform(),
        "操作系統(tǒng)版本號": platform.version(),
        "操作系統(tǒng)的位數(shù)": platform.architecture(),
        "計算機類型": platform.machine(),
        "網(wǎng)絡名稱": platform.node(),
        "處理器信息": platform.processor(),
    }
    return res


def getSystemStatus():
    """
    系統(tǒng)信息，僅支持centos進行查詢
    """
    # 系統(tǒng)
    OS = runCommand("uname -o")
    # 發(fā)行版本
    Release = runCommand("cat /etc/redhat-release 2>/dev/null")
    # 內(nèi)核
    Kernel = runCommand("uname -r")
    # 主機名
    Hostname = runCommand("uname -n")
    # 當前時間
    LocalTime = runCommand("date +'%F %T'")
    # 最后啟動
    LastReboot = runCommand("who -b | awk '{print $3,$4}'")
    # 運行時間
    Uptime = runCommand("date +'%F %T'")
    # 當前時區(qū)信息
    time_zone = runCommand("date -R")
    res = {
        "系統(tǒng)": OS,
        "發(fā)行版本": Release,
        "內(nèi)核": Kernel,
        "主機名": Hostname,
        "當前時間": LocalTime,
        "最后啟動": LastReboot,
        "運行時間": Uptime,
        "時區(qū)信息": time_zone
    }
    return res


def getCpuStatus():
    """
    CPU信息
    """
    # 物理CPU個數(shù)
    physical_cpus = runCommand("grep 'physical id' /proc/cpuinfo| sort | uniq | wc -l")
    # 邏輯CPU個數(shù)
    virt_cpus = runCommand("grep 'processor' /proc/cpuinfo | wc -l")
    # 每CPU核心數(shù)
    cpu_kernels = runCommand("grep 'cores' /proc/cpuinfo|uniq| awk -F ': ' '{print $2}'")
    # CPU型號
    cpu_type = runCommand("grep 'model name' /proc/cpuinfo | awk -F ': ' '{print $2}' | sort | uniq")
    # CPU架構
    cpu_arch = runCommand("uname -m")
    res = {
        '物理CPU個數(shù)': physical_cpus,
        '邏輯CPU個數(shù)': virt_cpus,
        '每CPU核心數(shù)': cpu_kernels,
        'CPU型號': cpu_type,
        'CPU架構': cpu_arch
    }
    return res


def getMemStatus():
    """
    內(nèi)存信息
    """
    # 總內(nèi)存
    MemTotal = runCommand("grep MemTotal /proc/meminfo| awk '{print $2}'")
    MemTotal_Num = map(float, MemTotal)[0]
    # 可用內(nèi)存
    MemFree = runCommand("grep MemFree /proc/meminfo| awk '{print $2}'")
    MemFree_Num = map(float, MemFree)[0]
    # 比例
    Proportion = '{:.4%}'.format(MemFree_Num / MemTotal_Num)
    res = {
        '總內(nèi)存(GB)': '{:.5}'.format(float(MemTotal_Num / 1024 / 1024)),
        '可用內(nèi)存(GB)': '{:.5}'.format(float(MemFree_Num / 1024 / 1024)),
        '已用比例(%)': Proportion
    }
    return res


def getMemStatusSimple():
    MemTotal = runCommand("free -h")
    res = {
        '內(nèi)存總覽': MemTotal
    }
    return res


def getDiskStatus():
    """
    磁盤檢查
    """
    # 生成臨時數(shù)據(jù)記錄文件
    # os.popen("df -TP | sed '1d' | awk '$2!='tmpfs'{print}'")
    # os.popen("df -hTP | sed 's/Mounted on/Mounted/'> /tmp/disk")
    # 硬盤總量
    DiskAllInfo = runCommand("df -h | grep -v docker")
    DiskTotal = runCommand("df -TP | sed '1d' | awk '$2!='tmpfs'{print}'| awk '{total+=$3}END{print total}'")
    DiskTotalNum = int(DiskTotal[0])
    # 硬盤使用量
    DiskUsed = runCommand("df -TP | sed '1d' | awk '$2!='tmpfs'{print}'| awk '{total+=$4}END{print total}'")
    DiskUsedNum = int(DiskUsed[0])
    # 硬盤空余量
    DiskFree = DiskTotalNum - DiskUsedNum
    # 硬盤使用比例
    DiskUsedPercent = '{:.2%}'.format(DiskUsedNum / DiskTotalNum)
    # 索引總量
    InodeTotal = runCommand("df -iTP | sed '1d' | awk '$2!='tmpfs'{print}' | awk '{total+=$3}END{print total}' ")
    InodeTotal_Num = int(InodeTotal[0])
    # 索引使用量
    InodeUsed = runCommand("df -iTP | sed '1d' | awk '$2!='tmpfs'{print}' | awk '{total+=$4}END{print total}' ")
    InodeUsed_Num = int(InodeUsed[0])
    # 索引剩余量
    InodeFree = InodeTotal_Num - InodeUsed_Num
    # 索引使用比例
    InodePercent = '{:.2%}'.format(InodeUsed_Num / InodeTotal_Num)
    res = {
        '磁盤總覽': DiskAllInfo,
        '硬盤總量(GB)': int(DiskTotalNum / 1024 / 1024),
        '硬盤使用量(GB)': int(DiskUsedNum / 1024 / 1024),
        '硬盤空余量(GB)': int(DiskFree / 1024 / 1024),
        '硬盤使用比例(%)': DiskUsedPercent,
        '索引總量(MB)': int(InodeTotal_Num / 1021),
        '索引使用量(MB)': int(InodeUsed_Num / 1021),
        '索引剩余量(MB)': int(InodeFree / 1021),
        '索引使用比例(%)': InodePercent,
    }
    return res


def getNetworkStatus():
    """
    網(wǎng)絡檢查
    """
    GATEWAY = runCommand("ip route | grep default | awk '{print $3}'")
    DNS = runCommand("grep nameserver /etc/resolv.conf| grep -v '#' | awk '{print $2}' | tr '\n' ',' | sed 's/,$//'")
    IP = runCommand(
        "ip -f inet addr | grep -v 127.0.0.1 | grep inet | awk '{print $NF,$2}' | tr '\n' ',' | sed 's/,$//'")
    # TODO 語句有問題會報錯，sed的錯誤，需要檢查下執(zhí)行情況
    # MAC = runCommand("ip link | grep -v 'LOOPBACK\|loopback' | awk '{print $2}' | sed 'N;s/\n//' | tr '\n' ',' | sed 's/,$//'")
    res = {
        'GATEWAY': GATEWAY,
        'DNS': DNS,
        'IP': IP
        # 'MAC': MAC
    }
    return res


def getUserStatus():
    """
    所有用戶和空密碼用戶
    """
    all_user = runCommand("awk -F':' '{ print $1}' /etc/passwd")
    empty_passwd_user = runCommand("getent shadow | grep -Po '^[^:]*(?=::)'")
    res = {
        '所有用戶名': all_user,
        '空密碼用戶': empty_passwd_user
    }
    return res


def getJdkStatus():
    """
    jdk信息
    """
    jdkInfo = runCommand("java -version 2>&1")
    res = {
        'jdk信息': jdkInfo
    }
    return res


def getFirewallStatus():
    """
    防火墻
    """
    firewall = runCommand("firewall-cmd --state 2>&1")
    # 兼容 ubuntu 防火墻命令報錯 sh: not found 特殊處理
    for info in firewall:
        if "not found" in info:
            firewall = runCommand("ufw status")
    res = {
        '防火墻狀態(tài)': firewall
    }
    return res


def sshStatus():
    """
    ssh 檢查
    """
    sshActive = runCommand("systemctl is-active sshd.service")
    sshNetstat = runCommand("sudo netstat -atlunp | grep sshd")
    res = {
        'ssh開啟狀態(tài)': sshActive,
        'ssh運行情況': sshNetstat
    }
    return res


def ntpStatus():
    """
    ntp 檢查
    """
    ntpActive = runCommand("systemctl is-active ntpd")
    res = {
        'ntp運行情況': ntpActive
    }
    return res


def dockerStatus():
    """
    docker 檢查
    """
    dk_version = runCommand("docker -v")
    dk_stats = []
    for info in dk_version:
        if "version" not in info:
            dk_version = "未安裝docker"
        else:
            lines = os.popen(
                "docker stats --all --no-stream").readlines()
            for line in lines:
                dk_stats.append(line.replace('\n', ''))
    dp_version = runCommand("docker-compose --version")
    for info in dp_version:
        if "version" not in info:
            dp_version = "未安裝docker-compose"
    res = {
        'docker version': dk_version,
        'docker-compose version': dp_version,
        'docker stats': dk_stats
    }
    return res


def createReportFile(name, text):
    """
    創(chuàng)建report的txt文件,并寫入數(shù)據(jù)
    """
    report_dir = os.getcwd() + os.sep + "report" + os.sep
    # 判斷當前路徑是否存在，沒有則創(chuàng)建new文件夾
    if not os.path.exists(report_dir):
        os.makedirs(report_dir)
    # 在當前py文件所在路徑下的new文件中創(chuàng)建txt
    report_file = report_dir + name + '.txt'
    # 打開文件，open()函數(shù)用于打開一個文件，創(chuàng)建一個file對象，相關的方法才可以調(diào)用它進行讀寫。
    file = open(report_file, 'w')
    # 寫入內(nèi)容信息
    file.write(text)
    file.close()
    print('report_file create success', report_file)


def printSinfcloud():
    print("+------------------------------------------------+")
    print("|       歡迎使用SinfCloud自動巡檢工具            |")
    print("| ____  _        __  ____ _                 _    |")
    print("|/ ___|(_)_ __  / _|/ ___| | ___  _   _  __| |   |")
    print("|\___ \| |  _ \| |_| |   | |/ _ \| | | |/ _  |   |")
    print("| ___) | | | | |  _| |___| | (_) | |_| | (_| |   |")
    print("||____/|_|_| |_|_|  \____|_|\___/ \__,_|\__,_|   |")
    print("|                                                |")
    print("+------------------------------------------------+")


if __name__ == '__main__':
    printSinfcloud()
    outputFileName = time.strftime('%Y-%m-%d', time.localtime(time.time())) + "_report"
    report = list()
    report.append(getSystemInfo())
    report.append(getSystemStatus())
    report.append(getCpuStatus())
    report.append(getMemStatusSimple())
    report.append(getDiskStatus())
    report.append(getNetworkStatus())
    report.append(getUserStatus())
    report.append(getJdkStatus())
    report.append(getFirewallStatus())
    report.append(sshStatus())
    report.append(ntpStatus())
    report.append(dockerStatus())
    createReportFile(outputFileName,
                     json.dumps(report, sort_keys=True, indent=4, separators=(',', ':'), ensure_ascii=False))