From 4ea5da44d010168a4d46072f24e00484af41cb60 Mon Sep 17 00:00:00 2001
From: Kegan Dougal <kegan@matrix.org>
Date: Thu, 21 May 2015 14:11:29 +0100
Subject: [PATCH] Include enum information.

---
 templating/internal/units.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/templating/internal/units.py b/templating/internal/units.py
index 328a9f07..c8d47d98 100644
--- a/templating/internal/units.py
+++ b/templating/internal/units.py
@@ -82,6 +82,9 @@ def _load_schemas():
                     value_type = "[%s]" % props[key_name]["items"]["type"]
             else:
                 value_type = props[key_name]["type"]
+                if props[key_name].get("enum"):
+                    value_type = "enum"
+                    desc += " One of: %s" % json.dumps(props[key_name]["enum"])
 
             fields["rows"].append({
                 "key": key_name,