Initial commit
This commit is contained in:
l_samenv
219
dummy.py
Executable file
219
dummy.py
Executable file
@ -0,0 +1,219 @@
|
||||
#!/usr/bin/env python
|
||||
|
||||
import json
|
||||
import sys
|
||||
import copy
|
||||
from collections import OrderedDict, deque
|
||||
import tcp_lineserver as lineserver
|
||||
import uuid
|
||||
import traceback
|
||||
|
||||
class Group(object):
|
||||
def __init__(self, name):
|
||||
self.name = name
|
||||
self.title = name
|
||||
self.components = []
|
||||
|
||||
class Component(object):
|
||||
def __init__(self, name, g):
|
||||
self.name = name
|
||||
self.group = g
|
||||
self.value = None
|
||||
self.properties = {}
|
||||
|
||||
init_values = ['device name', 'device stick_name']
|
||||
|
||||
def set_value(c, value):
|
||||
c.value = value
|
||||
print 'set',c.name,c.value,c.group
|
||||
for n in init_values:
|
||||
if c.name == n:
|
||||
init(cindex['device name'].value)
|
||||
break
|
||||
for client in clients.values():
|
||||
client.send_msg(dict(type='update', updates=[dict(name=c.name, value=c.value)]))
|
||||
|
||||
|
||||
def history_add(item):
|
||||
history.append(item)
|
||||
id, is_cmd, line = item
|
||||
for cid, client in console_clients.items():
|
||||
client.send_msg(dict(
|
||||
type = 'command' if is_cmd else 'reply',
|
||||
origin = 'self' if cid == id else 'other',
|
||||
line = line
|
||||
))
|
||||
|
||||
class ClientType:
|
||||
GROUP = 1
|
||||
CONSOLE = 2
|
||||
MAINUPDATE = 3
|
||||
|
||||
class DummyHandler(lineserver.LineHandler):
|
||||
def __init__(self, *args, **kwargs):
|
||||
lineserver.LineHandler.__init__(self, *args, **kwargs)
|
||||
self.id = None
|
||||
|
||||
def send_msg(self, msg):
|
||||
print '<', msg
|
||||
self.send_line(json.dumps(msg))
|
||||
|
||||
def handle_line(self, line):
|
||||
print '>', line
|
||||
msg = json.loads(line)
|
||||
msgtype = msg['type']
|
||||
if msgtype == 'init':
|
||||
id = msg.get('id', '0')
|
||||
if self.id != msg['id']:
|
||||
self.id = msg['id']
|
||||
clients[self.id] = self
|
||||
self.title = ""
|
||||
global main_update
|
||||
main_update.append(self)
|
||||
#self.send_msg({'type': 'accept-mainupdate'})
|
||||
init(cindex['device name'].value)
|
||||
elif msgtype == 'getblock':
|
||||
path = msg['path']
|
||||
grp = path.split(",")[-1]
|
||||
g = groups[grp]
|
||||
clist = []
|
||||
for c in g.components:
|
||||
cdict = copy.deepcopy(c.properties)
|
||||
cdict['name'] = c.name
|
||||
clist.append(cdict)
|
||||
self.send_msg(dict(type='draw', path=path, title=g.title, components=clist))
|
||||
elif msgtype == 'updateblock':
|
||||
path = msg['path']
|
||||
grp = path.split(",")[-1]
|
||||
g = groups[grp]
|
||||
clist = []
|
||||
updates = []
|
||||
for c in g.components:
|
||||
if c.properties.get('type','') != 'group':
|
||||
updates.append(dict(name=c.name, value=c.value))
|
||||
self.send_msg(dict(type='accept-block'))
|
||||
self.send_msg(dict(type='update', updates=updates))
|
||||
elif msgtype == 'console':
|
||||
print '*** make console ',self.id,self
|
||||
self.send_msg(dict(type='accept-console'))
|
||||
console_clients[self.id] = self
|
||||
for item in history:
|
||||
id, is_cmd, line = item
|
||||
self.send_msg(dict(
|
||||
type = 'command' if is_cmd else 'reply',
|
||||
origin = 'self' if self.id == id else 'other',
|
||||
line = line
|
||||
))
|
||||
elif msgtype == 'sendcommand':
|
||||
cmd = msg['command'].split(' ')
|
||||
history_add((self.id, True, msg['command']))
|
||||
if len(cmd) == 1:
|
||||
name = cmd[0]
|
||||
c = cindex.get(name, None)
|
||||
val = None
|
||||
else:
|
||||
name = cmd[0] + " " + cmd[1]
|
||||
try:
|
||||
c = cindex[name]
|
||||
except KeyError:
|
||||
try:
|
||||
c = cindex[cmd[0]]
|
||||
except KeyError:
|
||||
c = None
|
||||
else:
|
||||
name = cmd[0]
|
||||
val = " ".join(cmd[1:])
|
||||
else:
|
||||
if len(cmd) == 2:
|
||||
val = None
|
||||
else:
|
||||
val = " ".join(cmd[2:])
|
||||
if c == None:
|
||||
history_add((self.id, False, "ERROR: " + name + " not found"))
|
||||
elif val == None:
|
||||
history_add((self.id, False, name + " = " + c.value))
|
||||
else:
|
||||
set_value(c, val)
|
||||
history_add((self.id, False, "OK: " + name + " = " + c.value))
|
||||
self.send_msg(dict(type = 'accept-command'))
|
||||
|
||||
def handle_close(self):
|
||||
try:
|
||||
print 'close client', self
|
||||
del clients[self.id]
|
||||
except KeyError:
|
||||
print 'can not remove client'
|
||||
try:
|
||||
del console_clients[self.id]
|
||||
print '*** removed console client',self.id,self
|
||||
except KeyError:
|
||||
print '*** console client already closed',self.id,self
|
||||
self.close()
|
||||
|
||||
def init(device):
|
||||
global cindex, groups, actual_device
|
||||
if device != actual_device:
|
||||
print 'INIT', actual_device, device
|
||||
actual_device = device
|
||||
cindex = {} # component index
|
||||
groups = {}
|
||||
groupsfile = device + ".json"
|
||||
with open(groupsfile) as fil:
|
||||
groupdict = json.load(fil)
|
||||
for grp in groupdict:
|
||||
gdict = groupdict[grp]
|
||||
g = find_group(grp)
|
||||
g.title = gdict.get('title', grp)
|
||||
for cdict in gdict['components']:
|
||||
cname = cdict['name']
|
||||
c = Component(cname, g)
|
||||
g.components.append(c)
|
||||
cindex[cname] = c
|
||||
c.properties = dict((k, v) for k, v in cdict.items() if k != 'name' and k != 'value')
|
||||
c.value = cdict.get('value','')
|
||||
if c.properties.get('type','') == 'group':
|
||||
gg = find_group(cname)
|
||||
if 'title' in cdict:
|
||||
if gg.title == gg.name:
|
||||
gg.title = cdict['title']
|
||||
else:
|
||||
c.properties['title'] = gg.title
|
||||
devlist = [cindex[n].value for n in init_values]
|
||||
device_title = "DUMMY " + "/".join(devlist)
|
||||
print 'ENDINIT', device_title
|
||||
for m in main_update:
|
||||
if m.title != device_title:
|
||||
m.title=device_title
|
||||
m.send_msg(dict(type='id', id=m.id, title=device_title))
|
||||
|
||||
def find_group(grp):
|
||||
if grp in groups:
|
||||
g = groups[grp]
|
||||
else:
|
||||
g = Group(grp)
|
||||
groups[grp] = g
|
||||
return g
|
||||
|
||||
if __name__ == "__main__":
|
||||
console_clients = {}
|
||||
clients = {}
|
||||
cindex = {} # component index
|
||||
groups = {}
|
||||
history = deque(maxlen=50)
|
||||
console_id = 0
|
||||
main_update = []
|
||||
actual_device = ''
|
||||
|
||||
try:
|
||||
device = sys.argv[2]
|
||||
except IndexError:
|
||||
device = "dummy1"
|
||||
try:
|
||||
port = int(sys.argv[1])
|
||||
except IndexError:
|
||||
port = 5001
|
||||
|
||||
init(device)
|
||||
|
||||
server = lineserver.LineServer('localhost', port, DummyHandler)
|
||||
server.loop()
|
||||
Reference in New Issue
Block a user