Ejemplo n.º 1
0
    def __init__(self, core: Core) -> None:
        self.core = core
        user_storage = Storage.get_storage(self.core,
                                           "users",
                                           1,
                                           storage_init=lambda: {},
                                           loader=self._load_users,
                                           dumper=self._dump_users)
        self.users = DictWrapper(user_storage)
        user_storage.schedule_save(self.users)

        token_storage = Storage.get_storage(
            self.core,
            "refresh_tokens",
            1,
            storage_init=lambda: {},
            loader=self._load_refresh_tokens,
            dumper=self._dump_refresh_tokens,
        )
        self.refresh_tokens = DictWrapper(token_storage)
        self.auth_codes: Dict[str, AuthorizationCode] = {}
        self.credential_providers = {
            name: provider(self)
            for name, provider in CREDENTIAL_PROVIDERS.items()
        }
Ejemplo n.º 2
0
    def __init__(self, cfg_dir: str, loop: AbstractEventLoop) -> None:
        user_storage = Storage.get_storage("users",
                                           1,
                                           cfg_dir=cfg_dir,
                                           loop=loop,
                                           storage_init=lambda: {},
                                           loader=self._load_users,
                                           dumper=self._dump_users)
        self.loop = loop
        self.users = DictWrapper(user_storage)
        user_storage.schedule_save(self.users)

        token_storage = Storage.get_storage(
            "refresh_tokens",
            1,
            cfg_dir=cfg_dir,
            loop=loop,
            storage_init=lambda: {},
            loader=self._load_refresh_tokens,
            dumper=self._dump_refresh_tokens,
        )
        self.refresh_tokens = DictWrapper(token_storage)
        self.auth_codes: Dict[str, AuthorizationCode] = {}
        self.credential_providers = {
            name: provider(self)
            for name, provider in CREDENTIAL_PROVIDERS.items()
        }