diff options
Diffstat (limited to 'python/analytics/portfolio.py')
| -rw-r--r-- | python/analytics/portfolio.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/python/analytics/portfolio.py b/python/analytics/portfolio.py index 2872a7c2..62861922 100644 --- a/python/analytics/portfolio.py +++ b/python/analytics/portfolio.py @@ -81,6 +81,11 @@ class Portfolio: self.trade_ids.extend(other.trade_ids) return self + def __add__(self, other: Portfolio): + return Portfolio( + self.trades + other.trades, trade_ids=self.trade_ids + other.trade_ids + ) + def __getitem__(self, trade_id): for tid, trade in zip(self.trade_ids, self.trades): if tid == trade_id: |