elal
/
roomcomputer
mirror of https://github.com/E-Almqvist/roomcomputer
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Revert "Made code pep8 compliant once again"

Browse Source 
This reverts commit c0ad5ce2d7.
pull/3/head
Alve 5 years ago
parent c0ad5ce2d7
commit ea87ee293c
8 changed files with 287 additions and 306 deletions
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Show Stats Download Patch File Download Diff File
  1. 10
      hue_cmd.py
  2. 1
      modules/configloader/__init__.py
  3. 15
      modules/configloader/loader.py
  4. 299
      modules/hue/hue_controller.py
  5. 135
      modules/hue/hue_remote.py
  6. 11
      modules/hue/lib/func.py
  7. 68
      modules/speech/speech.py
  8. 54
      speech_daemon.py

10
hue_cmd.py
Unescape View File

@ -3,12 +3,10 @@
from modules.hue.hue_controller import controller from modules.hue.hue_controller import controller
from modules.hue.hue_remote import parseCommandline from modules.hue.hue_remote import parseCommandline
def init(): def init():
controller.init() # very important to initialize the controller controller.init() # very important to initialize the controller
parseCommandline() parseCommandline()
controller.end() # also to end it controller.end() # also to end it
if __name__ == "__main__": if __name__ == "__main__":
init() init()

1
modules/configloader/__init__.py
Unescape View File

@ -0,0 +1 @@

15
modules/configloader/loader.py
Unescape View File

@ -1,13 +1,12 @@
import json import json
def readconfig(path): def readconfig(path):
try: try:
with open(path) as cfg: with open(path) as cfg:
data = json.load(cfg) data = json.load(cfg)
return data return data
except: except:
print("[Error] Something went wrong reading the configuration file.") print("[Error] Something went wrong reading the configuration file.")
print("--", path) print("--", path)

299
modules/hue/hue_controller.py
Unescape View File

