Skip to content

Commit ecbac2c

Browse files
authored
Merge pull request #16 from Deigue/subdivision-description
Fix the detailed description for subdivision txn
2 parents 9de7320 + f738b0f commit ecbac2c

1 file changed

Lines changed: 6 additions & 2 deletions

File tree

ws_api/wealthsimple_api.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -488,11 +488,15 @@ def _activity_add_description(self, act):
488488
held_activity = next((activity for activity in child_activities if activity['entitlementType'] == 'HOLD'), None)
489489
receive_activity = next((activity for activity in child_activities if activity['entitlementType'] == 'RECEIVE'), None)
490490
if held_activity and receive_activity:
491+
held_shares: float = float(held_activity['quantity'])
492+
received_shares: float = float(receive_activity['quantity'])
493+
total_shares: float = held_shares + received_shares
491494
act['description'] = (
492-
f"Subdivision: {held_activity['quantity']} -> {receive_activity['quantity']} shares of {act['assetSymbol']}"
495+
f"Subdivision: {held_shares} -> {total_shares} shares of {act['assetSymbol']}"
493496
)
494497
else:
495-
act['description'] = f"Subdivision: Received {act['amount']} shares of {act['assetSymbol']}"
498+
received_shares: float = float(act['amount'])
499+
act['description'] = f"Subdivision: Received {received_shares} new shares of {act['assetSymbol']}"
496500

497501
elif act['type'] in ['DEPOSIT', 'WITHDRAWAL'] and act['subType'] in ['E_TRANSFER', 'E_TRANSFER_FUNDING']:
498502
direction = 'from' if act['type'] == 'DEPOSIT' else 'to'

0 commit comments

Comments
 (0)