GithubFollow/One-KVM
1
0
Fork 0
mirror of https://github.com/mofeng-git/One-KVM.git synced 2025-12-12 01:00:29 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

totp: custom name or from fqdn

Browse Source
This commit is contained in:
Maxim Devaev 2023-01-23 18:38:55 +02:00
parent acd6279113
commit a2e90c08d3
1 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
kvmd/apps/totp/__init__.py
View File

@ -20,6 +20,7 @@
# ========================================================================== # # ========================================================================== #
import socket
import argparse import argparse
import pyotp import pyotp
@ -53,16 +54,18 @@ def _cmd_init(config: Section, options: argparse.Namespace) -> None:
_cmd_show(config, options) _cmd_show(config, options)
def _cmd_show(config: Section, _: argparse.Namespace) -> None: def _cmd_show(config: Section, options: argparse.Namespace) -> None:
secret = _read_secret(config) secret = _read_secret(config)
if len(secret) == 0: if len(secret) == 0:
raise SystemExit("Error: TOTP secret is not configured") raise SystemExit("Error: TOTP secret is not configured")
uri = pyotp.totp.TOTP(secret).provisioning_uri(issuer_name="PiKVM") uri = pyotp.totp.TOTP(secret).provisioning_uri(
name=(options.name or socket.getfqdn()),
issuer_name="PiKVM",
)
qr = qrcode.QRCode() qr = qrcode.QRCode()
qr.add_data(uri) qr.add_data(uri)
print() print("\nSecret:", secret, "\n")
print(uri) print("URI:", uri, "\n")
print()
qr.print_ascii(invert=True) qr.print_ascii(invert=True)
print() print()
@ -90,6 +93,7 @@ def main(argv: (list[str] | None)=None) -> None:
cmd_setup_parser = subparsers.add_parser("init", help="Generate and show TOTP secret with QR code") cmd_setup_parser = subparsers.add_parser("init", help="Generate and show TOTP secret with QR code")
cmd_setup_parser.add_argument("-f", "--force", action="store_true", help="Overwrite an existing secret") cmd_setup_parser.add_argument("-f", "--force", action="store_true", help="Overwrite an existing secret")
cmd_setup_parser.add_argument("-n", "--name", default="", help="PiKVM instance name, FQDN by default")
cmd_setup_parser.set_defaults(cmd=_cmd_init) cmd_setup_parser.set_defaults(cmd=_cmd_init)
cmd_show_parser = subparsers.add_parser("show", help="Show the current TOTP secret with QR code") cmd_show_parser = subparsers.add_parser("show", help="Show the current TOTP secret with QR code")