125 lines
2.9 KiB
Python
125 lines
2.9 KiB
Python
import click
|
|
import json
|
|
import matplotlib
|
|
matplotlib.use('Agg') # don't show windows
|
|
import matplotlib.pyplot as plt
|
|
import networkx as nx
|
|
import os
|
|
|
|
from x import resource as xr
|
|
from x import signals as xs
|
|
|
|
|
|
@click.group()
|
|
def cli():
|
|
pass
|
|
|
|
|
|
def init_cli_resource():
|
|
@click.group()
|
|
def resource():
|
|
pass
|
|
|
|
cli.add_command(resource)
|
|
|
|
@click.command()
|
|
@click.argument('name')
|
|
@click.argument('base_path')
|
|
@click.argument('dest_path')
|
|
@click.argument('args')
|
|
def create(args, dest_path, base_path, name):
|
|
print 'create', name, base_path, dest_path, args
|
|
args = json.loads(args)
|
|
xr.create(name, base_path, dest_path, args)
|
|
|
|
resource.add_command(create)
|
|
|
|
@click.command()
|
|
@click.argument('path')
|
|
def show(path):
|
|
print xr.load(path)
|
|
|
|
resource.add_command(show)
|
|
|
|
@click.command()
|
|
@click.argument('path')
|
|
@click.argument('args')
|
|
def update(args, path):
|
|
print 'Update', path, args
|
|
args = json.loads(args)
|
|
# Need to load all resources for bubbling effect to take place
|
|
# TODO: resources can be scattered around, this is a simple
|
|
# situation when we assume resources are all in one directory
|
|
base_path, name = os.path.split(path)
|
|
all = xr.load_all(base_path)
|
|
r = all[name]
|
|
r.update(args)
|
|
|
|
resource.add_command(update)
|
|
|
|
|
|
def init_cli_connect():
|
|
@click.command()
|
|
@click.argument('emitter')
|
|
@click.argument('receiver')
|
|
@click.option('--mapping', default=None)
|
|
def connect(mapping, receiver, emitter):
|
|
print 'Connect', emitter, receiver
|
|
emitter = xr.load(emitter)
|
|
receiver = xr.load(receiver)
|
|
print emitter
|
|
print receiver
|
|
if mapping is not None:
|
|
mapping = json.loads(mapping)
|
|
xs.connect(emitter, receiver, mapping=mapping)
|
|
|
|
cli.add_command(connect)
|
|
|
|
@click.command()
|
|
@click.argument('emitter')
|
|
@click.argument('receiver')
|
|
def disconnect(receiver, emitter):
|
|
print 'Disconnect', emitter, receiver
|
|
emitter = xr.load(emitter)
|
|
receiver = xr.load(receiver)
|
|
print emitter
|
|
print receiver
|
|
xs.disconnect(emitter, receiver)
|
|
|
|
cli.add_command(disconnect)
|
|
|
|
|
|
def init_cli_connections():
|
|
@click.group()
|
|
def connections():
|
|
pass
|
|
|
|
cli.add_command(connections)
|
|
|
|
@click.command()
|
|
def show():
|
|
print json.dumps(xs.CLIENTS, indent=2)
|
|
|
|
connections.add_command(show)
|
|
|
|
# TODO: this requires graphing libraries
|
|
@click.command()
|
|
def graph():
|
|
g = xs.connection_graph()
|
|
pos = nx.spring_layout(g)
|
|
nx.draw_networkx_nodes(g, pos)
|
|
nx.draw_networkx_edges(g, pos, arrows=True)
|
|
nx.draw_networkx_labels(g, pos)
|
|
plt.axis('off')
|
|
plt.savefig('graph.png')
|
|
|
|
connections.add_command(graph)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
init_cli_resource()
|
|
init_cli_connect()
|
|
init_cli_connections()
|
|
|
|
cli()
|