@ -1,173 +1,168 @@
import requests as req # Used for HTTP requests for the Hue API import requests as req # Used for HTTP requests for the Hue API
import json # API uses JSON import json # API uses JSON
import asyncio # ASync stuff import asyncio # ASync stuff
import time import time
from .lib.func import * # useful functions from .lib.func import * # useful functions
from modules.configloader.loader import readconfig # used to read the config files from modules.configloader.loader import readconfig # used to read the config files
from os.path import expanduser # to get the home dir from os.path import expanduser # to get the home dir
homedir = expanduser("~") # get the home directory of the current user homedir = expanduser("~") # get the home directory of the current user
LIGHTS = {} # dictionary of all the lights LIGHTS = {} # dictionary of all the lights
CONFIG = {} # the configuration CONFIG = {} # the configuration
PRESETS = {} # the presets PRESETS = {} # the presets
PRE_URL = "" # prefix PRE_URL = "" # prefix
loop = asyncio.get_event_loop() # ASync loop
loop = asyncio.get_event_loop() # ASync loop
def genUrl(params: str): def genUrl(params: str):
return PRE_URL + params return PRE_URL + params
class APIrequest: class APIrequest:
# Get Req # Get Req
async def get(dest: str = "", payload: str = ""): async def get( dest: str="", payload: str="" ):
try: try:
apiReq = req.get(genUrl(dest), data=payload) apiReq = req.get( genUrl(dest), data = payload )
if(apiReq.status_code != 200): # print out the error if the status code is not 200 if( apiReq.status_code != 200 ): # print out the error if the status code is not 200
print(apiReq) print(apiReq)
print(apiReq.text) print(apiReq.text)
return apiReq return apiReq
except req.exceptions.RequestException as err: except req.exceptions.RequestException as err:
print(err) print(err)
# PUT Req # PUT Req
async def put(dest: str = "", payload: str = ""): async def put( dest: str="", payload: str="" ):
try: try:
apiReq = req.put(genUrl(dest), data=payload) # send the payload apiReq = req.put( genUrl(dest), data = payload ) # send the payload
if(apiReq.status_code != 200): if( apiReq.status_code != 200 ):
print(apiReq) print(apiReq)
print(apiReq.text) print(apiReq.text)
return apiReq return apiReq
except req.exceptions.RequestException as err: except req.exceptions.RequestException as err:
print(err) print(err)
class controller: class controller:
# Internal get functions # Internal get functions
async def getLights(): async def getLights():
return await APIrequest.get("/lights") return await APIrequest.get("/lights")
async def getLight(index: int = 1): async def getLight(index: int=1):
return await APIrequest.get("/lights/" + str(index)) return await APIrequest.get( "/lights/" + str(index) )
# Lower level light manipulation (async) # Lower level light manipulation (async)
async def toggleLight(index: int = 1, isOn: bool = True): async def toggleLight(index: int=1, isOn: bool=True):
await APIrequest.put("/lights/" + str(index) + "/state", '{"on":' + boolToString(isOn) + '}') await APIrequest.put( "/lights/" + str(index) + "/state", '{"on":' + boolToString(isOn) + '}' )
async def toggleLights(isOn: bool = True): async def toggleLights(isOn: bool=True):
for key in LIGHTS: for key in LIGHTS:
await controller.toggleLight(key, isOn) await controller.toggleLight(key, isOn)
async def setLightRGB(index: int, r: int, g: int, b: int): async def setLightRGB( index: int, r:int, g:int, b:int ):
h, s, v = rgbToHsv(r, g, b) h, s, v = rgbToHsv(r, g, b)
payload = '{"sat":' + str(s) + ', "bri":' + \ payload = '{"sat":' + str(s) + ', "bri":' + str(v) + ', "hue":' + str(h) + '}'
str(v) + ', "hue":' + str(h) + '}'
await APIrequest.put( "/lights/" + str(index) + "/state", payload )
await APIrequest.put("/lights/" + str(index) + "/state", payload)
# Normal functions
# Normal functions def switchLight( index: int=1 ):
def switchLight(index: int = 1): key = LIGHTS.get(str(index))
key = LIGHTS.get(str(index)) if(key):
if(key): if( key.get("state") ):
if(key.get("state")): curPower = LIGHTS[str(index)]["state"]["on"]
curPower = LIGHTS[str(index)]["state"]["on"] loop.run_until_complete( controller.toggleLight(index, not curPower))
loop.run_until_complete( else:
controller.toggleLight(index, not curPower)) print("Error: Light index '" + str(index) + "' out of range")
else:
print("Error: Light index '" + str(index) + "' out of range") def switchLights():
for key in LIGHTS:
def switchLights(): controller.switchLight(key)
for key in LIGHTS:
controller.switchLight(key) # Light control
def setLightColor( index:int, r:int, g:int, b:int ):
# Light control if( LIGHTS.get(str(index)) ):
def setLightColor(index: int, r: int, g: int, b: int): loop.run_until_complete( controller.setLightRGB(index, r, g, b) )
if(LIGHTS.get(str(index))): else:
loop.run_until_complete(controller.setLightRGB(index, r, g, b)) print("Error: Light index '" + str(index) + "' out of range")
else:
print("Error: Light index '" + str(index) + "' out of range") def setLightBrightness( index:int, b:int ):
if( LIGHTS.get(str(index)) ):
def setLightBrightness(index: int, b: int): payload = '{"bri":' + str(b) + '}'
if(LIGHTS.get(str(index))): loop.run_until_complete( APIrequest.put( "/lights/" + str(index) + "/state", payload ) )
payload = '{"bri":' + str(b) + '}' else:
loop.run_until_complete(APIrequest.put( print("Error: Light index '" + str(index) + "' out of range")
"/lights/" + str(index) + "/state", payload))
else: def setBrightness( b:int ):
print("Error: Light index '" + str(index) + "' out of range") for key in LIGHTS:
controller.setLightBrightness( key, b )
def setBrightness(b: int):
for key in LIGHTS: def setAllLightsColor( r:int, g:int, b:int ):
controller.setLightBrightness(key, b) for key in LIGHTS:
controller.setLightColor( key, r, g, b )
def setAllLightsColor(r: int, g: int, b: int):
for key in LIGHTS: def Power(isOn:bool=True): # Controlling the power of the lights
controller.setLightColor(key, r, g, b) loop.run_until_complete( controller.toggleLights(isOn) )
def Power(isOn: bool = True): # Controlling the power of the lights def powerLight( index:int, isOn:bool=True ):
loop.run_until_complete(controller.toggleLights(isOn)) loop.run_until_complete( controller.toggleLight( index, isOn ) )
def powerLight(index: int, isOn: bool = True): # Presets
loop.run_until_complete(controller.toggleLight(index, isOn)) def setLightPreset( index:int, p:str ):
if( LIGHTS.get(str(index)) ):
# Presets if( PRESETS.get(p) ):
def setLightPreset(index: int, p: str): preset = PRESETS[p]
if(LIGHTS.get(str(index))): r, g, b = preset["color"]
if(PRESETS.get(p)): brightness = preset["brightness"]
preset = PRESETS[p]
r, g, b = preset["color"] controller.setLightColor( index, r, g, b )
brightness = preset["brightness"] controller.setLightBrightness( index, brightness )
else:
controller.setLightColor(index, r, g, b) print("Error: Unknown preset '" + p + "'")
controller.setLightBrightness(index, brightness) else:
else: print("Error: Light index '" + str(index) + "' out of range")
print("Error: Unknown preset '" + p + "'")
else: def setPreset( presetID:str, index:int=-1 ):
print("Error: Light index '" + str(index) + "' out of range") if( PRESETS.get(presetID) ):
if( index == -1 ):
def setPreset(presetID: str, index: int = -1): for key in LIGHTS:
if(PRESETS.get(presetID)): controller.setLightPreset( key, presetID )
if(index == -1): else:
for key in LIGHTS: controller.setLightPreset( index, presetID )
controller.setLightPreset(key, presetID) else:
else: print("Error: Unknown preset '" + presetID + "'")
controller.setLightPreset(index, presetID)
else: def countLights():
print("Error: Unknown preset '" + presetID + "'") return len(LIGHTS)
def countLights(): # Controller "system" functions
return len(LIGHTS) def delay(n:int):
time.sleep(n)
# Controller "system" functions
def delay(n: int): def init( cfgPath="{0}/.config/roomcomputer/config.json".format(homedir), presetPath="{0}/.config/roomcomputer/presets.json".format(homedir) ):
time.sleep(n) config = readconfig(cfgPath)
presets = readconfig(presetPath)
def init(cfgPath="{0}/.config/roomcomputer/config.json".format(homedir), presetPath="{0}/.config/roomcomputer/presets.json".format(homedir)):
config = readconfig(cfgPath) global CONFIG
presets = readconfig(presetPath) CONFIG = config["hue"]
global CONFIG global PRESETS
CONFIG = config["hue"] PRESETS = presets
global PRESETS global PRE_URL
PRESETS = presets PRE_URL = "http://" + CONFIG["address"] + "/api/" + CONFIG["username"]
global PRE_URL jsonLights = loop.run_until_complete(APIrequest.get("/lights"))
PRE_URL = "http://" + CONFIG["address"] + "/api/" + CONFIG["username"] global LIGHTS
LIGHTS = json.loads(jsonLights.text)
jsonLights = loop.run_until_complete(APIrequest.get("/lights"))
global LIGHTS def end():
LIGHTS = json.loads(jsonLights.text) loop.close()
def end():
loop.close()

