bluetooth-RPi-wifi.py

import os
import subprocess
import select
import serial
import time
import json
import re

wpa_supplicant_conf = "/etc/wpa_supplicant/wpa_supplicant.conf"
sudo_mode = "sudo "
 
class SerialComm:
    def __init__(self):
        self.port = serial.Serial("/dev/rfcomm0", baudrate=9600, timeout=1)
 
    def read_serial(self):
        res = self.port.read(50)
        if len(res):
            return res.splitlines()
        else:
            return []
 
    def send_serial(self, text):
        self.port.write(text)

    def is_json(self, mJson):
        try:
            json_object = json.loads(mJson)
            if isinstance(json_object, int):
                return False

            if len(json_object) == 0:
                return False
        except ValueError, e:
            return False
        return True

    def isValidCommand(self, command, invalidCommand):
        if command not in invalidCommand:
            if re.match("^[a-zA-Z0-9. -]+$",command):
                return True
            
        return False

    def readExecuteSend(self, shell, ble_comm, ble_line):

        json_object = json.loads(ble_line)
        ip_address = ble_comm.wifi_connect(json_object['SSID'], json_object['PWD'])
        if ip_address == "<Not Set>":
            print("Fail to connect to Internet")
            #send back fail to configure wifi
            callback_message = {'result': "FAIL", 'IP': ip_address}
            callback_json = json.dumps(callback_message)
            ble_comm.send_serial(callback_json)
            return False
        
        else:
            #isConnected = True
            print("connect to Internet! your ip_address: " + ip_address)
            #send back configure wifi succesfully
            callback_message = {'result': "SUCCESS", 'IP': ip_address}
            callback_json = json.dumps(callback_message)
            ble_comm.send_serial(callback_json)

            return True
        

    def wifi_connect(self, ssid, psk):
        # write wifi config to file
        f = open('wifi.conf', 'w')
        f.write('country=US\n')
        f.write('ctrl_interface=DIR=/var/run/wpa_supplicant GROUP=netdev\n')
        f.write('update_config=1\n')
        f.write('\n')
        f.write('network={\n')
        f.write('    ssid="' + ssid + '"\n')
        f.write('    psk="' + psk + '"\n')
        f.write('}\n')
        f.close()

        cmd = 'sudo mv wifi.conf ' + wpa_supplicant_conf
        cmd_result = ""
        cmd_result = os.system(cmd)
        print cmd + " - " + str(cmd_result)


        # restart wifi adapter
        cmd = sudo_mode + 'ifdown wlan0'
        cmd_result = os.system(cmd)
        print cmd + " - " + str(cmd_result)

        time.sleep(2)

        cmd = sudo_mode + 'ifup wlan0'
        cmd_result = os.system(cmd)
        print cmd + " - " + str(cmd_result)

        time.sleep(10)

        cmd = 'iwconfig wlan0'
        cmd_result = os.system(cmd)
        print cmd + " - " + str(cmd_result)

        cmd = 'ifconfig wlan0'
        cmd_result = os.system(cmd)
        print cmd + " - " + str(cmd_result)

        p = subprocess.Popen(['ifconfig', 'wlan0'], stdout=subprocess.PIPE,
                                stderr=subprocess.PIPE)

        out, err = p.communicate()

        ip_address = "<Not Set>"
    
        for l in out.split('\n'):
            if l.strip().startswith("inet addr:"):
                ip_address = l.strip().split(' ')[1].split(':')[1]
                
        return ip_address
 
 
class ShellWrapper:
    def __init__(self):
        self.ps = subprocess.Popen(['bash'], stdout=subprocess.PIPE, stderr=subprocess.PIPE, stdin=subprocess.PIPE)
 
    def execute_command(self, command):
        self.ps.stdin.write(command + "\n")
 
    def get_output(self):
        timeout = False
        time_limit = .5
        lines = []
        while not timeout:
            poll_result = select.select([self.ps.stdout, self.ps.stderr], [], [], time_limit)[0]
            if len(poll_result):

                for p in poll_result:
                    lines.append(p.readline())              
            else:
                timeout = True
            
        if(len(lines)):
            return lines
        else:
            return None
 
 
def main():
    shell = ShellWrapper()
    invalidCommand = ['clear','head','sudo','nano','touch','vim']
    ble_comm = None
    isConnected = False
    
    while True:
        try:
            ble_comm = SerialComm()
            out = ble_comm.read_serial()
            for ble_line in out:
                print(out)
                if ble_comm.is_json(ble_line):
                    
                    if not isConnected:
                        isConnected = ble_comm.readExecuteSend(shell, ble_comm, ble_line)
                        break
                    else:
                        ble_comm.send_serial("Wifi has been configured")
                        break
                
                if ble_comm.isValidCommand(ble_line, invalidCommand):
                    
                    shell.execute_command(ble_line)
                    shell_out = shell.get_output()
                    if shell_out != None:
                        for l in shell_out:
                            print(l)
                            ble_comm.send_serial(l)
                    else:
                        ble_comm.send_serial("command '" + ble_line + "' return nothing ")
                else:
                  ble_comm.send_serial("command '" + ble_line + "' not support ")  
          
        except serial.SerialException:
            print("waiting for connection")
            ble_comm = None
            isConnected = False
            time.sleep(1)
            
if __name__ == "__main__":
    main()