diff apollo/ApolloUser.py @ 0:9945eba268e6 draft default tip

planemo upload for repository https://github.com/Yating-L/suite_gonramp_apollo.git commit 9009fa8663038d2cb4a1c5130600a51f31f654a1-dirty

--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/apollo/ApolloUser.py	Mon Nov 27 12:49:27 2017 -0500
@@ -0,0 +1,98 @@
+#!/usr/bin/python
+
+
+import os
+import sys
+import json
+import logging
+from util import subtools
+
+class ApolloUser(object):
+    """ 
+    This class is to manage Apollo users, such as create, delete, add users to a group or delete users from a group
+    
+    """
+
+    def __init__(self, users_list):
+        self.users_list = users_list
+        self.logger = logging.getLogger(__name__)
+
+
+    def createApolloUser(self):
+        for user in self.users_list:
+            if user['batch'] == "false":
+                subtools.arrow_create_user(user['useremail'], user['firstname'], user['lastname'], user['password'])
+            elif user['batch'] == "true":
+                users = self.parseUserInfoFile(user['format'], user['false_path'])
+                for u in users:
+                    if not 'useremail' in u:
+                        self.logger.error("Cannot find useremail in the text file, make sure you use the correct header, see README file for examples.")
+                    if not 'firstname' in u:
+                        self.logger.error("Cannot find firstname in the text file, make sure you use the correct header, see README file for examples.")
+                    if not 'lastname' in u:
+                        self.logger.error("Cannot find lastname in the text file, make sure you use the correct header, see README file for examples.")
+                    if not 'password' in u:
+                        self.logger.error("Cannot find password in the text file, make sure you use the correct header, see README file for examples.")
+                    subtools.arrow_create_user(u['useremail'], u['firstname'], u['lastname'], u['password'])
+    
+    
+    def parseUserInfoFile(self, file_format, filename):
+        if file_format == "tab":
+            delimiter = '\t'
+        elif file_format == "csv":
+            delimiter = ','
+        else:
+            self.logger.error("The %s format is not supported!", file_format)
+        with open(filename, 'r') as f:
+            lines = f.readlines()
+        headers = lines[0].rstrip().split(delimiter)
+        users = []
+        lines = lines[1:]
+        for l in lines:
+            print l
+            l = l.split(delimiter)
+            info = dict()
+            fields = len(l)
+            for i in range(fields):
+                info[headers[i]] = l[i]
+            users.append(info)
+        return users
+
+    def deleteApolloUser(self):
+        for user in self.users_list:
+            if user['batch'] == "false":
+                subtools.arrow_delete_user(user['useremail'])
+            elif user['batch'] == "true":
+                users = self.parseUserInfoFile(user['format'], user['false_path']) 
+                for u in users:
+                    if not 'useremail' in u:
+                        self.logger.error("Cannot find useremail in the text file, make sure you use the correct header, see README file for examples.")
+                    subtools.arrow_delete_user(u['useremail'])
+
+    def addApolloUserToGroup(self):
+        for user in self.users_list:
+            if user['batch'] == "false":
+                subtools.arrow_add_to_group(user['group'], user['useremail'])
+            elif user['batch'] == "true":
+                users = self.parseUserInfoFile(user['format'], user['false_path'])
+                for u in users:
+                    if not 'useremail' in u:
+                        self.logger.error("Cannot find useremail in the text file, make sure you use the correct header, see README file for examples.")
+                    if not 'group' in u:
+                        self.logger.error("Cannot find group in the text file, make sure you use the correct header, see README file for examples.")
+                    subtools.arrow_add_to_group(u['group'], u['useremail'])
+                
+    def removeApolloUserFromeGroup(self):
+        for user in self.users_list:
+            if user['batch'] == "false":
+                subtools.arrow_remove_from_group(user['group'], user['useremail'])
+            elif user['batch'] == "true":
+                users = self.parseUserInfoFile(user['format'], user['false_path'])
+                for u in users:
+                    if not 'useremail' in u:
+                        self.logger.error("Cannot find useremail in the text file, make sure you use the correct header, see README file for examples.")
+                    if not 'group' in u:
+                        self.logger.error("Cannot find group in the text file, make sure you use the correct header, see README file for examples.")
+                    subtools.arrow_add_to_group(u['group'], u['useremail'])
+                
+