pusher script moved to its own module and entrypoint

This commit is contained in:
Fabio Manganiello 2017-12-11 23:09:45 +01:00
parent 00e9965305
commit 7a58add0ad
4 changed files with 84 additions and 111 deletions

View file

@ -126,7 +126,7 @@ def parse_config_file(config_file=None):
def get_backends(config): def get_backends(config):
backends = [] backends = {}
for k in config.keys(): for k in config.keys():
if k.startswith('backend.'): if k.startswith('backend.'):
@ -143,7 +143,8 @@ def get_backends(config):
try: try:
b = getattr(module, cls_name)(config[k]) b = getattr(module, cls_name)(config[k])
backends.append(b) name = '.'.join((k.split('.'))[1:])
backends[name] = b
except AttributeError as e: except AttributeError as e:
logging.warn('No such class in {}: {}'.format( logging.warn('No such class in {}: {}'.format(
module.__name__, cls_name)) module.__name__, cls_name))
@ -151,6 +152,15 @@ def get_backends(config):
return backends return backends
def get_default_pusher_backend(config):
backends = ['.'.join((k.split('.'))[1:])
for k in config.keys() if k.startswith('backend.')
and 'pusher' in config[k] and config[k]['pusher'] is True]
return backends[0] if backends else None
def get_device_id(): def get_device_id():
global config global config
return config['device_id'] return config['device_id']
@ -196,7 +206,7 @@ Usage: {} [-v] [-h] [-c <config_file>]
mq = Queue() mq = Queue()
backends = get_backends(config) backends = get_backends(config)
for backend in backends: for backend in backends.values():
backend.mq = mq backend.mq = mq
backend.start() backend.start()

View file

@ -1,105 +0,0 @@
#!python
import argparse
import json
import os
import re
import sys
import yaml
from platypush import parse_config_file
from platypush.backend.kafka import KafkaBackend
from platypush.backend.local import LocalBackend
from platypush.backend.pushbullet import PushbulletBackend
def print_usage():
print ('''Usage: {} [-h|--help] <-t|--target <target name>> <-a|--action <action name>> payload
-h, --help:\t\tShow this help and exit
-b, --backend:\tBackend to deliver the message [pushbullet|kafka] (default: whatever specified in your config with pusher=True)
-t, --target:\tName of the target device/host
-a, --action\tAction to run, it includes both the package name and the method (e.g. shell.exec or music.mpd.play)
payload:\t\tArguments to the action
'''.format(sys.argv[0]))
def get_pb_device_by_name(pb, name):
devices = [
_ for _ in pb.devices if _.nickname == name
]
return devices[0] if devices else None
def get_backend(config):
if 'backend.local' in config \
and 'pusher' in config['backend.local'] \
and config['backend.local']['pusher']:
c = config['backend.local']
return LocalBackend({'fifo': c['fifo']})
elif 'backend.pushbullet' in config \
and 'pusher' in config['backend.pushbullet'] \
and config['backend.pushbullet']['pusher']:
c = config['backend.pushbullet']
return PushbulletBackend({'token':c['token'], 'device':c['device']})
elif 'backend.kafka' in config \
and 'pusher' in config['backend.kafka'] \
and config['backend.kafka']['pusher']:
c = config['backend.kafka']
return KafkaBackend({'server':c['server'], 'topic':c['topic']})
def main():
config = parse_config_file()
parser = argparse.ArgumentParser()
parser.add_argument('--target', '-t', dest='target', required=True,
help="Destination of the command")
parser.add_argument('--action', '-a', dest='action', required=True,
help="Action to execute, as package.method")
parser.add_argument('--backend', '-b', dest='backend', required=False,
help="Backend to deliver the message " +
"[pushbullet|kafka|local] (default: whatever " +
"specified in your config with pusher=True)")
opts, args = parser.parse_known_args(sys.argv[1:])
payload = {}
if len(args) % 2 != 0:
raise RuntimeError('Odd number of key-value options passed: {}'.
format(args))
if opts.target == 'localhost' and 'backend.local' in config:
cfg = config['backend.local']
cfg['pusher'] = True
config = { 'backend.local': cfg }
if opts.backend:
backend_cfg_name = 'backend.' + opts.backend
if backend_cfg_name not in config:
raise RuntimeError('{} backend specified, but no configuration ' +
'for it available'.format(backend_cfg_name))
cfg = config[backend_cfg_name]
cfg['pusher'] = True
config = { backend_cfg_name: cfg }
backend = get_backend(config)
for i in range(0, len(args), 2):
payload[re.sub('^-+', '', args[i])] = args[i+1]
msg = {
'target': opts.target,
'action': opts.action,
**payload,
}
print('msg: {}'.format(msg))
backend.send_msg(msg)
if __name__ == '__main__':
main()
# vim:sw=4:ts=4:et:

70
platypush/pusher/__init__.py Executable file
View file

@ -0,0 +1,70 @@
import argparse
import re
import sys
from platypush import get_backends, get_default_pusher_backend, parse_config_file
def print_usage():
print ('''Usage: {} [-h|--help] <-t|--target <target name>> <-a|--action <action name>> payload
-h, --help:\t\tShow this help and exit
-b, --backend:\tBackend to deliver the message [pushbullet|kafka] (default: whatever specified in your config with pusher=True)
-t, --target:\tName of the target device/host
-a, --action\tAction to run, it includes both the package name and the method (e.g. shell.exec or music.mpd.play)
payload:\t\tArguments to the action
'''.format(sys.argv[0]))
def pusher(target, action, backend=None, **kwargs):
config = parse_config_file()
msg = {
'target': target,
'action': action,
**kwargs,
}
if target == 'localhost':
backend = 'local'
elif not backend:
backend = get_default_pusher_backend(config)
backends = get_backends(config)
if backend not in backends:
raise RuntimeError('No such backend configured: {}'.format(backend))
b = backends[backend]
b.send_msg(msg)
def main():
parser = argparse.ArgumentParser()
parser.add_argument('--target', '-t', dest='target', required=True,
help="Destination of the command")
parser.add_argument('--action', '-a', dest='action', required=True,
help="Action to execute, as package.method")
parser.add_argument('--backend', '-b', dest='backend', required=False,
help="Backend to deliver the message " +
"[pushbullet|kafka|local] (default: whatever " +
"specified in your config with pusher=True)")
opts, args = parser.parse_known_args(sys.argv[1:])
if len(args) % 2 != 0:
raise RuntimeError('Odd number of key-value options passed: {}'.
format(args))
payload = {}
for i in range(0, len(args), 2):
payload[re.sub('^-+', '', args[i])] = args[i+1]
pusher(target=opts.target, action=opts.action,
backend=opts.backend if 'backend' in opts else None, **payload)
if __name__ == '__main__':
main()
# vim:sw=4:ts=4:et:

View file

@ -40,13 +40,11 @@ setup(
python_requires = '>= 3', python_requires = '>= 3',
keywords = "pushbullet notifications automation", keywords = "pushbullet notifications automation",
url = "https://github.com/BlackLight/platypush", url = "https://github.com/BlackLight/platypush",
# packages = ['platypush'],
packages = find_packages(), packages = find_packages(),
# package_data = {'': plugins},
scripts = ['platypush/bin/pusher'],
entry_points = { entry_points = {
'console_scripts': [ 'console_scripts': [
'platypush=platypush:main', 'platypush=platypush:main',
'pusher=platypush.pusher:main',
], ],
}, },
data_files = [ data_files = [