Merge pull request #128 from tenable/task/as-49

Add support for user authorizations endpoints and bump to 1.9.0

Author: tenableAndrew

.gitignore

@@ -12,8 +12,7 @@ tenable_io.ini
 # Test state / virtualenvs
 .cache
 .tox
-.venv
-.venv3
+.venv*
 
 # IntelliJ / PyCharm IDE
 .idea/

CHANGELOG.rst

@@ -2,6 +2,10 @@
 CHANGELOG
 =========
 
+1.9.0
+==========
+* Added: Support for user authorizations endpoints
+
 1.8.0
 ==========
 * Added: Support for creating dynamic tags

tenable_io/__init__.py

@@ -1 +1 @@
-__version__ = "1.8.0"
+__version__ = "1.9.0"

tenable_io/api/models.py

@@ -2149,6 +2149,9 @@ def __init__(
             type=None,
             login_fail_count=None,
             last_login_attempt=None,
+            uuid=None,
+            uuid_id=None,
+            container_uuid=None
     ):
         self.id = id
         self.username = username
@@ -2159,6 +2162,9 @@ def __init__(
         self.type = type
         self.login_fail_count = login_fail_count
         self.last_login_attempt = last_login_attempt
+        self.uuid = uuid
+        self.uuid_id = uuid_id
+        self.container_uuid = container_uuid
 
 
 class UserKeys(BaseModel):
@@ -2209,6 +2215,23 @@ def users(self, users):
         self._users = users
 
 
+class UserAuthorizations(BaseModel):
+
+    def __init__(
+            self,
+            account_uuid=None,
+            user_uuid=None,
+            api_permitted=None,
+            password_permitted=None,
+            saml_permitted=None
+    ):
+        self.account_uuid = account_uuid
+        self.user_uuid = user_uuid
+        self.api_permitted = api_permitted
+        self.password_permitted = password_permitted
+        self.saml_permitted = saml_permitted
+
+
 class AssetSeverity(BaseModel):
 
     def __init__(

tenable_io/api/users.py

@@ -1,7 +1,7 @@
 from json import loads
 
 from tenable_io.api.base import BaseApi, BaseRequest
-from tenable_io.api.models import User, UserKeys, UserList
+from tenable_io.api.models import User, UserKeys, UserList, UserAuthorizations
 
 
 class UsersApi(BaseApi):
@@ -95,6 +95,27 @@ def enabled(self, user_id, enabled):
         self._client.put('users/%(user_id)s/enabled', {'enabled': enabled}, {'user_id': user_id})
         return True
 
+    def authorizations(self, user_id):
+        """Returns authorizations for a specified user.
+
+        :param user_id: The user ID.
+        :raise TenableIOApiException:  When API error is encountered.
+        :return: An instance of :class:`tenable_io.api.models.UserAuthorizations`
+        """
+        response = self._client.get('users/%(user_id)s/authorizations', {'user_id': user_id})
+        return UserAuthorizations.from_json(response.text)
+
+    def update_authorizations(self, user_id, authorizations_update):
+        """Updates a user's authorizations.
+
+        :param user_id: The user ID.
+        :param authorizations_update: An instance of :class:`UserAuthorizationsRequest`.
+        :raise TenableIOApiException:  When API error is encountered.
+        :return: True if successful.
+        """
+        self._client.put('users/%(user_id)s/authorizations', authorizations_update, {'user_id': user_id})
+        return True
+
 
 class UserCreateRequest(BaseRequest):
 
@@ -126,3 +147,16 @@ def __init__(
         self.permissions = permissions
         self.name = name
         self.email = email
+
+class UserAuthorizationsRequest(BaseRequest):
+
+    def __init__(
+            self,
+            api_permitted=None,
+            password_permitted=None,
+            saml_permitted=None
+    ):
+        self.api_permitted = api_permitted
+        self.password_permitted = password_permitted
+        self.saml_permitted = saml_permitted
+