1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
|
from __future__ import annotations
import abc
import os
class Credential(metaclass=abc.ABCMeta):
"""Abstract class to manage credentials"""
@abc.abstractproperty
def username(self) -> str: ...
@abc.abstractproperty
def password(self) -> str: ...
def _vars(self) -> dict[str, str]:
return dict(username=self.username, password=self.password)
class SimpleCredential(Credential):
"""Simple credentials implementation"""
def __init__(self, username: str, password: str):
self._username = username
self._password = password
@property
def username(self) -> str:
return self._username
@property
def password(self) -> str:
return self._password
class AnonymousCredential(SimpleCredential):
def __init__(self, password: str):
self._password = password
@property
def username(self) -> str:
raise ValueError("Anonymous credential has no username")
def _vars(self) -> dict[str, str]:
return dict(password=self.password)
class EnvironCredential(Credential):
"""
Source credentials from environment variables.
Actual sourcing is deferred until requested.
Supports comparison by equality.
>>> e1 = EnvironCredential('a', 'b')
>>> e2 = EnvironCredential('a', 'b')
>>> e3 = EnvironCredential('a', 'c')
>>> e1 == e2
True
>>> e2 == e3
False
"""
def __init__(self, user_env_var: str, pwd_env_var: str):
self.user_env_var = user_env_var
self.pwd_env_var = pwd_env_var
def __eq__(self, other: object) -> bool:
return vars(self) == vars(other)
def _get_env(self, env_var: str) -> str:
"""Helper to read an environment variable"""
value = os.environ.get(env_var)
if not value:
raise ValueError(f'Missing environment variable:{env_var}')
return value
@property
def username(self) -> str:
return self._get_env(self.user_env_var)
@property
def password(self) -> str:
return self._get_env(self.pwd_env_var)
|
