codegen.py now works with latest rabbit-codegen

1 files changed, 3 insertions, 1 deletions

diff --git a/codegen.py b/codegen.py
index 6fcc67d0..50804990 100644
--- a/codegen.py
+++ b/codegen.py
@@ -516,5 +516,7 @@ def generateSpec(specPath):
     genSpec(AmqpSpec(specPath))
 
 if __name__ == "__main__":
-    do_main(generateHrl, generateSpec, generateErl)
+    do_main_dict({"header": generateHrl,
+                  "spec": generateSpec,
+                  "body": generateErl})