smart_brain/house_n_gui_sim.py

import config
import logging
import mqtt
import readline
import report
from __simulation__.rooms import house
from __simulation__.test import test_smarthome
import time

if __name__ == "__main__":
    report.stdoutLoggingConfigure(((config.APP_NAME, logging.WARNING), ), report.SHORT_FMT)
    #
    mc = mqtt.mqtt_client(host=config.MQTT_SERVER, port=config.MQTT_PORT, username=config.MQTT_USER,
                          password=config.MQTT_PASSWORD, name=config.APP_NAME + '_simulation')
    #
    COMMANDS = ['quit', 'help', 'test.smoke', 'test.full']
    #
    h = house(mc)
    for name in h.getmembers():
        d = h.getobjbyname(name)
        for c in d.capabilities():
            COMMANDS.append(name + '.' + c)
    #
    ts = test_smarthome(h)

    def reduced_list(text):
        """
        Create reduced completation list
        """
        reduced_list = {}
        for cmd in COMMANDS:
            next_dot = cmd[len(text):].find('.')
            if next_dot >= 0:
                reduced_list[cmd[:len(text) + next_dot + 1]] = None
            else:
                reduced_list[cmd] = None
        return reduced_list.keys()

    def completer(text, state):
        """
        Our custom completer function
        """
        options = [x for x in reduced_list(text) if x.startswith(text)]
        return options[state]

    def complete(text, state):
        for cmd in COMMANDS:
            if cmd.startswith(text):
                if not state:
                    hit = ""
                    index = 0
                    sub_list = cmd.split('.')
                    while len(text) >= len(hit):
                        hit += sub_list[index] + '.'
                    return hit  # cmd
                else:
                    state -= 1

    readline.parse_and_bind("tab: complete")
    readline.set_completer(completer)
    time.sleep(0.3)
    print("\nEnter command: ")

    while True:
        userfeedback = input('')
        command = userfeedback.split(' ')[0]
        if userfeedback == 'quit':
            break
        elif userfeedback == 'help':
            print("Help is not yet implemented!")
        elif userfeedback == 'test.full':
            ts.full()
        elif userfeedback == 'test.smoke':
            ts.smoke()
        elif command in COMMANDS[2:]:
            h.command(userfeedback)
        elif userfeedback != "":
            print("Unknown command!")
        else:
            print()