135
modules/hue/hue_remote.py
Unescape View File

@ -3,101 +3,96 @@
import sys import sys
from modules.hue import hue_controller as hue # Actual controller from modules.hue import hue_controller as hue # Actual controller
cmd = "hue" cmd = "hue"
def help(): def help():
print("--Help page--") print("--Help page--")
print("'" + cmd + "' : Display this help page")
print("'" + cmd + " light (index)' ... : Specify light target, from 1-" +
str(hue.controller.countLights()))
print("'" + cmd + " lights' ... : Specify all lights\n")
print("--Commands--") print( "'" + cmd + "' : Display this help page" )
print("'on'/'off' : Turn light(s) on/off") print( "'" + cmd + " light (index)' ... : Specify light target, from 1-" + str(hue.controller.countLights()) )
print("'switch' : Switch the light(s) power") print( "'" + cmd + " lights' ... : Specify all lights\n" )
print("'set ...'")
print(" 'preset (preset ID)' : Set the preset (from presets.py)")
print(" 'color (red) (green) (blue)' : Set the color, from 0-255")
print(" 'brightness (brightness)' : Set the brightness, from 0-255")
print("\nExamples:\n'hue light 2 on' : Turn on light 2\n'hue lights set color 255 255 255' : Set all lights colors to white") print("--Commands--")
print( "'on'/'off' : Turn light(s) on/off" )
print( "'switch' : Switch the light(s) power" )
print( "'set ...'" )
print( " 'preset (preset ID)' : Set the preset (from presets.py)" )
print( " 'color (red) (green) (blue)' : Set the color, from 0-255" )
print( " 'brightness (brightness)' : Set the brightness, from 0-255" )
print("\nExamples:\n'hue light 2 on' : Turn on light 2\n'hue lights set color 255 255 255' : Set all lights colors to white")
boolConvert = { boolConvert = {
"on": True, "on": True,
"off": False "off": False
} }
# this is the most spaghetti-ish code I have ever written but it works # this is the most spaghetti-ish code I have ever written but it works
def parseCommand( cmd:list, pos:int, index=-1, displayHelp=True ):
try:
if( cmd[pos] == "on" or cmd[pos] == "off" ):
if( index == -1 ):
hue.controller.Power( boolConvert[cmd[pos]] )
else:
hue.controller.powerLight( index, boolConvert[cmd[pos]] )
def parseCommand(cmd: list, pos: int, index=-1, displayHelp=True): return
try:
if(cmd[pos] == "on" or cmd[pos] == "off"):
if(index == -1):
hue.controller.Power(boolConvert[cmd[pos]])
else:
hue.controller.powerLight(index, boolConvert[cmd[pos]])
return
elif(cmd[pos] == "switch"): elif( cmd[pos] == "switch" ):
if(index == -1): if(index == -1):
hue.controller.switchLights() hue.controller.switchLights()
else: else:
hue.controller.switchLight(index) hue.controller.switchLight(index)
return return
elif(cmd[pos] == "set"): elif( cmd[pos] == "set" ):
if(cmd[pos+1] == "preset"): if( cmd[pos+1] == "preset" ):
hue.controller.setPreset(cmd[pos+2], index) hue.controller.setPreset( cmd[pos+2], index )
return return
elif(cmd[pos+1] == "color"): elif( cmd[pos+1] == "color" ):
if(len(cmd) > pos+4): if( len(cmd) > pos+4 ):
r, g, b = int(cmd[pos+2]), int(cmd[pos+3]), int(cmd[pos+4]) r, g, b = int(cmd[pos+2]), int(cmd[pos+3]), int(cmd[pos+4])
if(index == -1): if( index == -1 ):
hue.controller.setAllLightsColor( hue.controller.setAllLightsColor( r, g, b ) # this code is bad
r, g, b) # this code is bad else:
else: hue.controller.setLightColor( index, r, g, b )
hue.controller.setLightColor(index, r, g, b)
return return
else: else:
print("Error: Missing parameters") print("Error: Missing parameters")
help() help()
elif(cmd[pos+1] == "brightness"): elif( cmd[pos+1] == "brightness" ):
if(len(cmd) > pos+2): if( len(cmd) > pos+2 ):
bri = int(cmd[pos+2]) bri = int(cmd[pos+2])
if(index == -1): if( index == -1 ):
hue.controller.setBrightness(bri) hue.controller.setBrightness(bri)
else: else:
hue.controller.setLightBrightness(index, bri) hue.controller.setLightBrightness( index, bri )
return return
help() # display help if function did nothing help() # display help if function did nothing
except (RuntimeError, TypeError, NameError, IndexError) as err: except (RuntimeError, TypeError, NameError, IndexError) as err:
if(displayHelp): if(displayHelp):
help() # display the help page if parameters are missing (it will give out an IndexError) help() # display the help page if parameters are missing (it will give out an IndexError)
print("\n\nError: " + str(err)) print( "\n\nError: " + str(err) )
def parseCommandline(cmd=sys.argv, needHelp=True): def parseCommandline( cmd=sys.argv, needHelp=True ):
if(len(cmd) > 1): if( len(cmd) > 1 ):
if(cmd[1] == "light"): if( cmd[1] == "light" ):
parseCommand(cmd, 3, cmd[2], displayHelp=needHelp) parseCommand( cmd, 3, cmd[2], displayHelp=needHelp )
elif(cmd[1] == "lights"): elif( cmd[1] == "lights" ):
parseCommand(cmd, 2, displayHelp=needHelp) parseCommand( cmd, 2, displayHelp=needHelp )
elif(needHelp): elif( needHelp ):
help() help()

