From 37b18e9297daa0fd4c4a70755fbf6c8fa564abf8 Mon Sep 17 00:00:00 2001 From: aozhiwei Date: Wed, 17 Nov 2021 16:37:33 +0800 Subject: [PATCH] 1 --- bin/mydoc | 5 ++++- scripts/mydoc/mydoc.py | 26 +++++++++++++++++++++++++- 2 files changed, 29 insertions(+), 2 deletions(-) diff --git a/bin/mydoc b/bin/mydoc index 6da9a5a..7678210 100755 --- a/bin/mydoc +++ b/bin/mydoc @@ -8,4 +8,7 @@ import mydoc workDir = os.getcwd() doc = mydoc.MyDoc(workDir) -doc.convert() +if len(sys.argv) > 1 and sys.argv[1] == '-g': + doc.generate() +else: + doc.convert() diff --git a/scripts/mydoc/mydoc.py b/scripts/mydoc/mydoc.py index f7e8e78..7e17332 100644 --- a/scripts/mydoc/mydoc.py +++ b/scripts/mydoc/mydoc.py @@ -130,5 +130,29 @@ class MyDoc(object): outRsp(item, 1) else: [outRsp(p, 1) for p in item.fields] - self._wf.write(' * }\n') + self._wf.write(' * }\n') self._wf.write(' */\n') + + def generate(self): + scheme = json.loads(open(self._workDir + 'apilist.json', 'r').read()) + for key, val in scheme['apis'].items(): + wf = open(self._workDir + key.replace('Controller', '') + '.py', 'w') + wf.write('# -*- coding: utf-8 -*-\n\n') + wf.write('import _common\n\n') + wf.write('class %s(object):\n\n' % (key.replace('Controller', ''))) + wf.write(' def __init__(self):\n' % ()) + wf.write(' self.apis = [\n' % ()) + for api in val: + wf.write(" {\n" % ()) + wf.write(" 'desc': '%s',\n" % (api)) + wf.write(" 'group': '%s',\n" % (key.replace('Controller', ''))) + wf.write(" 'url': 'webapp/index.php?c=%s&a=%s',\n" % (key.replace('Controller', ''), api)) + wf.write(" 'params': [\n" % ()) + wf.write(" _common.ReqHead(),\n" % ()) + wf.write(" ],\n" % ()) + wf.write(" 'response': [\n" % ()) + wf.write(" _common.RspHead(),\n" % ()) + wf.write(" ]\n" % ()) + wf.write(" },\n" % ()) + wf.write(' ]\n' % ()) + wf.close()