From 649af4d749250ddc818164f20384a4dea67bb30a Mon Sep 17 00:00:00 2001
From: Tony Mack <tmack@paris.CS.Princeton.EDU>
Date: Mon, 10 Dec 2012 21:40:14 -0500
Subject: [PATCH] use prepare_response() for format json response

---
 PLC/RestAPI.py | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/PLC/RestAPI.py b/PLC/RestAPI.py
index 1e4f8ed5..e82ea3a8 100644
--- a/PLC/RestAPI.py
+++ b/PLC/RestAPI.py
@@ -16,6 +16,10 @@ get_response = """
 </body></html>
 """
 
+def prepare_response(result):
+    response = {'response': result}
+    return jsonify(response)
+
 @app.route('/')
 def api_root():
     return 'Welcome'
@@ -24,13 +28,15 @@ def api_root():
 def AuthCheck():
     if request.method == 'POST':
         args = [request.json.get('auth')]
-        return api.call(None, 'AuthCheck', *args)
+        result = api.call(request.remote_addr, 'AuthCheck', *args)
+        return prepare_response(result)
     return get_response
       
 @app.route('/roles', methods=['GET', 'POST'])
 def GetRoles():
     if request.method == 'POST':
-        logger.error(request.data) 
-        return jsonify(**request.json)
+        args = [request.json.get('auth')]
+        result = api.call(request.remote_addr, 'AuthCheck', *args)
+        return jsonify(result)
     return get_response 
 
-- 
2.47.0