40 lines
1,004 B
Python
40 lines
1,004 B
Python
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
|