-

Author: previ

templates/config.html

@@ -16,6 +16,7 @@ <h1>CoderBot</h1>
       <li><a href="#t_start" data-ajax="true">{% trans %}Start{% endtrans %}</a></li>
       <li><a href="#t_button" data-ajax="true">{% trans %}Button{% endtrans %}</a></li>
       <li><a href="#t_ui" data-ajax="true">{% trans %}User Interface{% endtrans %}</a></li>
+      <li><a href="#t_wifi" data-ajax="true">{% trans %}WiFi{% endtrans %}</a></li>
      </ul>
     </div>
     <div id="t_control" class="ui-body-d ui-content">
@@ -128,6 +129,20 @@ <h1>CoderBot</h1>
       <input type="checkbox" name="show_page_prefs" id="c_show_page_prefs" value="true" {%if config.show_page_prefs=='true'%}checked="checked"{%endif%}>
      </fieldset>
     </div>
+   <div id="t_wifi" class="ui-body-d ui-content">
+     <fieldset data-role="controlgroup" data-mini="true">
+      <legend>{% trans %}WiFi mode{% endtrans %}</legend>
+      <input type="radio" name="wifi_mode" id="r_wifi_mode_a" value="ap" {%if config.wifi_mode=='ap'%}checked="checked"{%endif%}>
+      <label for="r_wifi_mode_a">{% trans %}Access Point{% endtrans %}</label>
+      <input type="radio" name="wifi_mode" id="r_wifi_mode_b" value="client" {%if config.wifi_mode=='client'%}checked="checked"{%endif%}>
+      <label for="r_wifi_mode_b">{% trans %}Client{% endtrans %}</label>
+     </fieldset>
+     <legend>WiFi Client mode access data</legend>
+     <label for="i_wifi_ssid">{% trans %}WiFi name{% endtrans %}</label>
+     <input type="text" id="i_wifi_ssid" name="wifi_ssid" value="{{config.wifi_ssid}}">
+     <label for="i_wifi_psk">{% trans %}WiFi password{% endtrans %}</label>
+     <input type="text" id="i_wifi_psk" name="wifi_psk" value="{{config.wifi_psk}}">
+    </div>
     <div class="ui-grid-a">
      <div class="ui-block-a"><a href="#" data-rel="close" class="ui-btn ui-shadow ui-corner-all ui-btn-b ui-mini">{% trans %}Cancel{% endtrans %}</a></div>
      <div class="ui-block-b"><button type="submit" data-rel="close" class="ui-btn ui-shadow ui-corner-all ui-btn-a ui-mini">{% trans %}Save{% endtrans %}</button></div>

viz/image.py

@@ -9,7 +9,8 @@
 MIN_MATCH_COUNT = 10
 
 class Image():
-    _face_cascade = cv2.CascadeClassifier('/usr/local/share/OpenCV/haarcascades/haarcascade_frontalface_default.xml')
+    #_face_cascade = cv2.CascadeClassifier('/usr/local/share/OpenCV/haarcascades/haarcascade_frontalface_default.xml')
+    _face_cascade = cv2.CascadeClassifier('/usr/local/share/OpenCV/lbpcascades/lbpcascade_frontalface.xml')
     _kernel = np.ones((3,3),np.uint8)
 
     def __init__(self, array):

wifi.py

