示例#1
0
文件: main.py 项目: cerndb/wls-cli
 def validate_security_data(self):
     self.model.update(MessageType.INFO, "Authentication - validating")
     self.validator = SecurityValidator(self.model.data_storage)
     self.validator.validate_security_data()
     self.model.update(MessageType.INFO, "Authentication - SUCCESS")
示例#2
0
文件: main.py 项目: aGGre55or/wls-cli
 def validate_security_data(self):
     self.model.update(MessageType.INFO, "Authentication - validating")
     self.validator = SecurityValidator(self.model.data_storage)
     self.validator.validate_security_data()
     self.model.update(MessageType.INFO, "Authentication - SUCCESS")
示例#3
0
文件: main.py 项目: cerndb/wls-cli
class ServiceManager(object):
    def __init__(self, model):
        """ Constructor """
        self.model = model
        self.service_objects_mapping = {
            AuthOperation.COOKIE: security.CookieAuthenticationService,
            AuthOperation.NETRC: security.NetrcAuthenticationService,
            AuthOperation.SCRIPT: security.ScriptAuthenticationService,
            AuthOperation.CERN_DATA: security.CERNAuthenticationService,
            AuthOperation.CREDS_IN_PARAMETERS: security.CredentialsAuthenticationService,
            AuthOperation.INTERACTIVE: security.CredentialsAuthenticationService,
        }

    def authorise(self, auth_operation, omit_auth):
        try:
            self.get_security_data(auth_operation)
            if not omit_auth:
                self.validate_security_data()

        except AuthenticationError:
            self.handle_authentication_error()

    def get_security_data(self, auth_operation):
        self.auth_operation = auth_operation
        security_service_object = self.service_objects_mapping[auth_operation](self.model.data_storage)
        security_service_object.get_data()

    def validate_security_data(self):
        self.model.update(MessageType.INFO, "Authentication - validating")
        self.validator = SecurityValidator(self.model.data_storage)
        self.validator.validate_security_data()
        self.model.update(MessageType.INFO, "Authentication - SUCCESS")

    def execute_request(self, operation):
        self.service_object = WeblogicServicesFactory(type(operation), self.model)
        # process the request
        self.model.update(MessageType.INFO, "Operation: '" + str(operation) + "'")
        return self.service_object.run(operation)

    def handle_authentication_error(self):

        if self.auth_operation == AuthOperation.COOKIE:
            self.display_error_msg("Cookie expired.")
            self.authorise(AuthOperation.INTERACTIVE)

        elif self.auth_operation == AuthOperation.NETRC:
            self.display_error_msg("Credentials in .netrc are invalid.")
            raise Exception()

        elif self.auth_operation == AuthOperation.SCRIPT:
            self.display_error_msg("Credentials provided by user script are invalid.")
            raise Exception()

        elif self.auth_operation == AuthOperation.CERN_DATA:
            self.display_error_msg("Credentials provided by CERN infrastructure are invalid.")
            raise Exception()

        elif self.auth_operation == AuthOperation.CREDS_IN_PARAMETERS:
            self.display_error_msg("Credentials are invalid.")
            raise Exception()

        elif self.auth_operation == AuthOperation.INTERACTIVE:
            self.display_error_msg("Credentials are invalid.")
            self.authorise(self.auth_operation, False)

        else:
            raise Exception("Unrecognized auth operation")

    def display_msg(self, message):
        self.model.update(MessageType.INFO, message)

    def display_error_msg(self, message):
        self.model.update(MessageType.ERROR, message)
示例#4
0
文件: main.py 项目: aGGre55or/wls-cli
class ServiceManager(object):
    def __init__(self, model):
        ''' Constructor '''
        self.model = model
        self.service_objects_mapping = \
            {AuthOperation.COOKIE: security.CookieAuthenticationService,
             AuthOperation.NETRC: security.NetrcAuthenticationService,
             AuthOperation.SCRIPT: security.ScriptAuthenticationService,
             AuthOperation.CERN_DATA: security.CERNAuthenticationService,
             AuthOperation.CREDS_IN_PARAMETERS: security.CredentialsAuthenticationService,
             AuthOperation.INTERACTIVE: security.CredentialsAuthenticationService}

    def authorise(self, auth_operation, omit_auth):
        try:
            self.get_security_data(auth_operation)
            if not omit_auth:
                self.validate_security_data()

        except AuthenticationError:
            self.handle_authentication_error()

    def get_security_data(self, auth_operation):
        self.auth_operation = auth_operation
        security_service_object = \
            self.service_objects_mapping[auth_operation](self.model.data_storage)
        security_service_object.get_data()

    def validate_security_data(self):
        self.model.update(MessageType.INFO, "Authentication - validating")
        self.validator = SecurityValidator(self.model.data_storage)
        self.validator.validate_security_data()
        self.model.update(MessageType.INFO, "Authentication - SUCCESS")

    def execute_request(self, operation):
        self.service_object = WeblogicServicesFactory(type(operation),
                                                      self.model)
        #process the request
        self.model.update(MessageType.INFO, "Operation: '" + \
                          str(operation) + "'")
        return self.service_object.run(operation)

    def handle_authentication_error(self):

        if self.auth_operation == AuthOperation.COOKIE:
            self.display_error_msg("Cookie expired.")
            self.authorise(AuthOperation.INTERACTIVE)

        elif self.auth_operation == AuthOperation.NETRC:
            self.display_error_msg("Credentials in .netrc are invalid.")
            raise Exception()

        elif self.auth_operation == AuthOperation.SCRIPT:
            self.display_error_msg(
                "Credentials provided by user script are invalid.")
            raise Exception()

        elif self.auth_operation == AuthOperation.CERN_DATA:
            self.display_error_msg(
                "Credentials provided by CERN infrastructure are invalid.")
            raise Exception()

        elif self.auth_operation == AuthOperation.CREDS_IN_PARAMETERS:
            self.display_error_msg("Credentials are invalid.")
            raise Exception()

        elif self.auth_operation == AuthOperation.INTERACTIVE:
            self.display_error_msg("Credentials are invalid.")
            self.authorise(self.auth_operation, False)

        else:
            raise Exception("Unrecognized auth operation")

    def display_msg(self, message):
        self.model.update(MessageType.INFO, message)

    def display_error_msg(self, message):
        self.model.update(MessageType.ERROR, message)