ctpbee-opt-api 0.1.0__cp310-cp310-win_amd64.whl

ctpbee_api/ctp_mini/generator/generate_data_type.py

@@ -0,0 +1,76 @@
+""""""
+
+TYPE_CPP2PY = {
+    "int": "int",
+    "char": "char",
+    "double": "double",
+    "short": "int",
+}
+
+
+class DataTypeGenerator:
+    """DataType生成器"""
+
+    def __init__(self, filename: str, prefix: str):
+        """Constructor"""
+        self.filename = filename
+        self.prefix = prefix
+
+    def run(self):
+        """主函数"""
+        self.f_cpp = open(self.filename, "r")
+        self.f_define = open(f"{self.prefix}_constant.py", "w")
+        self.f_typedef = open(f"{self.prefix}_typedef.py", "w")
+
+        for line in self.f_cpp:
+            self.process_line(line)
+
+        self.f_cpp.close()
+        self.f_define.close()
+        self.f_typedef.close()
+
+        print("DataType生成完毕")
+
+    def process_line(self, line: str):
+        """处理每行"""
+        line = line.replace("\n", "")
+        line = line.replace(";", "")
+
+        if line.startswith("#define"):
+            self.process_define(line)
+        elif line.startswith("typedef"):
+            self.process_typedef(line)
+
+    def process_define(self, line: str):
+        """处理常量定义"""
+        words = line.split(" ")
+        words = [word for word in words if word]
+        if len(words) < 3:
+            return
+
+        name = words[1]
+        value = words[2]
+
+        new_line = f"{name} = {value}\n"
+        self.f_define.write(new_line)
+
+    def process_typedef(self, line: str):
+        """处理类型定义"""
+        words = line.split(" ")
+        words = [word for word in words if word != " "]
+
+        name = words[2]
+        typedef = TYPE_CPP2PY[words[1]]
+        print(name)
+        if typedef == "char":
+            if "[" in name:
+                typedef = "string"
+                name = name[:name.index("[")]
+
+        new_line = f"{name} = \"{typedef}\"\n"
+        self.f_typedef.write(new_line)
+
+
+if __name__ == "__main__":
+    generator = DataTypeGenerator("../include/mini/ThostFtdcUserApiDataType.h", "mini")
+    generator.run()

ctpbee_api/ctp_mini/generator/generate_struct.py

@@ -0,0 +1,85 @@
+""""""
+import importlib
+
+
+class StructGenerator:
+    """Struct生成器"""
+
+    def __init__(self, filename: str, prefix: str):
+        """Constructor"""
+        self.filename = filename
+        self.prefix = prefix
+        self.typedefs = {}
+
+        self.load_constant()
+
+    def load_constant(self):
+        """"""
+        module_name = f"{self.prefix}_typedef"
+        module = importlib.import_module(module_name)
+
+        for name in dir(module):
+            if "__" not in name:
+                self.typedefs[name] = getattr(module, name)
+
+    def run(self):
+        """运行生成"""
+        self.f_cpp = open(self.filename, "r")
+        self.f_struct = open(f"{self.prefix}_struct.py", "w")
+
+        for line in self.f_cpp:
+            self.process_line(line)
+
+        self.f_cpp.close()
+        self.f_struct.close()
+
+        print("Struct生成成功")
+
+    def process_line(self, line: str):
+        """处理每行"""
+        line = line.replace(";", "")
+        line = line.replace("\n", "")
+
+        if line.startswith("struct"):
+            self.process_declare(line)
+        elif line.startswith("{"):
+            self.process_start(line)
+        elif line.startswith("}"):
+            self.process_end(line)
+        elif "\t" in line and "///" not in line:
+            self.process_member(line)
+
+    def process_declare(self, line: str):
+        """处理声明"""
+        words = line.split(" ")
+        name = words[1]
+        end = "{"
+
+        new_line = f"{name} = {end}\n"
+        self.f_struct.write(new_line)
+
+    def process_start(self, line: str):
+        """处理开始"""
+        pass
+
+    def process_end(self, line: str):
+        """处理结束"""
+        new_line = "}\n\n"
+        self.f_struct.write(new_line)
+
+    def process_member(self, line: str):
+        """处理成员"""
+        words = line.split("\t")
+        words = [word for word in words if word]
+        print(words)
+        py_type = self.typedefs[words[0]]
+        name = words[1]
+
+        new_line = f"    \"{name}\": \"{py_type}\",\n"
+        self.f_struct.write(new_line)
+
+
+if __name__ == "__main__":
+    generator = StructGenerator(
+        "../include/mini/ThostFtdcUserApiStruct.h", "mini")
+    generator.run()