@@ -0,0 +1,150 @@
+#!/usr/bin/python
+
+import socket
+import subprocess
+import shutil
+import sys
+import os
+import urllib2
+import fcntl
+import struct
+import json
+
+class WiFi():
+
+  CONFIG_FILE = "/etc/coderbot_wifi.conf"
+  adapters = ["RT5370", "RTL8188CUS"] 
+  hostapds = {"RT5370": "hostapd.RT5370", "RTL8188CUS": "hostapd.RTL8188"} 
+  web_url = "http://coderbotsrv.appspot.com/register_ip"
+  wifi_client_conf_file = "/etc/wpa_supplicant/wpa_supplicant.conf"
+  _config = {}
+
+  @classmethod
+  def load_config(cls):
+    f = open(cls.CONFIG_FILE)
+    cls._config = json.load(f)
+    return cls._config
+
+  @classmethod
+  def save_config(cls):
+    f = open(cls.CONFIG_FILE, 'w')
+    json.dump(cls._config, f)
+    return cls._config
+
+  @classmethod
+  def get_config(cls):
+    return cls._config
+
+  @classmethod
+  def get_adapter_type(cls):
+    lsusb_out = subprocess.check_output("lsusb")
+    for a in cls.adapters:
+      if a in lsusb_out:
+        return a
+    return None
+    
+  @classmethod
+  def start_hostapd(cls):
+    adapter = cls.get_adapter_type()
+    hostapd_type = cls.hostapds.get(adapter)
+    try:
+      print "starting hostapd..."
+      os.system("start-stop-daemon --start --oknodo --quiet --exec /usr/sbin/" + hostapd_type + " -- /etc/hostapd/" + hostapd_type + " &")
+
+    except subprocess.CalledProcessError as e:
+      print e.output
+
+  @classmethod
+  def stop_hostapd(cls):
+    try:
+      out = subprocess.check_output(["pkill", "-9", "hostapd"])
+      print out
+    except subprocess.CalledProcessError as e:
+      print e.output
+
+  @classmethod
+  def get_ipaddr(cls, ifname):
+    s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+    return socket.inet_ntoa(fcntl.ioctl(
+        s.fileno(),
+        0x8915,  # SIOCGIFADDR
+        struct.pack('256s', ifname[:15])
+    )[20:24])
+    ipaddr = socket.gethostbyname(socket.gethostname())
+    return ipaddr
+
+  @classmethod
+  def register_ipaddr(cls, ipaddr, botname):
+    try:
+      ret = urllib2.urlopen(cls.web_url + "?name=" + botname + "&ipaddr=" + ipaddr)
+      if ret.getcode() != 200:
+        raise Exception()
+    except URLError as e:
+      print e
+      raise
+    print botname, ": ", ipaddr
+
+  @classmethod
+  def get_wlans(cls):
+    out = subprocess.check_output(["iwlist", "wlan0", "scan"])  
+
+  @classmethod
+  def set_client_params(cls, wssid, wpsk):
+    f = open (cls.wifi_client_conf_file, "w+")
+    f.write("""ctrl_interface=DIR=/var/run/wpa_supplicant GROUP=netdev
+update_config=1
+network={\n""")
+    f.write("  ssid=\""+wssid+"\"\n")
+    f.write("  psk=\""+wpsk+"\"\n")
+    f.write("}")
+
+  @classmethod
+  def set_start_as_client(cls):
+    shutil.copy("/etc/network/interfaces_cli", "/etc/network/interfaces")
+    cls._config["wifi_mode"] = "client"
+    cls.save_config()
+
+  @classmethod
+  def start_as_client(cls):
+    cls.stop_hostapd()
+    try:
+      out = subprocess.check_output(["ifdown", "wlan0"])
+      out = subprocess.check_output(["ifup", "wlan0"])
+    except subprocess.CalledProcessError as e:
+      print e.output
+      raise
+
+  @classmethod
+  def set_start_as_ap(cls):
+    shutil.copy("/etc/network/interfaces_ap", "/etc/network/interfaces")
+    cls._config["wifi_mode"] = "ap"
+    cls.save_config()
+
+  @classmethod
+  def start_as_ap(cls):
+    out = subprocess.check_output(["ifdown", "wlan0"])
+    out = subprocess.check_output(["ifup", "wlan0"])
+    cls.start_hostapd()
+
+
+def main():
+  print "starting wifi service..."
+  w = WiFi()
+  config = w.load_config()
+
+  if config["wifi_mode"] == "ap":
+    print "starting as ap..."
+    w.start_as_ap()
+  elif config["wifi_mode"] == "client":
+    print "starting as client..."
+    w.start_as_client()
+    try:
+      print "registering ip..."
+      w.register_ipaddr(w.get_ipaddr("wlan0"), "CoderBot")
+    except:
+      print "Unable to register ip, revert to ap mode"
+      w.start_as_ap()
+
+if __name__ == "__main__":
+  main()
+