Add team invite delete

Author: hovnatan

docs/source/superannotate.sdk.rst

@@ -110,6 +110,7 @@ _________________
 
 .. autofunction:: superannotate.get_team_metadata
 .. autofunction:: superannotate.invite_contributor_to_team
+.. autofunction:: superannotate.delete_contributor_to_team_invitation
 
 ----------
 

superannotate/__init__.py

@@ -61,7 +61,10 @@
     upload_preannotations_from_folder_to_project, upload_video_to_project,
     upload_videos_from_folder_to_project
 )
-from .db.teams import get_team_metadata, invite_contributor_to_team
+from .db.teams import (
+    delete_contributor_to_team_invitation, get_team_metadata,
+    invite_contributor_to_team
+)
 from .db.users import search_team_contributors
 from .dicom_converter import dicom_to_rgb_sequence
 from .exceptions import (

superannotate/db/teams.py

@@ -52,18 +52,25 @@ def get_team_metadata(convert_users_role_to_string=False):
     return res
 
 
-# def delete_team_contributor_invitation(invitation):
-#     """Deletes team contributor invitation
-
-#     :param invite: invitation metadata returned from invite_contributor_to_team
-#     :type project: dict
-#     """
-#     data = {'token': invitation["token"], 'e_mail': invitation['email']}
-#     response = _api.send_request(
-#         req_type='DELETE', path=f'/team/{_api.team_id}/invite', json_req=data
-#     )
-#     if not response.ok:
-#         raise SABaseException(
-#             response.status_code,
-#             "Couldn't delete contributor invite. " + response.text
-#         )
+def delete_contributor_to_team_invitation(email):
+    """Deletes team contributor invitation
+
+    :param email: invitation email
+    :type email: str
+    """
+    team_metadata = get_team_metadata()
+    for invite in team_metadata["pending_invitations"]:
+        if invite["email"] == email:
+            break
+    else:
+        raise SABaseException(0, "Couldn't find user " + email + " invitation")
+
+    data = {'token': invite["token"], 'e_mail': invite['email']}  # pylint: disable=undefined-loop-variable
+    response = _api.send_request(
+        req_type='DELETE', path=f'/team/{_api.team_id}/invite', json_req=data
+    )
+    if not response.ok:
+        raise SABaseException(
+            response.status_code,
+            "Couldn't delete contributor invite. " + response.text
+        )

superannotate/version.py

@@ -1 +1 @@
-__version__ = "2.3.5"
+__version__ = "2.3.6"

tests/test_team_invite.py

@@ -2,14 +2,13 @@
 
 import superannotate as sa
 
+TEST_EMAIL = "hovnata1@superannotate.com"
+
 
 def test_team_invite():
-    try:
-        sa.invite_contributor_to_team("hovnatan@superannotate.com")
-    except sa.SABaseException as e:
-        assert e.message == 'Couldn\'t invite to team. {"error":"User with hovnatan@superannotate.com email is already member of this team."}'
-    else:
-        assert False
+    sa.invite_contributor_to_team(TEST_EMAIL)
+    sa.delete_contributor_to_team_invitation(TEST_EMAIL)
+
     # sa.delete_team_contributor_invitation(invite)
 
     # print(sa.search_team_contributors())