11
modules/hue/lib/func.py
Unescape View File

@ -5,24 +5,19 @@ boolStr = {
False: "false" False: "false"
} }
def boolToString(v: bool): # To fix the dumb python syntax
def boolToString(v: bool): # To fix the dumb python syntax
return boolStr[v] return boolStr[v]
def rgbToDecimal( r:int, g:int, b:int ):
def rgbToDecimal(r: int, g: int, b: int):
return round(r/255, 1), round(g/255, 1), round(b/255, 1) return round(r/255, 1), round(g/255, 1), round(b/255, 1)
def svNumFix(n: float): def svNumFix(n: float):
return int(round(n*254, 0)) return int(round(n*254, 0))
def hueNumFix(n: float): def hueNumFix(n: float):
return int(round(n*65535, 0)) return int(round(n*65535, 0))
def rgbToHsv( r:int, g:int, b:int ):
def rgbToHsv(r: int, g: int, b: int):
R, G, B = rgbToDecimal(r, g, b) R, G, B = rgbToDecimal(r, g, b)
H, S, V = colorsys.rgb_to_hsv(R, G, B) H, S, V = colorsys.rgb_to_hsv(R, G, B)
return hueNumFix(H), svNumFix(S), svNumFix(V) return hueNumFix(H), svNumFix(S), svNumFix(V)

