Put config schema inside the main module. flyte#13

MANIFEST.in

@@ -1,3 +1,2 @@
 # http://packages.python.org/distribute/setuptools.html#including-data-files
 include *requirements.txt
-recursive-include . *.yml

pi_mqtt_gpio/__init__.py

@@ -0,0 +1,119 @@
+import yaml
+
+CONFIG_SCHEMA = yaml.load("""
+mqtt:
+  type: dict
+  required: yes
+  schema:
+    host:
+      type: string
+      empty: no
+      required: no
+      default: localhost
+    port:
+      type: integer
+      min: 1
+      max: 65535
+      required: no
+      default: 1883
+    user:
+      type: string
+      required: no
+      default: ""
+    password:
+      type: string
+      required: no
+      default: ""
+    topic_prefix:
+      type: string
+      required: no
+      default: ""
+      coerce: rstrip_slash
+    protocol:
+      type: string
+      required: no
+      empty: no
+      coerce: tostring
+      default: "3.1.1"
+      allowed:
+        - "3.1"
+        - "3.1.1"
+
+gpio_modules:
+  type: list
+  required: yes
+  schema:
+    type: dict
+    allow_unknown: yes
+    schema:
+      name:
+        type: string
+        required: yes
+        empty: no
+      module:
+        type: string
+        required: yes
+        empty: no
+
+digital_inputs:
+  type: list
+  required: no
+  default: []
+  schema:
+    type: dict
+    schema:
+      name:
+        type: string
+        required: yes
+        empty: no
+      module:
+        type: string
+        required: yes
+        empty: no
+      pin:
+        type: integer
+        required: yes
+        min: 0
+      on_payload:
+        type: string
+        required: yes
+        empty: no
+      off_payload:
+        type: string
+        required: yes
+        empty: no
+      pullup:
+        type: boolean
+        required: no
+        default: no
+      pulldown:
+        type: boolean
+        required: no
+        default: no
+
+digital_outputs:
+  type: list
+  required: no
+  default: []
+  schema:
+    type: dict
+    schema:
+      name:
+        type: string
+        required: yes
+      module:
+        type: string
+        required: yes
+      pin:
+        type: integer
+        required: yes
+        min: 0
+      on_payload:
+        type: string
+        required: no
+        empty: no
+      off_payload:
+        type: string
+        required: no
+        empty: no
+""")

pi_mqtt_gpio/server.py

@@ -10,6 +10,7 @@
 import paho.mqtt.client as mqtt
 import cerberus
 
+from pi_mqtt_gpio import CONFIG_SCHEMA
 from pi_mqtt_gpio.modules import PinPullup, PinDirection, BASE_SCHEMA
 from pi_mqtt_gpio.scheduler import Scheduler, Task
 
@@ -30,11 +31,6 @@
 OUTPUT_TOPIC = "output"
 INPUT_TOPIC = "input2"
 
-# @TODO: Don't load this at module level
-THIS_DIR = os.path.dirname(os.path.realpath(__file__))
-with open(os.path.join(THIS_DIR, "..", "config.schema.yml")) as schema:
-    CONFIG_SCHEMA = yaml.load(schema)
-
 _LOG = logging.getLogger(__name__)
 _LOG.addHandler(logging.StreamHandler())
 _LOG.setLevel(logging.DEBUG)