fixed interactive STOCKTYPE_A_BJ

fasiondog · Apr 25, 2024 · 4b2c6b6 · 4b2c6b6
1 parent 6b2036a
commit 4b2c6b6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/hikyuu/interactive.py b/hikyuu/interactive.py
@@ -127,7 +127,7 @@
 # ==============================================================================
 blocka = Block("A", "ALL")
 for s in sm:
-    if s.type in (constant.STOCKTYPE_A, constant.STOCKTYPE_GEM):
+    if s.type in (constant.STOCKTYPE_A, constant.STOCKTYPE_A_BJ, constant.STOCKTYPE_GEM):
         blocka.add(s)
 zsbk_a = blocka
 
@@ -144,7 +144,7 @@
 zsbk_sz = blocksz
 
 blockbj = Block("A", "BJ")
-for s in sm:
+for s in blocka:
     if s.market == "BJ":
         blockbj.add(s)
 zsbk_bj = blockbj

diff --git a/hikyuu_pywrap/_Constant.cpp b/hikyuu_pywrap/_Constant.cpp
@@ -144,7 +144,7 @@ void export_Constant(py::module& m) {
       .def_readonly("STOCKTYPE_GEM", &Constant::STOCKTYPE_GEM, "创业板")
       .def_readonly("STOCKTYPE_START", &Constant::STOCKTYPE_START, "科创板")
       .def_readonly("STOCKTYPE_CRYPTO", &Constant::STOCKTYPE_START, "数字币")
-      .def_readonly("STOCKTYPE_A_BJ", &Constant::STOCKTYPE_START, "A股北交所")
+      .def_readonly("STOCKTYPE_A_BJ", &Constant::STOCKTYPE_A_BJ, "A股北交所")
       .def_readonly("STOCKTYPE_TMP", &Constant::STOCKTYPE_TMP, "临时Stock");
 
     m.attr("constant") = Constant();