Adding easier way to get to subaccounts and set subaccounts. (#92)

Author: Kristinn

jupyter/laceworkjupyter/helper.py

@@ -76,6 +76,19 @@ def __init__(
             else:
                 setattr(self, wrapper, wrapper_object)
 
+    @property
+    def subaccount(self):
+        """Returns the subaccount that is in use."""
+        return self.sdk.subaccount
+
+    @subaccount.setter
+    def subaccount(self, subaccount):
+        """Changes the subaccount that is in use."""
+        if subaccount == self.subaccount:
+            return
+
+        self.sdk.set_subaccount(subaccount)
+
     def __enter__(self):
         """
         Support the with statement in python.

laceworksdk/api/__init__.py

@@ -178,6 +178,13 @@ def __init__(self,
         self.vulnerability_exceptions = VulnerabilityExceptionsAPI(self._session)
         self.vulnerability_policies = VulnerabilityPoliciesAPI(self._session)
 
+    @property
+    def subaccount(self):
+        """
+        Returns the value of the session's subaccount.
+        """
+        return self._session.subaccount
+
     def set_org_level_access(self, org_level_access):
         """
         A method to set whether the client should use organization-level API calls.
@@ -193,4 +200,4 @@ def set_subaccount(self, subaccount):
         A method to update the subaccount the client should use for API calls.
         """
 
-        self._session._subaccount = subaccount
+        self._session.subaccount = subaccount

laceworksdk/http_session.py

@@ -266,6 +266,20 @@ def _request(self, method, uri, **kwargs):
 
         return response
 
+    @property
+    def subaccount(self):
+        """
+        Returns the current subaccount for the session.
+        """
+        return self._subaccount
+
+    @subaccount.setter
+    def subaccount(self, subaccount):
+        """
+        Modifies the value of the sessions's subaccount.
+        """
+        self._subaccount = subaccount
+
     def get(self, uri, params=None, **kwargs):
         """
         A method to build a GET request to interact with Lacework.