#!/usr/bin/env python

#encoding:utf8

# desc: self-inspection

# args:

#     reboot :  reboot AP

#     check  :  check AP syslog

#     getApNum: return AP Number Online

#     checkApOnline: ok: return ok; false: return ap ip : local apInfo 读取这个文件开始测试是否在线

import os

import sys

from paramikoHelper import myParamiko

from time import sleep

from telnetAP import rebTelnet

import threading

# AP IP信息文件

apInfoFile='/opt/script/apInfo'

# 全局线程锁

lock = threading.RLock()

# 多线程计数器

class Counter:

    count = 0

    def __init__(self, count):

        self.count = count

    def add(self, n = 1):

        lock.acquire()

        self.count = self.count + n

        lock.release()

    def delete(self, n = 1):

        lock.acquire()

        self.count = self.count - n

        lock.release()

    def get(self):

        return self.count

# 读取命令执行的结果

def getRunCommRes(cmd):

    return os.popen(cmd).read().strip('\n').strip(' ')

# 获取所有AP的IP地址列表

def getAllIP():

    return open(apInfoFile).read().split('\n')[:-1]

# 通用工具类

# 规定:

#    定义普通方法的时候,必须定义两个默认且两个参数 arg1 = "", arg2 = ""

#    定义被threadBox调用的target方法的时候,

#         要满足4个参数,且第一个参数为ip(threadBox定义的),后面如果没有指定参数,需要用默认参数 arg2 = "", arg3 = "", arg4 = ""

#         只有一个参数IP时: def oTarget(ip, arg2 = "", arg3 = "", arg4 = ""):

class CommonUtil:

    def __init__(self):

        pass

    # 多线程执行

    def threadBox(slef, myTarget, arg1 = "", arg2 = "", arg3 = ""):

        threads = []

        for ip in getAllIP():

            t = threading.Thread(target=myTarget, args=(ip, arg1, arg2, arg3))

            threads.append(t)

            t.start()

        for t in threads:

            t.join()

    # 获取设备在线时长

    def uptime(self, arg1 = "", arg2 = ""):

        print getRunCommRes("uptime | cut -d ',' -f1 | awk -F'up' '{print $2}'")

    # 获取设备网络配置

    def netInfo(self, arg1 = "", arg2 = ""):

        ip = getRunCommRes("ip a show eth0 | grep inet | grep eth0 | awk '{print $2}'")

        gateway = getRunCommRes("ip route  | grep default | awk '{print $3}'")

        print ip,gateway

    # 获取设备联盟号

    def getPortalId(self, arg1 = "", arg2 = ""):

        print getRunCommRes('cat /usr/local/etc/chilli/wid')

    # 运行指定的脚本文件

    def runScript(self, arg1, arg2 = ""):

        print getRunCommRes('/opt/script/' + arg1)

    # ping Target

    def pingTarget(self, ip, res, downCount, args4 = ""):

        code = os.system('/bin/ping -c 1 -w 3 ' + ip + '>/dev/null')

        if code != 0:

            code1 = os.system('/bin/ping -c 1 -w 3 ' + ip + '>/dev/null')

            if code1 != 0:

                res.append(ip.split('.')[-1])

                downCount.add() 

    # 获取AP在线数

    # :::ex251::: 判断是否还有192.168.188.251的IP地址可以ping通

    #             如果可以ping通表示还有AP没有修改IP地址

    # a= 表示一共多少台AP, d=表示有多少台掉线, ::后面的数字表示192.168.188.+数字表示AP的ip地址

    #

    def checkApOnline(self, arg1 = "", arg2 = ""):

        # 检查接口是否正常

        interfaceName = getRunCommRes('ip addr show| grep 192.168.188.254').split('scope global')[1].strip(' ').split(':')[0]

        code = os.system("ip link show " + interfaceName + " | grep 'state UP' 1>/dev/null 2>&1 ")

        if code != 0:

            if 'eth2' == interfaceName:

                print "err;if=down:eth1,," + interfaceName

            else:

                print "err;if=down:" + interfaceName

        else:

            res = []

            total = len(getAllIP())

            downCount = Counter(0)

            ms251 = ""

            # 检查是否还存在251 IP的AP信息

            state251 = os.system('/bin/ping -c 1 -w 3 192.168.188.251 >/dev/null')

            if state251 == 0:

                ms251 = ':::ex251:::'

            self.threadBox(self.pingTarget, res, downCount)

            if len(res) == 0:

                print  ms251 + 'ok;' + str(total)

            else:

                print  ms251 + 'err;a=' + str(total) + ';d=' + str(downCount.get()) + '::' + str(res)        

    # reboot target

    def rebootTarget(self, ip, resDict, args3 = "", args4 = ""):

        try:

            sshobj = myParamiko(ip, 'root', 'password@123')

            resInfo = sshobj.run_cmd("reboot")

            resDict[ip] = resInfo["stdin"].strip('\n')

            sleep(1)

            sshobj.close()

        except Exception, e:

            resDict[ip] = "Error: " + str(e)

            if 'Authentication failed' in str(e):

                try:

                    rebTelnet(ip, 'admin', 'password')

                    resDict[ip] = "telRebootAp!"

                except Exception, e1:

                    resDict[ip] = "Error: " + str(e1)

    # 重启AP

    def reboot(self):

        resDict = {}

        self.threadBox(self.rebootTarget, resDict)

        print resDict 

# 执行前环境初始化

def initSysDev():

    # 检查chilli程序是否异常退出

    chilliNum = getRunCommRes('ps -aef | grep chilli| grep -v grep | wc -l')

    cwatchNum = getRunCommRes('ps -aef | grep cwatch| grep -v grep | wc -l ')

    #if int(chilliNum) == 0 and int(cwatchNum) == 0:

    if int(chilliNum) == 0 :

        os.system('/usr/local/sbin/cwatch.sh &')

    # 根据网络方式配置IP地址,方便和AP通信

    os.system('ip addr del 192.168.188.254/24 dev eth1 1>/dev/null 2>&1')

    os.system('ip addr del 192.168.188.254/24 dev eth2 1>/dev/null 2>&1')

    os.system('ip addr del 192.168.188.254/24 dev eth3 1>/dev/null 2>&1')

    code = os.system('ifconfig | grep ^br0 >/dev/null')

    if code == 0:

        os.system('ip addr add 192.168.188.254/24 dev br0 1>/dev/null 2>&1')

    else:

        code1 = os.system("ip link show eth1 | grep 'state UP' 1>/dev/null 2>&1 ")

        if code1 == 0:

            os.system("ifconfig eth1:1 192.168.188.254 up 1>/dev/null 2>&1")

        else:

            os.system("ifconfig eth2:1 192.168.188.254 up 1>/dev/null 2>&1")

# 执行开始

def main():

    initSysDev()

    ComUtil = CommonUtil()

    if len(sys.argv) == 1:

        print "please add args!"

    elif len(sys.argv) == 2:

        myFun = getattr(ComUtil, sys.argv[1])

        myFun()

    elif len(sys.argv) == 3:

        myFun = getattr(ComUtil, sys.argv[1])

        myFun(sys.argv[2])

    elif len(sys.argv) == 4:

        myFun = getattr(ComUtil, sys.argv[1])

        myFun(sys.argv[2], sys.argv[3])

    else:

        print "args Error!"

if __name__ == "__main__":

    main()