21 lines
531 B
Python
Executable File
21 lines
531 B
Python
Executable File
import json
|
|
|
|
from pyotp import TOTP
|
|
|
|
|
|
class OTPCredentials:
|
|
def __init__(self, name, realm, seed):
|
|
self.name = name # type: str
|
|
self.realm = realm # type: str
|
|
self.seed = seed # type: str
|
|
|
|
def get_json(self):
|
|
return {"name": self.name, "realm": self.realm, "token": TOTP(self.seed).now()}
|
|
|
|
|
|
def load_dump_pretty(content):
|
|
if isinstance(content, bytes):
|
|
content = content.decode("utf-8")
|
|
parsed = json.loads(content)
|
|
return json.dumps(parsed, indent=4, sort_keys=True)
|