Class AuthSourceManager

web/pgadmin/authenticate/__init__.py:206–324

This class will manage all the authentication sources.

class AuthSourceManager:
    """This class will manage all the authentication sources.
     """

    def __init__(self, form, sources):
        self.form = form
        self.auth_sources = sources
        self.source = None
        self.source_friendly_name = INTERNAL
        self.current_source = INTERNAL
        self.update_auth_sources()

    def as_dict(self):
        """
        Returns the dictionary object representing this object.
        """

        res = dict()
        res['source_friendly_name'] = self.source_friendly_name
        res['auth_sources'] = self.auth_sources
        res['current_source'] = self.current_source

        return res

    def update_auth_sources(self):
        # Only mutate the ordered list of auth sources when a user explicitly
        # selected an auth method on the login form.
        #
        # Without this guard, a plain internal login POST (email/password) can
        # incorrectly drop INTERNAL/LDAP and try OAUTH2 first, which then fails
        # because no oauth2 provider button was provided.
        if request.method != 'POST':
            return

        if 'internal_button' in request.form:
            for auth_src in [KERBEROS, OAUTH2]:
                if auth_src in self.auth_sources:
                    self.auth_sources.remove(auth_src)
            return

        if 'oauth2_button' in request.form:
            if INTERNAL in self.auth_sources:
                self.auth_sources.remove(INTERNAL)
            if LDAP in self.auth_sources:
                self.auth_sources.remove(LDAP)
            return

    def set_current_source(self, source):
        self.current_source = source

    @property
    def get_current_source(self):
        return self.current_source

    def set_source(self, source):
        self.source = source

    @property