Add initial version

master
0xee 2019-11-25 16:59:08 +01:00
commit 7d2d89aabd
4 changed files with 67 additions and 0 deletions

10
default.nix Normal file
View File

@ -0,0 +1,10 @@
{ python3 }:
with python3.pkgs;
buildPythonPackage rec {
name = "mqtt-config";
src = ./.;
propagatedBuildInputs = [];
buildInputs = [ pyyaml paho-mqtt ];
doCheck = false;
shellHook = "";
}

44
mqtt-config Executable file
View File

@ -0,0 +1,44 @@
#!/usr/bin/env python
import argparse
import yaml
import paho.mqtt.publish
def read_config(config_dict, base_topic=''):
ret = {}
for elem in config_dict:
if 'topic' not in elem:
raise Exception("Invalid config")
sep = '/' if base_topic else ''
topic = '{}{}{}'.format(base_topic.rstrip("/"), sep,
elem["topic"].lstrip("/"))
if 'children' in elem:
ret.update(read_config(elem['children'], base_topic=topic))
if 'value' in elem:
ret[topic] = elem['value']
return ret
def main():
options = argparse.ArgumentParser()
options.add_argument('--broker', '-b', default='localhost')
options.add_argument('--port', '-p', default='1883', type=int)
options.add_argument('config')
args = options.parse_args()
with open(args.config, 'r') as f:
config_dict = yaml.load(f, Loader=yaml.Loader)
topics = read_config(config_dict)
msgs = [(topic, value, 1, True) for topic, value in topics.items()]
paho.mqtt.publish.multiple(msgs,
hostname=args.broker,
port=args.port,
client_id='configuration')
if __name__ == '__main__':
main()

11
setup.py Normal file
View File

@ -0,0 +1,11 @@
#!/usr/bin/env python
from distutils.core import setup
setup(name='mqtt-config',
version='0.1',
description='MQTT Configurator',
author='0xee',
author_email='mqtt-config@0xee.eu',
packages=[],
scripts=['mqtt-config'])

2
shell.nix Normal file
View File

@ -0,0 +1,2 @@
{ pkgs ? import <nixpkgs> {} }:
pkgs.callPackage ./. {}