Merge pull request #1087 from yutiansut/master

update fix bug for portfolio
yutiansut · Apr 2, 2019 · 8a6a9a8 · 8a6a9a8
2 parents 15a4382 + 302dfb4
commit 8a6a9a8
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/QUANTAXIS/QAARP/QAPortfolio.py b/QUANTAXIS/QAARP/QAPortfolio.py
@@ -201,6 +201,7 @@ def add_account(self, account):
                 account.user_cookie = self.user_cookie
                 self.cash.append(self.cash_available - account.init_cash)
                 self.account_list.append(account.account_cookie)
+                account.save()
                 return account
         else:
             pass
@@ -217,7 +218,8 @@ def drop_account(self, account_cookie):
 
         if account_cookie in self.account_list:
             res = self.account_list.remove(account_cookie)
-            self.cash.append(self.cash[-1] + res.init_cash)
+            self.cash.append(
+                self.cash[-1] + self.get_account_by_cookie(res).init_cash)
             return True
         else:
             raise RuntimeError(
@@ -296,7 +298,7 @@ def get_account_by_cookie(self, cookie):
         '''
         try:
             return QA_Account(
-                account_cookie=account_cookie,
+                account_cookie=cookie,
                 user_cookie=self.user_cookie,
                 portfolio_cookie=self.portfolio_cookie,
                 auto_reload=True
@@ -388,7 +390,6 @@ def receive_deal(self):
     def table(self):
         return pd.concat([acc.table for acc in self.accounts.values()], axis=1)
 
-
     @property
     def portfolioView(self):
         return []