ucloud-api/helper.py

import binascii
import requests

from decouple import config
from pyotp import TOTP


def check_otp(name, realm, token):
    try:
        data = {
            "auth_name": config('AUTH_NAME', ''),
            "auth_token": TOTP(config('AUTH_SEED', '')).now(),
            "auth_realm": config('AUTH_REALM', ''),
            "name": name,
            "realm": realm,
            "token": token
        }
    except binascii.Error:
        return 400

    response = requests.post(
        "{OTP_SERVER}{OTP_VERIFY_ENDPOINT}".format(
            OTP_SERVER=config('OTP_SERVER', ''),
            OTP_VERIFY_ENDPOINT=config('OTP_VERIFY_ENDPOINT', 'verify/')
        ),
        data=data
    )
    return response.status_code


def add_otp_args(parser):
    parser.add_argument("name", required=True)
    parser.add_argument("realm", required=True)
    parser.add_argument("token", required=True)
    return parser


def add_vmid_args(parser):
    parser.add_argument("vmid", required=True)
    return parser