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 = [
item['portfolio_cookie'] for item in DATABASE.portfolio.find(
{'user_cookie': self.user_cookie}, {
'portfolio_cookie': 1,
'_id': 0
})
]
# 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):
'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
