uncloud-cli/commands/helper.py

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)