forked from uncloud/uncloud
27 lines
614 B
Python
27 lines
614 B
Python
|
from uncloud.common.shared import shared
|
||
|
|
from pyotp import TOTP
|
||
|
|
|
||
|
|
|
||
|
|
def get_token(seed):
|
||
|
|
if seed is not None:
|
||
|
|
try:
|
||
|
|
token = TOTP(seed).now()
|
||
|
|
except Exception:
|
||
|
|
raise Exception('Invalid seed')
|
||
|
|
else:
|
||
|
|
return token
|
||
|
|
|
||
|
|
|
||
|
|
def resolve_otp_credentials(kwargs):
|
||
|
|
d = {
|
||
|
|
'name': shared.settings['client']['name'],
|
||
|
|
'realm': shared.settings['client']['realm'],
|
||
|
|
'token': get_token(shared.settings['client']['seed'])
|
||
|
|
}
|
||
|
|
|
||
|
|
for k, v in d.items():
|
||
|
|
if k in kwargs and kwargs[k] is None:
|
||
|
|
kwargs.update({k: v})
|
||
|
|
|
||
|
|
return d
|