diff --git a/pecan-swagger/examples/industrial-ai-apiserver/industrialai/api/controllers/root.py b/pecan-swagger/examples/industrial-ai-apiserver/industrialai/api/controllers/root.py index aefd782d67a5ca1c99d42d13e999710ebfad57ce..1450b12304027c5949ddedde3690ea40709c4276 100644 --- a/pecan-swagger/examples/industrial-ai-apiserver/industrialai/api/controllers/root.py +++ b/pecan-swagger/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/pecan-swagger/pecan_swagger/decorators.py b/pecan-swagger/pecan_swagger/decorators.py index d84bf64e4a815096452151b023a38c5fe47d2ac5..3be6739baecfb64f6badc0470facf068b70aa6b1 100644 --- a/pecan-swagger/pecan_swagger/decorators.py +++ b/pecan-swagger/pecan_swagger/decorators.py @@ -54,4 +54,15 @@ def description(method, des): m.__swag['method'][method]['description'] = des 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