mirror of
https://gitee.com/fasiondog/hikyuu.git
synced 2024-12-03 20:37:50 +08:00
100 lines
2.6 KiB
C++
100 lines
2.6 KiB
C++
/*
|
|
* _Block.cpp
|
|
*
|
|
* Created on: 2015年2月10日
|
|
* Author: fasiondog
|
|
*/
|
|
|
|
#include <boost/python.hpp>
|
|
#include <hikyuu/serialization/Block_serialization.h>
|
|
#include <boost/python/suite/indexing/vector_indexing_suite.hpp>
|
|
#include "pickle_support.h"
|
|
|
|
using namespace boost::python;
|
|
using namespace hku;
|
|
|
|
#if defined(_MSC_VER)
|
|
#pragma warning(disable : 4267)
|
|
#endif
|
|
|
|
string (Block::*getCategory)() const = &Block::category;
|
|
void (Block::*setCategory)(const string&) = &Block::category;
|
|
string (Block::*getName)() const = &Block::name;
|
|
void (Block::*setName)(const string&) = &Block::name;
|
|
|
|
bool (Block::*add_1)(const Stock&) = &Block::add;
|
|
bool (Block::*add_2)(const string&) = &Block::add;
|
|
bool (Block::*remove_1)(const Stock&) = &Block::remove;
|
|
bool (Block::*remove_2)(const string&) = &Block::remove;
|
|
|
|
void export_Block() {
|
|
class_<Block>("Block", "板块类，可视为证券的容器", init<>())
|
|
.def(init<const string&, const string&>())
|
|
.def(init<const Block&>())
|
|
|
|
.def(self_ns::str(self))
|
|
.def(self_ns::repr(self))
|
|
|
|
.add_property("category", getCategory, setCategory, "板块所属分类")
|
|
.add_property("name", getName, setName, "板块名称")
|
|
|
|
.def("empty", &Block::empty, R"(empty(self)
|
|
|
|
是否为空)")
|
|
|
|
.def("add", add_1, R"(add(self, stock)
|
|
|
|
加入指定的证券
|
|
|
|
:param Stock stock: 待加入的证券
|
|
:return: 是否成功加入
|
|
:rtype: bool)")
|
|
|
|
.def("add", add_2, R"(add(self, market_code)
|
|
|
|
根据"市场简称证券代码"加入指定的证券
|
|
|
|
:param str market_code: 市场简称证券代码
|
|
:return: 是否成功加入
|
|
:rtype: bool)")
|
|
|
|
.def("remove", remove_1, R"(remove(self, stock)
|
|
|
|
移除指定证券
|
|
|
|
:param Stock stock: 指定的证券
|
|
:return: 是否成功
|
|
:rtype: bool)")
|
|
|
|
.def("remove", remove_2, R"(remove(market_code)
|
|
|
|
移除指定证券
|
|
|
|
:param str market_code: 市场简称证券代码
|
|
:return: True 成功 | False 失败
|
|
:rtype: bool)")
|
|
|
|
.def("clear", &Block::clear, "移除包含的所有证券")
|
|
|
|
.def("__len__", &Block::size, "包含的证券数量")
|
|
|
|
.def("__getitem__", &Block::get, R"(__getitem__(self, market_code)
|
|
|
|
:param str market_code: 证券代码
|
|
:return: Stock 实例)")
|
|
|
|
.def("__iter__", iterator<const Block>())
|
|
|
|
#if HKU_PYTHON_SUPPORT_PICKLE
|
|
.def_pickle(normal_pickle_suite<Block>())
|
|
#endif
|
|
;
|
|
|
|
class_<BlockList>("BlockList")
|
|
.def(vector_indexing_suite<BlockList>())
|
|
#if HKU_PYTHON_SUPPORT_PICKLE
|
|
.def_pickle(normal_pickle_suite<BlockList>())
|
|
#endif
|
|
;
|
|
}
|