python處理apiDoc轉swagger
現在我需要轉換的介面全是nodejs寫的資料，而且均為post傳輸的json格式介面
apiDoc程式碼中的格式如下：
/**
* @api {方法} 路徑 標題
* @apiGroup Group
* @apiDescription 描述這個API的資訊
*
* @apiParam {String} userName 使用者名稱
* @apiParamExample {json} request-example
* {
* "userName": "Eve"
* }
*
* @apiError {String} message 錯誤資訊
* @apiErrorExample {json} error-example
* {
* "message": "使用者名稱不存在"
* }
*
*
* @apiSuccess {String} userName 使用者名稱
* @apiSuccess {String} createTime 建立時間
* @apiSuccess {String} updateTime 更新時間
* @apiSuccessExample {json} success-example
* {
* "userName": "Eve",
* "createTime": "1568901681"
* "updateTime": "1568901681"
* }
*/function getUserInfo(username) {
// 假如這個函數是根據使用者名稱返回使用者資訊的
}
使用npm安裝apidoc外掛：
npm install apidoc
再新建對應的apidoc.json，格式如下：
{
"name": "檔名",
"version": "版本號",
"description": "解釋",
"title": "標題",
"url" : "地址"
}
然後在apidoc.json路徑下執行命令可以生成介面檔案（src是介面程式碼檔案夾，apidoc是生成檔案的資料夾）：
apidoc -i src/ -o apidoc/
生成後可以在apidoc資料夾中開啟index.html檢視生成的介面檔案，生成檔案時會生成一個api_data.json，下面會用到
這裡我們暫時只需要關注引數為json的介面格式
{
"swagger": "2.0",
"info": {
"description": "1.0版本介面檔案",
"version": "1.0.5",
"title": "智慧醫療輔助平臺",
"termsOfService": "http://swagger.io/terms/"
},
"host": "http://localhost:8080",
"basePath": "/",
"tags": [],
"paths": {},
"definitions": {}
}
其中path是存放介面的，tags是存放的分組名列表，definitions是實體列表（json引數）
使用apidoc包生成apidoc的json格式資料，然後使用python讀取出介面地址、名字、組名、輸入引數格式和例子、輸出引數格式和例子等，然後根據swagger格式填入對應的資料即可生成swagger的json格式
我的話是會直接使用處理出的swagger的json格式的資料匯入yApi中
程式碼雖然在下面，但是是我臨時著急用寫的，有的地方是寫死的，需要改，這裡放出來主要是講個大致的思路
import re
import json
import demjson
import decimal
# 儲存時會出現byte格式問題，使用這個處理
class DecimalEncoder(json.JSONEncoder):
def default(self, o):
if isinstance(o, decimal.Decimal):
return float(o)
super(DecimalEncoder, self).default(o)
# 分析例子轉json，在這裡可以自己新增規則
def analyze_demjson(json_data):
item = json_data.replace("\\n", "").replace("\\", "").replace(" ", "")
result_item = {}
try:
result_item = demjson.decode(item, encoding='UTF-8')
except:
print(item)
return result_item
# 獲取解析apidoc資料
def get_api_doc_data(name):
data_list = None
group_list = {}
with open(name, mode='r', encoding="UTF-8") as f:
data_list = json.load(f)
for data in data_list:
if data['group'] in group_list:
group_list[data['group']].append(data)
else:
group_list[data['group']] = [data]
return group_list
# 轉為swagger寫入
def set_swagger_data(data):
swagger_json = {
"swagger": "2.0",
"info": {
"description": "1.0版本介面檔案",
"version": "1.0.5",
"title": "智慧醫療輔助平臺",
"termsOfService": "http://swagger.io/terms/"
},
"host": "http://localhost:8080",
"basePath": "/",
"tags": [],
"paths": {},
"definitions": {}
}
# 新增分組
for group_key in data:
swagger_json['tags'].append({
"name": group_key,
"description": group_key
})
# 新增介面資訊
# 迴圈分組
for group_key in data:
# 迴圈每組列表
for interface in data[group_key]:
parameters = {}
if 'parameter' in interface and 'fields' in interface['parameter']:
# 獲取引數demo資訊
content = ""
if 'examples' in interface['parameter']:
content = analyze_demjson(interface['parameter']['examples'][0]['content'])
# 新增引數資訊
parameter_dict = {}
for parameter in interface['parameter']['fields']['Parameter']:
parameter_type = "None"
if "type" in parameter:
parameter_type = parameter['type'].lower()
if parameter_type == 'number':
parameter_type = "integer"
parameter_item = {
"description": parameter['description'].replace('<p>', '').replace('</p>', ''),
"required": parameter['optional'],
"type": parameter_type,
"default": ''
}
if parameter['field'] in content:
parameter_item['default'] = content[parameter['field']]
parameter_dict[parameter['field']] = parameter_item
parameters = {
"in": "body",
"name": interface['name'],
"description": interface['name'],
"required": "true",
"schema": {
"originalRef": interface['name'],
"$ref": "#/definitions/" + interface['name']
}
}
swagger_json['definitions'][interface['name']] = {
"type": "object",
"properties": parameter_dict
}
# 新增返回資訊
responses = {
"200": {
"description": "successful operation",
"schema": {
"originalRef": interface['name'] + "_response",
"$ref": "#/definitions/" + interface['name'] + "_response"
}
}
}
schema = {
"type": "object",
"properties": {
"errcode": {
"type": "integer",
"default": 0,
"description": "編碼，成功返回1"
},
"data": {
"type": "object",
"default": {},
"description": "監管物件明細,包含表頭和資料內容兩部分"
},
"errmsg": {
"type": "string",
"default": "ok",
"description": '編碼提示資訊，成功時返回 "ok"'
}
}
}
# 返回例子
if "success" in interface:
response_example = ""
if len(interface['success']['examples']) == 1:
response_example = analyze_demjson(interface['success']['examples'][0]['content'])
else:
response_example = analyze_demjson(interface['success']['examples']['content'])
if 'data' in response_example and response_example['data'] != {}:
schema['properties']['data'] = response_example['data']
swagger_json['definitions'][interface['name'] + "_response"] = schema
# 加入
swagger_json['paths'][interface['url']] = {
interface['type']: {
"tags": [group_key],
"summary": interface['title'].replace(interface['url'] + '-', ''),
"description": interface['title'],
"consumes": [
"application/json"
],
"produces": [
"application/json"
],
"parameters": [parameters],
"responses": responses
}}
# 寫入json檔案
with open('swagger_data.json', 'w', encoding="UTF-8") as json_file:
json.dump(swagger_json, json_file, cls=DecimalEncoder, indent=4, ensure_ascii=False)
if __name__ == '__main__':
group_data = get_api_doc_data('api_data.json')
set_swagger_data(group_data)