KQuery的recoverType可设定

fasiondog · Jan 25, 2022 · 72ac855 · 72ac855
1 parent a240375
commit 72ac855
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/hikyuu_cpp/hikyuu/KQuery.h b/hikyuu_cpp/hikyuu/KQuery.h
@@ -163,6 +163,11 @@ class HKU_API KQuery {
         return m_recoverType;
     }
 
+    /** 设置复权类型 */
+    void recoverType(RecoverType recoverType) {
+        m_recoverType = recoverType;
+    }
+
     /** 获取queryType名称，用于显示输出 */
     static string getQueryTypeName(QueryType);
 

diff --git a/hikyuu_pywrap/_KQuery.cpp b/hikyuu_pywrap/_KQuery.cpp
@@ -12,6 +12,9 @@
 using namespace boost::python;
 using namespace hku;
 
+KQuery::RecoverType (KQuery::*get_recoverType)() const = &KQuery::recoverType;
+void (KQuery::*set_recoverType)(KQuery::RecoverType recoverType) = &KQuery::recoverType;
+
 void export_KQuery() {
     scope in_Query =
       class_<KQuery>("Query", "K线数据查询条件", init<>())
@@ -31,7 +34,8 @@ void export_KQuery() {
                       "结束日期，当按索引查询方式创建时无效，为 constant.null_datetime")
         .add_property("query_type", &KQuery::queryType, "查询方式 Query.QueryType")
         .add_property("ktype", &KQuery::kType, "查询的K线类型 Query.KType")
-        .add_property("recover_type", &KQuery::recoverType, "查询的复权类型 Query.RecoverType")
+        .add_property("recover_type", get_recoverType, set_recoverType,
+                      "查询的复权类型 Query.RecoverType")
 
 #if HKU_PYTHON_SUPPORT_PICKLE
         .def_pickle(normal_pickle_suite<KQuery>())