def reload(self, message):
"""恢复方法
Arguments:
message {[type]} -- [description]
"""
self.phone = message.get('phone')
self.level = message.get('level')
self.utype = message.get('utype')
self.coins = message.get('coins')
self.wechat_id = message.get('wechat_id')
self.coins_history = message.get('coins_history')
self.money = message.get('money')
self._subscribed_strategy = message.get('subuscribed_strategy')
self._subscribed_code = message.get('subscribed_code')
self.username = message.get('username')
self.password = message.get('password')
self.user_cookie = message.get('user_cookie')
#
portfolio_list = message.get('portfolio_list')
if len(portfolio_list) > 0:
self.portfolio_list = dict(
zip(portfolio_list, [
QA_Portfolio(user_cookie=self.user_cookie,
portfolio_cookie=item)
for item in portfolio_list
]))
else:
self.portfolio_list = {}
def new_portfolio(self,
portfolio_cookie=None,
strategy_name=None,
init_cash=100000000,
sell_available=None,
market_type=MARKET_TYPE.STOCK_CN,
running_environment=RUNNING_ENVIRONMENT.BACKETEST):
'''
根据 self.user_cookie 创建一个 portfolio
:return:
如果存在 返回 新建的 QA_Portfolio
如果已经存在 返回 这个portfolio
'''
_portfolio = QA_Portfolio(user_cookie=self.user_cookie,
portfolio_cookie=portfolio_cookie,
strategy_name=strategy_name,
init_cash=init_cash,
sell_available=sell_available,
market_type=market_type,
running_environment=running_environment)
if _portfolio.portfolio_cookie not in self.portfolio_list.keys():
self.portfolio_list[_portfolio.portfolio_cookie] = _portfolio
return _portfolio
else:
print(" prortfolio with user_cookie ", self.user_cookie,
" already exist!!")
return self.portfolio_list[portfolio_cookie]
def get_portfolio(self, portfolio_cookie: str):
'''
'get a portfolio'
从 portfolio_list dict字典中 根据 portfolio key 获取
:param portfolio: QA_Portfolio类型
:return: QA_Portfolio类型
'''
# return self.portfolio_list[portfolio]
# fix here use cookie as key to find value in dict
return QA_Portfolio(user_cookie=self.user_cookie, portfolio_cookie=portfolio_cookie)
def new_portfolio(self, portfolio_cookie=None):
'''
根据 self.user_cookie 创建一个 portfolio
:return:
如果存在 返回 新建的 QA_Portfolio
如果已经存在 不返回 None
'''
_portfolio = QA_Portfolio(user_cookie=self.user_cookie, portfolio_cookie=portfolio_cookie)
if _portfolio.portfolio_cookie not in self.portfolio_list.keys():
self.portfolio_list[_portfolio.portfolio_cookie] = _portfolio
return _portfolio
else:
print(" prortfolio with user_cookie ", self.user_cookie , " already exist!!")
def new_portfolio(self, portfolio_cookie=None):
'''
根据 self.user_cookie 创建一个 portfolio
:return:
如果存在 返回 新建的 QA_Portfolio
如果已经存在 返回 这个portfolio
'''
if portfolio_cookie not in self.portfolio_list:
self.portfolio_list.append(portfolio_cookie)
return QA_Portfolio(user_cookie=self.user_cookie,
portfolio_cookie=portfolio_cookie)
else:
print(" prortfolio with user_cookie ", self.user_cookie,
" already exist!!")
return self.get_portfolio(portfolio_cookie)
def get_portfolio(self, user_cookie, portfolio_cookie):
"""首先进行变量检查
Arguments:
user_cookie {[type]} -- [description]
portfolio_cookie {[type]} -- [description]
Returns:
[type] -- [description]
"""
if user_cookie is None or portfolio_cookie is None:
return False
else:
try:
portfolio = QA_Portfolio(user_cookie=user_cookie,
portfolio_cookie=portfolio_cookie)
return portfolio
except:
return False
def new_portfolio(self):
'create a portfolio'
_portfolio = QA_Portfolio()
if _portfolio.portfolio_cookie not in self.portfolio_list.keys():
self.portfolio_list[_portfolio.portfolio_cookie] = _portfolio
return _portfolio.portfolio_cookie
