-
Notifications
You must be signed in to change notification settings - Fork 32
/
create_api_tester.py
176 lines (143 loc) · 7.09 KB
/
create_api_tester.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
# -*- coding: utf8 -*-
import sys
import os
import cpphelper
from jinja2 import Environment, FileSystemLoader
from os import path
def addEnumInfo(fields,typedefDict,enumDict):
'''
对一个field列表增加枚举信息
fields field的列表
typedefDict 类型定义字典,通过调用cpphelper.getTypedefDict获得
enumDict 枚举类型字典，通过cpphelper.getEnumDict获得
返回 无返回信息直接作用在fields中
'''
for field in fields:
typedef = typedefDict[ field['type'] ]
field['original'] = typedef['type']
field['len'] = typedef['len']
field['enums'] = enumDict.get(field['type'])
def main():
'''
从命令行上获取一个API名称（不含Req和OnRsp前缀），生成相关的API组合的测试代码
命令行参数1 API名称（如:QryTradingAccount，对应的请求函数为ReqQryTradingAccount，对应的相应函数为OnRspQryTradingAccount）
'''
if len(sys.argv) != 2 :
print '命令使用格式不正确,请参照以下例子:'
print '$ python create_api_tester.py API名称'
print '其中API名称形如QryTradingAccount，对应的请求函数为ReqQryTradingAccount，对应的相应函数为OnRspQryTradingAccount。'
return(0)
apiName = sys.argv[1]
print '开始生成%s的接口测试代码:...' % apiName
# 检查目录是否存在
#apiName = 'QryInstrument'
if path.exists(apiName):
print '同名(%s)目录(或文件)已经存在，请确认并修改其位置或重命名后再执行命令\n'
return(0)
# 创建目录
os.makedirs(apiName)
# 读取相关cpp头文件
ThostFtdcTraderApi_h = cpphelper.getCppHeader('include/ThostFtdcTraderApi.h',['TRADER_API_EXPORT'])
ThostFtdcUserApiStruct_h = cpphelper.getCppHeader('include/ThostFtdcUserApiStruct.h')
typedefDict = cpphelper.getTypedefDict('include/ThostFtdcUserApiDataType.h')
enumDict = cpphelper.getEnumDict('include/ThostFtdcUserApiDataType.h')
# 将TThostFtdcOffsetFlagType类型数据映射到TThostFtdcCombOffsetFlagType
# TThostFtdcCombOffsetFlagType表示多个TThostFtdcOffsetFlagType位的组合其实际字段含义在TThostFtdcOffsetFlagType中说明
enumDict['TThostFtdcCombOffsetFlagType'] = enumDict['TThostFtdcOffsetFlagType']
# 情况同TThostFtdcCombOffsetFlagType
enumDict['TThostFtdcCombHedgeFlagType'] = enumDict['TThostFtdcHedgeFlagType']
# 获取响应函数相关信息
responseMethodName = 'OnRsp%s' % apiName
CThostFtdcTraderSpi = cpphelper.getClass(ThostFtdcTraderApi_h,'CThostFtdcTraderSpi')
responseMethod = cpphelper.getClassMethod(CThostFtdcTraderSpi,'public',responseMethodName)
respParameters = cpphelper.getMethodParameters(responseMethod)
# 获取调用函数相关信息
requestMethodName = 'Req%s' % apiName
CThostFtdcTraderApi = cpphelper.getClass(ThostFtdcTraderApi_h,'CThostFtdcTraderApi')
requestMethod = cpphelper.getClassMethod(CThostFtdcTraderApi,'public',requestMethodName)
reqParameters = cpphelper.getMethodParameters(requestMethod)
# 读取报单通知回调函数信息
onRtnOrderMethodName = 'OnRtnOrder'
onRtnOrderMethod = cpphelper.getClassMethod(CThostFtdcTraderSpi,'public',onRtnOrderMethodName)
onRtnOrderParameters =cpphelper.getMethodParameters(onRtnOrderMethod)
# 读取成交通知回调函数信息
onRtnTradeMethodName = 'OnRtnTrade'
onRtnTradeMethod = cpphelper.getClassMethod(CThostFtdcTraderSpi,'public',onRtnTradeMethodName)
onRtnTradeParameters =cpphelper.getMethodParameters(onRtnTradeMethod)
# 读取入单错误回调函数信息
onErrRtnOrderInsertMethodName = 'OnErrRtnOrderInsert'
onErrRtnOrderInsertMethod = cpphelper.getClassMethod(CThostFtdcTraderSpi,'public',onErrRtnOrderInsertMethodName)
onErrRtnOrderInsertParameters =cpphelper.getMethodParameters(onErrRtnOrderInsertMethod)
# 检查请求函数的参数格式是否符合预期
methedDeclare = '%s(%s)' % (requestMethod['name'],
','.join('%s %s' % (parameter['type'],parameter['name']) for parameter in reqParameters))
if len(reqParameters) != 2 :
print '无法处理的请求API结构:%s' % methedDeclare
return(0)
if reqParameters[1]['name'] != 'nRequestID' :
print '无法处理的请求API结构:%s' % methedDeclare
return(0)
if reqParameters[0]['pointer'] == False :
print '无法处理的请求API结构:%s' % methedDeclare
return(0)
# 读取响应函数的返回的数据类型的所有字段
responseDataStruct = cpphelper.getClass(ThostFtdcUserApiStruct_h,respParameters[0]['raw_type'])
responseFields = cpphelper.getStructFields(responseDataStruct)
addEnumInfo(responseFields,typedefDict,enumDict)
# 读取请求类型的所有字段列表和原始类型
requestDataStruct = cpphelper.getClass(ThostFtdcUserApiStruct_h,reqParameters[0]['raw_type'])
requestFields = cpphelper.getStructFields(requestDataStruct)
addEnumInfo(requestFields,typedefDict,enumDict)
# 读取报单通知返回结构的所有字段列表和原始类型
onRtnOrderDataStruct = cpphelper.getClass(ThostFtdcUserApiStruct_h,onRtnOrderParameters[0]['raw_type'])
onRtnOrderFields = cpphelper.getStructFields(onRtnOrderDataStruct)
addEnumInfo(onRtnOrderFields,typedefDict,enumDict)
# 读取成交通知返回结构的所有字段列表和原始类型
onRtnTradeDataStruct = cpphelper.getClass(ThostFtdcUserApiStruct_h,onRtnTradeParameters[0]['raw_type'])
onRtnTradeFields = cpphelper.getStructFields(onRtnTradeDataStruct)
addEnumInfo(onRtnTradeFields,typedefDict,enumDict)
# 读取订单录入错误信息回报消息的字段列表和原始类型
onErrRtnOrderInsertDataStruct = cpphelper.getClass(ThostFtdcUserApiStruct_h,onErrRtnOrderInsertParameters[0]['raw_type'])
onErrRtnOrderInsertFields = cpphelper.getStructFields(onErrRtnOrderInsertDataStruct)
addEnumInfo(onErrRtnOrderInsertFields,typedefDict,enumDict)
# 生成模板所需的信息集
data = {
'apiName' : apiName,
# 响应函数
'responseMethod' : responseMethod,
'respParameters' : respParameters,
'responseFields' : responseFields,
# 请求函数
'requestMethod' : requestMethod,
'reqParameters' : reqParameters,
'requestFields' : requestFields,
# 报单通知函数
'onRtnOrderMethod' : onRtnOrderMethod,
'onRtnOrderParameters' : onRtnOrderParameters,
'onRtnOrderFields' : onRtnOrderFields,
# 成交通知函数
'onRtnTradeMethod' : onRtnTradeMethod,
'onRtnTradeParameters' : onRtnTradeParameters,
'onRtnTradeFields' : onRtnTradeFields,
# 订单错误回报函数
'onErrRtnOrderInsertMethod' : onErrRtnOrderInsertMethod,
'onErrRtnOrderInsertParameters' : onErrRtnOrderInsertParameters,
'onErrRtnOrderInsertFields' : onErrRtnOrderInsertFields
}
# 读取模板信息
env = Environment(loader=FileSystemLoader('templates'))
# 生成代码文件
source = env.get_template('source.tpl.cpp')
sourcePath = path.join(apiName,'%s.cpp' % apiName)
with open(sourcePath, 'w') as f :
# 间模板生成到文件
f.write(source.render(**data).encode('utf-8'))
# 使用astyle对文件进行格式化
os.system('astyle %s' % sourcePath)
# 生成make文件
makefile = env.get_template('Makefile.tpl')
makefilePath = path.join(apiName,'Makefile')
with open(makefilePath, 'w') as f :
f.write(makefile.render(**data).encode('utf-8'))
if __name__ == '__main__':
main()