RobotFramework Library

2019独角兽企业重金招聘Python工程师标准>>>

import os
import sys
import pexpect
import re
import subprocess
import time

class Wifi:

    def __init__(self):
        """"""
        self.cfg_file = '/tmp/wifi_config_file'

    def raiseError(self, Message=None):
        raise Exception, Message

    def ExcuteCMD(self, cmd):
        "Excute shell command ,return code and output"
        cmd = cmd
        print 'CMD is ### ' + cmd + ' ###'
        rc = 1
        content = ""
        # print (datetime.datetime.now())
        p = subprocess.Popen(cmd, shell=True, stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE)

        p.wait()

        rc = p.returncode
        print 'rc is ' + str(rc)

        content = p.stdout.read().strip()
        if content:
            print ">" + content + "<"

        if p.returncode == 0:
            # print 'AT_INFO : "' + cmd + '" Excute SUCESS!'
            rc = True
        else:
            print 'return code :' + str(rc)
            print 'AT_WARNING : "' + cmd + '" Excute FAIL!'
            rc = False
        p.stdout.close()
        print '*'*100
        return rc, content

    def Get_Wlan_Card_Name(self):
        cmd = 'ifconfig -a |grep -o "^ *wlan[0-9][0-9]* *"|awk ' + "'{print $1}'"
        rc, result = self.ExcuteCMD(cmd)
        if rc:
            data = result.strip()
        else:
            data = None
            self.raiseError("error")
        return data

    def Up_Monitor_Interface(self):
        wlan = self.Get_Wlan_Card_Name()
        mon = 'mon_' + str(wlan)
        if wlan:

            if self.ExcuteCMD(cmd='ifconfig |grep "^ *' + wlan + '"')[0]:
                pass
            else:
                if self.ExcuteCMD(cmd='ifconfig ' + wlan + ' up')[0]:
                    pass
                else:
                    print 'AT_ERROR : ' + 'ifconfig ' + wlan + ' up' + ' FAIL FAIL!'
                    return False, None

            if self.ExcuteCMD(cmd='ifconfig |grep ' + mon)[0]:
                print 'AT_INFO : Monitor Interface is already Up PASS PASS!'
                return True, mon
            elif self.ExcuteCMD(cmd='ifconfig -a|grep ' + mon)[0]:
                print 'AT_INFO : Monitor Interface is exist,but NOT UP!'
                if self.ExcuteCMD(cmd='ifconfig ' + mon + ' up')[0]:
                    print 'AT_INFO : Monitor Interface is already Up PASS PASS!'
                    return True, mon
                else:
                    print 'AT_ERROR : Monitor Interface Up FAIL FAIL!'
                    return False, None
            else:
                if self.ExcuteCMD(cmd='iw dev ' + wlan + ' interface add ' + mon + ' type monitor')[0]:
                    if self.ExcuteCMD(cmd='ifconfig |grep ' + mon)[0]:
                        print 'AT_INFO : Monitor Interface is already Up PASS PASS!'
                        return True, mon
                    elif self.ExcuteCMD(cmd='ifconfig -a|grep ' + mon)[0]:
                        if self.ExcuteCMD(cmd='ifconfig ' + mon + ' up')[0]:
                            print 'AT_INFO : Monitor Interface Up PASS PASS!'
                            return True, mon
                        else:
                            print 'AT_ERROR : Monitor Interface Up FAIL FAIL!'
                            return False, None
                else:
                    print 'AT_ERROR : Monitor Interface Add FAIL FAIL!'
                    return False, mon
        else:
            print 'AT_ERROR : NO Exist WLAN Card!'
            return False, None


    def readfile(self, file, filter=''):
        filter = filter
        match_flag = False
        try:
            file = open(file, 'r')
        except Exception, e:
            print e
            return False
        for line in file:
            print line,
            if filter:
                m = re.findall(filter, line)
                if m:
                    match_flag = True
        file.close()
        return match_flag


    def Scan_SSID(self, ssid='', bssid=''):

        wlan = self.Get_Wlan_Card_Name()
        ssid=str(ssid)
        bssid=str(bssid)
        cmd = 'iw dev ' + wlan + ' scan |grep "' + ssid +'" -8' + '|grep ' + bssid

        if self.ExcuteCMD(cmd)[0]:
            return True
        else:
            print 'AT_WARNING : San_SSID RUN FAIL FAIL!'

            self.raiseError("error")

    def Start_WPASupplicant(self):
        wlan = self.Get_Wlan_Card_Name()
        os.system('killall wpa_supplicant')
        time.sleep(5)
        cmd = 'nohup wpa_supplicant -d -i '+ wlan + ' -c ' + self.cfg_file + ' 2>&1 & '
        if os.system(cmd) == 0:
            print 'AT_INFO : Start_WPASupplicant RUN SUCCESS SUCCESS!'
            return True
        else:
            print 'AT_WARNING : Start_WPASupplicant RUN FAIL FAIL!'
            self.raiseError("error")

    def Check_WPASupplicant(self):
        time.sleep(10)
        cmd = 'wpa_cli status'
        rc,result=self.ExcuteCMD(cmd)
        if rc:
            m=re.findall(r'wpa_state=COMPLETED',result)
            if m:
                print 'AT_INFO : Check_WPASuppcliant RUN SUCCESS SUCCESS! And wpa_state is COMPLETED '
                return True
            else:
                print 'AT_INFO : Check_WPASuppcliant RUN SUCCESS SUCCESS! But wpa_state is not COMPLETED '
                self.raiseError("error")

        else:
            print 'AT_WARNING : Check_WPASuppcliant RUN FAIL FAIL!'
            self.raiseError("error")



    def Generate_Config_file(self, ssid='', bssid='', type='', key=''):
        """Generate Wifi Config file for Wireless Card"""
        ssid=str(ssid)
        bssid=str(bssid)
        type=str(type)
        key=str(key)
        try:
            file = open(self.cfg_file, 'w')
        except Exception, e:
            print e
            return False

        file.writelines('ctrl_interface=/var/run/wpa_supplicant' + os.linesep)
        file.writelines('eapol_version=1' + os.linesep)
        file.writelines('ap_scan=1' + os.linesep)
        file.writelines('fast_reauth=1' + os.linesep)
        file.writelines('network={' + os.linesep)
        file.writelines('    ssid=\"' + ssid + '\"' + os.linesep)
        file.writelines('    bssid=' + bssid  + os.linesep)
        file.writelines('    scan_ssid=1' + os.linesep)
        file.writelines('    key_mgmt=' + type + os.linesep)
        file.writelines('    priority=5' + os.linesep)
        file.writelines('    pairwise=CCMP' + os.linesep)
        file.writelines('    proto=WPA RSN' + os.linesep)
        file.writelines('    psk=\"' + key + '\"' + os.linesep)
        file.writelines('}' + os.linesep)
        file.close()
        self.readfile(self.cfg_file)
        return True

        file.close()
        self.readfile(self.cfg_file)
        return True



    def connect_SSID(self, ssid='', bssid='', type='', key=''):
        """check Wireless Card can connect Wireless AP or not

        -Give this Keyword the 4 arguments of a AP: ssid, bssid, type and key

        -Then Case Word will scan SSID Exist or not ,and generate a Wifi config file
        If  Scan SSID Succeed  and config file is Wright, it will invoke wireless
        Tool wpa_supplicant to connect the AP and check the wireless status with
        Wireless Tool wpa_cli.

        -When wpa_state is "COMPLETED", the Keyword run succeed, otherwise it fails.


        Examples:
        | connect SSID | bssid=10:9f:a9:70:01:03 | type=WPA-PSK   |   key=1234567890   |


        """

        ssid=str(ssid)
        bssid=str(bssid)
        type=str(type)
        key=str(key)

        if self.Up_Monitor_Interface():
            pass
        else:
               self.raiseError("error")


        if self.Generate_Config_file(ssid=ssid, bssid=bssid, type=type,  key=key):
            pass
        else:
            self.raiseError("error")

        if self.Scan_SSID(ssid=ssid, bssid=bssid):
            print "scan ssid pass"
            pass
        else:
            print "scan ssid false"
            self.raiseError("error")

        if self.Start_WPASupplicant():
            pass
        else:
            self.raiseError("error")


        if self.Check_WPASupplicant():
            pass
        else:
            self.raiseError("error")

转载于:https://my.oschina.net/xxjbs001/blog/304884