From bd66322164b0629a3afddc6dc637312bb5f27bc6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=88=98=E8=96=AA?= <1537749411@qq.com>
Date: Fri, 20 May 2022 15:38:25 +0800
Subject: [PATCH] add parameters

---
 .../industrialai/api/controllers/root.py             |  6 +++++-
 .../pecan_swagger/decorators.py                      | 12 ++++++++++++
 2 files changed, 17 insertions(+), 1 deletion(-)

diff --git a/examples/industrial-ai-apiserver/industrialai/api/controllers/root.py b/examples/industrial-ai-apiserver/industrialai/api/controllers/root.py
index aefd782..1450b12 100644
--- a/examples/industrial-ai-apiserver/industrialai/api/controllers/root.py
+++ b/examples/industrial-ai-apiserver/industrialai/api/controllers/root.py
@@ -35,7 +35,11 @@ class registryForm:
 class RootController(object):
 
     _custom_actions = {
-        'test': ['GET']
+        'test': ['GET'],
+        'signin': ['POST'],
+        'pwdcheck': ['POST'],
+        'myself': ['POST'],
+        'index': ['POST'],
     }
 
     @expose('json')
diff --git a/examples/industrial-ai-apiserver/pecan_swagger/decorators.py b/examples/industrial-ai-apiserver/pecan_swagger/decorators.py
index 5b8ef4a..7d3d026 100644
--- a/examples/industrial-ai-apiserver/pecan_swagger/decorators.py
+++ b/examples/industrial-ai-apiserver/pecan_swagger/decorators.py
@@ -36,3 +36,15 @@ def method(method):
         m.__swag = dict(method=method)
         return m
     return decorator
+
+
+def parameter(parameter,*args):
+    def decorator(m):
+        if not m.__swag['method'][parameter].get('parameter'):
+            m.__swag['method'][parameter]['parameter']=[]
+        for i in args:
+            if i.get('schema'):
+                i['schema']={'$ref':'#/definitions/'+i['schema']}
+            m.__swag['method'][parameter]['parameter'].append(i)
+        return m
+    return decorator
\ No newline at end of file
-- 
Gitee