new: support for Wealthsimple Visa Infinite credit card account & transactions

Fixes #9

Author: gboudreau

setup.py

@@ -3,7 +3,7 @@
 setuptools.setup(
     name = 'WS-API',
     packages = ['ws_api'],
-    version = '0.16.0',
+    version = '0.17.0',
     license = 'GPL-3.0',
     description = 'Access your Wealthsimple account using their (GraphQL) API.',
     author = 'Guillaume Boudreau',

ws_api/wealthsimple_api.py

@@ -350,6 +350,8 @@ def _account_add_description(account):
             account['description'] = "Crypto"
         elif account['unifiedAccountType'] == 'SELF_DIRECTED_RRIF':
              account['description'] = f"RRIF: self-directed - {account['currency']}"
+        elif account['unifiedAccountType'] == 'CREDIT_CARD':
+             account['description'] = "Credit card"
         # TODO: Add other types as needed
 
     def get_account_balances(self, account_id):
@@ -551,6 +553,23 @@ def _activity_add_description(self, act):
             type_ = act['type'].capitalize()
             act['description'] = f"{type_}"
 
+        elif act['type'] == 'CREDIT_CARD' and act['subType'] == 'PURCHASE':
+            merchant = act['spendMerchant']
+            status = '(Pending) ' if act['status'] == 'authorized' else '' # Posted purchase transactions have status = settled
+            act['description'] = f"{status}Credit card purchase: {merchant}"
+
+        elif act['type'] == 'CREDIT_CARD' and act['subType'] == 'HOLD':
+            merchant = act['spendMerchant']
+            status = '(Pending) ' if act['status'] == 'authorized' else '' # Posted return transactions have subType = REFUND and status = settled
+            act['description'] = f"{status}Credit card refund: {merchant}"
+
+        elif act['type'] == 'CREDIT_CARD' and act['subType'] == 'REFUND':
+            merchant = act['spendMerchant']
+            act['description'] = f"Credit card refund: {merchant}"
+
+        elif act['type'] == 'CREDIT_CARD' and act['subType'] == 'PAYMENT':
+            act['description'] = "Credit card payment"
+
         # TODO: Add other types as needed
 
     def security_id_to_symbol(self, security_id: str) -> str: