Add option to add users to gitlab using API

amfoss · Mar 28, 2021 · ce49cba · ce49cba
1 parent 50f5c74
commit ce49cba
Show file tree

Hide file tree

Showing 3 changed files with 30 additions and 1 deletion.
diff --git a/framework/api/mutation.py b/framework/api/mutation.py
@@ -79,5 +79,33 @@ def mutate(self, info, username, github=None, gitlab=None, telegram=None, cloudf
                                code='ONLY_SUPERUSER_HAS_ACCESS')
 
 
+class AddToPlatform(graphene.Mutation):
+    class Arguments:
+        usernames = graphene.List(graphene.String)
+        platform = graphene.String()
+
+    Output = statusObj
+
+    def mutate(self, info, usernames, platform):
+        if info.context.user.is_superuser:
+            for username in usernames:
+                profile = Profile.objects.get(user__username=username)
+                if platform is not None:
+                    if platform == "gitlab":
+                        GitLab(profile.gitlabUsername).addUser()
+                    elif platform == "github":
+                        GitHub(profile.githubUsername).addUser()
+                    elif platform == "telegram":
+                        Telegram(profile.telegram_id).addUser()
+                else:
+                    raise APIException('Platform is required to perform this action',
+                                       code='PLATFORM_IS_REQUIRED')
+            return statusObj(status=True)
+        else:
+            raise APIException('Only Superusers have access',
+                               code='ONLY_SUPERUSER_HAS_ACCESS')
+
+
 class Mutation(object):
     change_user_platform = ChangeUserPlatform.Field()
+    addToPlatform = AddToPlatform.Field()
diff --git a/framework/platforms/gitlab.py b/framework/platforms/gitlab.py
@@ -22,7 +22,7 @@ def addUser(self):
         gl.auth()
         group = gl.groups.get('amfoss')
         userID = gl.users.list(username=self.username)[0].id
-        group.members.create({'user_id': userID, 'access_level': gitlab.GUEST_ACCESS})
+        group.members.create({'user_id': userID, 'access_level': gitlab.REPORTER_ACCESS})
 
     def checkIfUserExists(self):
         GITLAB_TOKEN = Token.objects.values().get(key='GITLAB_TOKEN')['value']

diff --git a/framework/schema.py b/framework/schema.py
@@ -127,6 +127,7 @@ class userResponseObj(graphene.ObjectType):
 class userStatusObj(graphene.ObjectType):
     status = graphene.String()
 
+
 class ApproveUsers(graphene.Mutation):
     class Arguments:
         usernames = graphene.List(graphene.String)