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