68
modules/speech/speech.py
Unescape View File

@ -1,37 +1,37 @@
import speech_recognition as sr import speech_recognition as sr
class voiceInput(object): class voiceInput(object):
recognizer = sr.Recognizer() recognizer = sr.Recognizer()
muted = True muted = True
# "Error codes", can be used to check stuff # "Error codes", can be used to check stuff
what = "??" what = "??"
error = "ERROR" error = "ERROR"
def start(self, deviceIndex=30): # a generator for everything that is said def start( self, deviceIndex=30 ): # a generator for everything that is said
while(True): # loop while( True ): # loop
try: try:
if(not self.muted): # this thing is not the NSA if( not self.muted ): # this thing is not the NSA
with sr.Microphone(deviceIndex) as src: with sr.Microphone( deviceIndex ) as src:
self.recognizer.adjust_for_ambient_noise(src, 0.2) self.recognizer.adjust_for_ambient_noise( src, 0.2 )
print("Listening...") print("Listening...")
audio = self.recognizer.listen( audio = self.recognizer.listen( src, phrase_time_limit=5 )
src, phrase_time_limit=5) print("Thinking...")
print("Thinking...") text = self.recognizer.recognize_google(audio)
text = self.recognizer.recognize_google(audio) yield text
yield text
except sr.RequestError as err:
except sr.RequestError as err: print("Unable to request results: {0}".format(err))
print("Unable to request results: {0}".format(err)) yield self.error
yield self.error
except sr.UnknownValueError:
except sr.UnknownValueError: yield self.what
yield self.what
def setMuted(self, setm: bool = True): def setMuted( self, setm: bool=True ):
self.muted = setm self.muted = setm
def switchMute(self): def switchMute( self ):
self.setMuted(not self.muted) self.setMuted( not self.muted )

54
speech_daemon.py
Unescape View File

@ -13,43 +13,41 @@ homedir = expanduser("~")
CONFIG = {} CONFIG = {}
class speech_daemon(object): class speech_daemon(object):
voiceInpObj = None voiceInpObj = None
deviceIndex = 30 deviceIndex = 30
def __init__(self, deviceIndex=30):
self.voiceInpObj = voiceInput()
self.voiceInpObj.setMuted(False)
self.deviceIndex = deviceIndex def __init__(self, deviceIndex=30):
self.voiceInpObj = voiceInput()
self.voiceInpObj.setMuted(False)
def loadconfig(self): self.deviceIndex = deviceIndex
path = homedir + "/.config/roomcomputer/config.json"
# if no config path is
# specified then choose the users default
if(len(sys.argv) > 1): def loadconfig(self):
path = sys.argv[1] path = homedir + "/.config/roomcomputer/config.json"
# if no config path is
# specified then choose the users default
cfg = readconfig(path) # read the config if( len(sys.argv) > 1 ):
path = sys.argv[1]
global CONFIG cfg = readconfig(path) # read the config
CONFIG = cfg
def start(self): global CONFIG
controller.init() CONFIG = cfg
for inp in self.voiceInpObj.start(self.deviceIndex): def start(self):
cmdBuf = inp.lower().split(" ") controller.init()
if(cmdBuf[0] in CONFIG["speech"]["prefixes"]):
print("CMD:", cmdBuf)
parseCommandline(cmdBuf, False)
controller.end() for inp in self.voiceInpObj.start( self.deviceIndex ):
cmdBuf = inp.lower().split(" ")
if( cmdBuf[0] in CONFIG["speech"]["prefixes"] ):
print("CMD:", cmdBuf)
parseCommandline( cmdBuf, False )
controller.end()
if __name__ == "__main__": if __name__ == "__main__":
daemon = speech_daemon() daemon = speech_daemon()
daemon.loadconfig() daemon.loadconfig()
daemon.start() daemon.start()

Write Preview
Loading…
Cancel
Save