prime PortfolioAccount add update timestamp

Author: tigersk

tigeropen/trade/domain/prime_account.py

@@ -9,8 +9,13 @@ class PortfolioAccount:
     """
     prime/paper account assets
     """
-    def __init__(self, account):
+    def __init__(self, account, timestamp=None):
+        """
+        :param account:
+        :param timestamp: asset update timestamp in milliseconds.
+        """
         self.account = account
+        self.timestamp = timestamp
         self._segments = dict()
 
     @property
@@ -28,7 +33,7 @@ def add_segment(self, segment):
             return self._segments.get(segment.category)
 
     def __repr__(self):
-        d = {'account': self.account, 'segments': self.segments}
+        d = {'account': self.account, 'timestamp': self.timestamp, 'segments': self.segments}
         return MODEL_REPR.format(self.__class__.__name__, d)
 
 

tigeropen/trade/response/prime_assets_response.py

@@ -20,7 +20,7 @@ def parse_response_content(self, response_content):
             self._is_success = response['is_success']
 
         if self.data:
-            assets = PortfolioAccount(self.data['accountId'])
+            assets = PortfolioAccount(self.data.get('accountId'), self.data.get('updateTimestamp'))
 
             for segment_data in self.data.get('segments', list()):
                 segment = Segment()