Merge pull request #119 from alpacahq/feature/account-activity

Add support for account activity API

Author: ttt733

alpaca_trade_api/entity.py

@@ -55,6 +55,10 @@ class Position(Entity):
     pass
 
 
+class AccountActivity(Entity):
+    pass
+
+
 class Bar(Entity):
     def __getattr__(self, key):
         if key == 't':

alpaca_trade_api/rest.py

@@ -10,8 +10,8 @@
     get_api_version,
 )
 from .entity import (
-    Account, AccountConfigurations, Asset,
-    Order, Position, BarSet, Clock, Calendar,
+    Account, AccountConfigurations, AccountActivity,
+    Asset, Order, Position, BarSet, Clock, Calendar,
 )
 from . import polygon
 
@@ -346,6 +346,34 @@ def get_clock(self):
         resp = self.get('/clock')
         return Clock(resp)
 
+    def get_activities(
+        self,
+        activity_types=None,
+        until=None,
+        after=None,
+        direction=None,
+        date=None,
+        page_size=None,
+    ):
+        url = '/account/activities'
+        params = {}
+        if isinstance(activity_types, list):
+            params['activity_types'] = ','.join(activity_types)
+        elif activity_types is not None:
+            url += '/{}'.format(activity_types)
+        if after is not None:
+            params['after'] = after
+        if until is not None:
+            params['until'] = until
+        if direction is not None:
+            params['direction'] = direction
+        if date is not None:
+            params['date'] = date
+        if page_size is not None:
+            params['page_size'] = page_size
+        resp = self.get(url, data=params)
+        return [AccountActivity(o) for o in resp]
+
     def get_calendar(self, start=None, end=None):
         params = {}
         if start is not None: