Pagination (#24)

Author: TomAFrench

monzo/monzo.py

@@ -5,6 +5,9 @@
 """
 
 from monzo.auth import MonzoOAuth2Client
+from datetime import datetime
+from functools import partial
+
 import string
 import random
 
@@ -75,15 +78,37 @@ def get_first_account(self):
             raise LookupError('There are no accounts associated with this user.')
         return accounts['accounts'][0]
 
-    def get_transactions(self, account_id):
+    def get_transactions(self, account_id, before=None, since=None, limit=None):
         """Get all transactions of a given account. (https://monzo.com/docs/#list-transactions)
 
            :param account_id: The unique identifier for the account which the transactions belong to.
+           :param before: A datetime representing the time of the earliest transaction to return (Can't take transaction id as input)
+           :param since: A datetime representing the time of the earliest transaction to return. (Can also take a transaction id)
+           :param limit: The maximum number of transactions to return (Max = 100)
            :rtype: A collection of transaction objects for specific user.
         """
+        if isinstance(before, datetime):
+            before = before.isoformat() + 'Z'
+        if isinstance(since, datetime):
+            since = since.isoformat() + 'Z'
         url = "{0}/transactions".format(self.API_URL)
-        params = {'expand[]': 'merchant', 'account_id': account_id}
+        params = {
+            'expand[]': 'merchant',
+            'account_id': account_id,
+            'before': before,
+            'since': since,
+            'limit': limit,
+            }
         response = self.oauth_session.make_request(url, params=params)
+        if any([before,since,limit]):
+            last_transaction_id = response['transactions'][-1]['id']
+            next_page = partial(self.get_transactions,
+                                account_id,
+                                before=before,
+                                since = last_transaction_id,
+                                limit = limit)
+            response.update({'next_page':next_page})
+        
         return response
 
     def get_transaction(self, transaction_id):