Moved setup metadata to setup.cfg

.gitignore

@@ -2,3 +2,4 @@ venv
 .idea
 /test/clients.json
 *.pyc
+*.egg-info

pyproject.toml

@@ -0,0 +1,3 @@
+[build-system]
+requires = ["setuptools", "wheel"]
+build-backend = "setuptools.build_meta"

setup.cfg

@@ -0,0 +1,19 @@
+[metadata]
+name = torchsub
+version = 0.0.1
+author = Benjamin Dweck
+author_email = bjdweck@gmail.com
+url = http://git.rudefox.io/bj/torch-subscriber-simple
+classifiers =
+    Programming Language :: Python :: 3
+    License :: OSI Approved :: MIT License
+
+[options]
+packages = find:
+install_requires = paho-mqtt~=1.5.1
+
+[options.entry_points]
+console_scripts = torch-sub=torchsub.torch_sub:main
+
+[options.packages.find]
+exclude=test

setup.py

@@ -1,16 +1,3 @@
-from setuptools import setup
+import setuptools
 
-setup(
-    name='torch-subscriber-simple',
-    version='0.0.1',
-    packages=['torch_sub', 'torch_sub.test'],
-    url='http://git.rudefox.io/bj/torch-subscriber-simple',
-    license='MIT',
-    author='Benjamin Dweck',
-    author_email='bjdweck@gmail.com',
-    description='',
-    classifiers=[
-        "Programming Language :: Python :: 3",
-        "License :: OSI Approved :: MIT License",
-    ],
-)
+setuptools.setup()

test/test_integration.py

@@ -10,7 +10,7 @@ from unittest.case import TestCase
 
 import paho.mqtt.client as mqtt
 
-from torch_sub import torch_sub
+from torchsub import torch_sub
 
 broker_hostname = "mqtt.example.com"
 broker_port = 8883

torch_sub/torch_sub.py

@@ -1,39 +0,0 @@
-import json
-import os
-import threading
-
-import paho.mqtt.subscribe as mqtt
-
-database_filename = "clients.json"
-database_lock = threading.Lock()
-
-# noinspection PyUnusedLocal
-def update_client_record(client, userdata, message):
-
-    database_lock.acquire()
-
-    if not os.path.exists(database_filename):
-        with open(database_filename, 'w') as database_blank:
-            json.dump({}, database_blank)
-
-    with open(database_filename, 'r') as infile:
-        database = json.load(infile)
-
-    payload = message.payload.decode('utf-8')
-    response = json.loads(payload)
-
-    database[response['clientId']] = response
-
-    with open(database_filename, 'w') as outfile:
-        json.dump(database, outfile)
-
-    database_lock.release()
-
-
-def subscribe(broker_hostname, broker_port, topic="torch", tls=None, auth=None):
-    mqtt.callback(update_client_record,
-                  topic,
-                  hostname=broker_hostname,
-                  port=broker_port,
-                  tls=tls,
-                  auth=auth)

torchsub/torch_sub.py

@@ -0,0 +1,108 @@
+import argparse
+import configparser
+import json
+import os
+import sys
+import threading
+
+import paho.mqtt.subscribe as mqtt
+
+database_filename = "clients.json"
+database_lock = threading.Lock()
+
+
+# noinspection PyUnusedLocal
+def update_client_record(client, userdata, message):
+    database_lock.acquire()
+
+    if not os.path.exists(database_filename):
+        with open(database_filename, 'w') as database_blank:
+            json.dump({}, database_blank)
+
+    with open(database_filename, 'r') as infile:
+        database = json.load(infile)
+
+    payload = message.payload.decode('utf-8')
+    response = json.loads(payload)
+
+    database[response['clientId']] = response
+
+    with open(database_filename, 'w') as outfile:
+        json.dump(database, outfile)
+
+    database_lock.release()
+
+
+def subscribe(broker_hostname, broker_port, topic="torch", tls=None):
+    mqtt.callback(update_client_record,
+                  topic,
+                  hostname=broker_hostname,
+                  port=broker_port,
+                  tls=tls)
+
+
+class Config:
+    def __init__(self):
+        self.broker_hostname = None
+        self.broker_port = None
+        self.topic = None
+        self.tls = None
+        parser = self.do_cli_argument_parsing()
+        (config_path, config_filename) = self.get_config_path(parser.parse_args())
+        print("Using torch configuration path: " + config_path)
+        self.read_configuration_file(config_path, config_filename)
+
+    @staticmethod
+    def do_cli_argument_parsing():
+        parser = argparse.ArgumentParser(description='Simple Torch Flat-file Database Subscriber')
+        parser.add_argument('--config-dir',
+                            nargs='?',
+                            dest='configPath',
+                            default='/etc/torch',
+                            help='configuration directory (default: /etc/torch)')
+        parser.add_argument('--config-filename',
+                            nargs='?',
+                            dest='configFilename',
+                            default='torch-sub.conf',
+                            help='configuration filename (default: torch-sub.conf)')
+        return parser
+
+    @staticmethod
+    def get_config_path(args):
+        config_path = args.configPath
+        if "TORCH_CONFIG_DIR" in os.environ:
+            config_path = os.environ.get("TORCH_CONFIG_DIR")
+        if not config_path.endswith("/"):
+            config_path = config_path + "/"
+        return config_path, args.configFilename
+
+    def read_configuration_file(self, config_dir, config_filename):
+        config_file_path = os.path.join(config_dir, config_filename)
+        if not os.path.exists(config_file_path):
+            print("Unable to locate config file at '%s'" % config_file_path)
+            sys.exit(1)
+        config = configparser.ConfigParser()
+        config.read(config_file_path)
+        mqtt_config = config['mqtt']
+        self.broker_hostname = mqtt_config.get('BrokerHost', fallback="localhost")
+        self.broker_port = mqtt_config.getint('BrokerPort', fallback=1883)
+        self.topic = mqtt_config.get('Topic', fallback="torch/+/onion_url")
+        require_certificate = mqtt_config.getboolean('RequireCertificate', fallback=False)
+        ca_file = config_dir + mqtt_config.get('CaFile', "")
+        cert_file = config_dir + mqtt_config.get('CertFile', "")
+        key_file = config_dir + mqtt_config.get('KeyFile', "")
+        self.tls = None
+        if require_certificate:
+            self.tls = {
+                'ca_certs': ca_file,
+                'certfile': cert_file,
+                'keyfile': key_file
+            }
+
+
+def main():
+    config = Config()
+    subscribe(config.broker_hostname,
+              config.broker_port,
+              config.topic,
+              config.tls)