pferd/PFERD/auth/tfa.py

37 lines
1016 B
Python
Raw Normal View History

2021-05-15 18:27:16 +02:00
from typing import Tuple
from ..config import Config
from ..logging import log
2021-05-15 18:27:16 +02:00
from ..utils import ainput
from .authenticator import Authenticator, AuthError, AuthSection
2021-05-15 18:27:16 +02:00
class TfaAuthenticator(Authenticator):
def __init__(
self,
name: str,
section: AuthSection,
config: Config,
) -> None:
super().__init__(name, section, config)
2021-05-15 18:27:16 +02:00
async def username(self) -> str:
raise AuthError("TFA authenticator does not support usernames")
2021-05-15 18:27:16 +02:00
async def password(self) -> str:
async with log.exclusive_output():
2021-05-15 18:27:16 +02:00
code = await ainput("TFA code: ")
return code
async def credentials(self) -> Tuple[str, str]:
raise AuthError("TFA authenticator does not support usernames")
2021-05-15 18:27:16 +02:00
def invalidate_username(self) -> None:
raise AuthError("TFA authenticator does not support usernames")
2021-05-15 18:27:16 +02:00
def invalidate_password(self) -> None:
pass
def invalidate_credentials(self) -> None:
pass