Account transaction history api support

oandapy/oandapy.py

@@ -1,5 +1,10 @@
+import os
+import sys
 import json
+import time
+import zipfile
 import requests
+from .utils import json_to_csv, ensure_directory_exists
 from .exceptions import BadEnvironment, OandaError
 
 """ OANDA API wrapper for OANDA's REST API """
@@ -167,6 +172,59 @@ def get_transaction(self, account_id, transaction_id):
                    (account_id, transaction_id)
         return self.request(endpoint)
 
+    def get_account_history(self, account_id, directory=None, **params):
+        """ Returns path to the csv file, containing account history information
+        If directory is specified, file will be created in that directory,
+        otherwise it will be created near the main module.
+        Docs: http://developer.oanda.com/rest-live/transaction-history
+        """
+        endpoint = '%s/v1/accounts/%s/alltransactions' % (self.api_url,
+                                                          account_id)
+        response = self.client.get(endpoint, params=params)
+        file_url = response.headers['Location']
+        if not directory:
+            directory = os.path.dirname(
+                os.path.abspath(sys.modules['__main__'].__file__)
+            )
+        ensure_directory_exists(directory)
+        max_tries = 5
+        timeout = 3
+        result = None
+        for step in xrange(max_tries):
+            result = self._get_account_history_file(file_url, directory)
+            if result:
+                break
+            time.sleep(timeout)
+        return result
+
+    def _get_account_history_file(self, file_url, directory, **params):
+        """Downloads account history to specified folder in both csv and json
+        Docs: http://developer.oanda.com/rest-live/transaction-history
+        """
+        params['stream'] = True
+        response = self.client.get(file_url, params=params)
+        if response.status_code == 404:
+            return None
+        out_file_name = 'account_history'
+        zip_file_name = os.path.join(directory, out_file_name+'.zip')
+        csv_file = None
+        with open(zip_file_name, 'a+b') as f:
+            for chunk in response:
+                f.write(chunk)
+            f.seek(0)
+            zip_file = zipfile.ZipFile(f)
+            for name in zip_file.namelist():
+                zip_file.extract(name, directory)
+                full_path = os.path.join(directory, name)
+                new_full_path = os.path.join(directory, out_file_name+'.json')
+                if os.path.exists(new_full_path):
+                    os.remove(new_full_path)
+                os.rename(full_path, new_full_path)
+                csv_file = json_to_csv(new_full_path, out_file_name+'.csv')
+            zip_file.close()
+        os.remove(zip_file_name)
+        return csv_file
+
     """Forex Labs"""
 
     def get_eco_calendar(self, **params):

oandapy/utils.py

@@ -0,0 +1,34 @@
+import os
+import csv
+import json
+
+
+def json_to_csv(json_file_path, csv_name, delete_source=False):
+    """
+    Converts json file to csv.
+    :param json_file_path: json file which needs to be converted to csv
+    :param csv_name: name for the csv file
+    :param delete_source: should json file be deleted or not
+    :return: path to new csv file
+    """
+    with open(json_file_path, 'r') as f:
+        data = json.loads(f.read())
+    headers = []
+    for line in data:
+        for key in line.iterkeys():
+            if key not in headers:
+                headers.append(key)
+    csv_file_path = os.path.join(os.path.dirname(json_file_path), csv_name)
+    with open(csv_file_path, 'w') as f:
+        csv_file = csv.DictWriter(f, headers,
+                                  delimiter=',', lineterminator='\n')
+        csv_file.writeheader()
+        csv_file.writerows(data)
+    if delete_source:
+        os.remove(json_file_path)
+    return csv_file_path
+
+
+def ensure_directory_exists(directory):
+    if not os.path.exists(directory):
+        os.makedirs(directory)