mirror of
https://github.com/Kozea/Radicale.git
synced 2025-08-01 18:18:31 +00:00
More type hints
This commit is contained in:
parent
12fe5ce637
commit
cecb17df03
51 changed files with 1374 additions and 957 deletions
|
@ -28,18 +28,23 @@ Take a look at the class ``BaseAuth`` if you want to implement your own.
|
|||
|
||||
"""
|
||||
|
||||
from radicale import utils
|
||||
from typing import Sequence, Tuple, Union
|
||||
|
||||
INTERNAL_TYPES = ("none", "remote_user", "http_x_remote_user", "htpasswd")
|
||||
from radicale import config, types, utils
|
||||
|
||||
INTERNAL_TYPES: Sequence[str] = ("none", "remote_user", "http_x_remote_user",
|
||||
"htpasswd")
|
||||
|
||||
|
||||
def load(configuration):
|
||||
def load(configuration: "config.Configuration") -> "BaseAuth":
|
||||
"""Load the authentication module chosen in configuration."""
|
||||
return utils.load_plugin(INTERNAL_TYPES, "auth", "Auth", configuration)
|
||||
return utils.load_plugin(INTERNAL_TYPES, "auth", "Auth", BaseAuth,
|
||||
configuration)
|
||||
|
||||
|
||||
class BaseAuth:
|
||||
def __init__(self, configuration):
|
||||
|
||||
def __init__(self, configuration: "config.Configuration") -> None:
|
||||
"""Initialize BaseAuth.
|
||||
|
||||
``configuration`` see ``radicale.config`` module.
|
||||
|
@ -49,7 +54,8 @@ class BaseAuth:
|
|||
"""
|
||||
self.configuration = configuration
|
||||
|
||||
def get_external_login(self, environ):
|
||||
def get_external_login(self, environ: types.WSGIEnviron) -> Union[
|
||||
Tuple[()], Tuple[str, str]]:
|
||||
"""Optionally provide the login and password externally.
|
||||
|
||||
``environ`` a dict with the WSGI environment
|
||||
|
@ -61,7 +67,7 @@ class BaseAuth:
|
|||
"""
|
||||
return ()
|
||||
|
||||
def login(self, login, password):
|
||||
def login(self, login: str, password: str) -> str:
|
||||
"""Check credentials and map login to internal user
|
||||
|
||||
``login`` the login name
|
||||
|
|
|
@ -49,18 +49,23 @@ When passlib[bcrypt] is installed:
|
|||
|
||||
import functools
|
||||
import hmac
|
||||
from typing import Any
|
||||
|
||||
from passlib.hash import apr_md5_crypt
|
||||
|
||||
from radicale import auth
|
||||
from radicale import auth, config
|
||||
|
||||
|
||||
class Auth(auth.BaseAuth):
|
||||
def __init__(self, configuration):
|
||||
|
||||
_filename: str
|
||||
_encoding: str
|
||||
|
||||
def __init__(self, configuration: config.Configuration) -> None:
|
||||
super().__init__(configuration)
|
||||
self._filename = configuration.get("auth", "htpasswd_filename")
|
||||
self._encoding = self.configuration.get("encoding", "stock")
|
||||
encryption = configuration.get("auth", "htpasswd_encryption")
|
||||
self._encoding = configuration.get("encoding", "stock")
|
||||
encryption: str = configuration.get("auth", "htpasswd_encryption")
|
||||
|
||||
if encryption == "plain":
|
||||
self._verify = self._plain
|
||||
|
@ -82,17 +87,17 @@ class Auth(auth.BaseAuth):
|
|||
raise RuntimeError("The htpasswd encryption method %r is not "
|
||||
"supported." % encryption)
|
||||
|
||||
def _plain(self, hash_value, password):
|
||||
def _plain(self, hash_value: str, password: str) -> bool:
|
||||
"""Check if ``hash_value`` and ``password`` match, plain method."""
|
||||
return hmac.compare_digest(hash_value.encode(), password.encode())
|
||||
|
||||
def _bcrypt(self, bcrypt, hash_value, password):
|
||||
def _bcrypt(self, bcrypt: Any, hash_value: str, password: str) -> bool:
|
||||
return bcrypt.verify(password, hash_value.strip())
|
||||
|
||||
def _md5apr1(self, hash_value, password):
|
||||
def _md5apr1(self, hash_value: str, password: str) -> bool:
|
||||
return apr_md5_crypt.verify(password, hash_value.strip())
|
||||
|
||||
def login(self, login, password):
|
||||
def login(self, login: str, password: str) -> str:
|
||||
"""Validate credentials.
|
||||
|
||||
Iterate through htpasswd credential file until login matches, extract
|
||||
|
|
|
@ -26,9 +26,14 @@ if the reverse proxy is not configured properly.
|
|||
|
||||
"""
|
||||
|
||||
import radicale.auth.none as none
|
||||
from typing import Tuple, Union
|
||||
|
||||
from radicale import types
|
||||
from radicale.auth import none
|
||||
|
||||
|
||||
class Auth(none.Auth):
|
||||
def get_external_login(self, environ):
|
||||
|
||||
def get_external_login(self, environ: types.WSGIEnviron) -> Union[
|
||||
Tuple[()], Tuple[str, str]]:
|
||||
return environ.get("HTTP_X_REMOTE_USER", ""), ""
|
||||
|
|
|
@ -26,5 +26,6 @@ from radicale import auth
|
|||
|
||||
|
||||
class Auth(auth.BaseAuth):
|
||||
def login(self, login, password):
|
||||
|
||||
def login(self, login: str, password: str) -> str:
|
||||
return login
|
||||
|
|
|
@ -25,9 +25,14 @@ It's intended for use with an external WSGI server.
|
|||
|
||||
"""
|
||||
|
||||
import radicale.auth.none as none
|
||||
from typing import Tuple, Union
|
||||
|
||||
from radicale import types
|
||||
from radicale.auth import none
|
||||
|
||||
|
||||
class Auth(none.Auth):
|
||||
def get_external_login(self, environ):
|
||||
|
||||
def get_external_login(self, environ: types.WSGIEnviron
|
||||
) -> Union[Tuple[()], Tuple[str, str]]:
|
||||
return environ.get("REMOTE_USER", ""), ""
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue