mini2fa 0.2.1__tar.gz → 0.2.3__tar.gz

{mini2fa-0.2.1/src/mini2fa.egg-info → mini2fa-0.2.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mini2fa
-Version: 0.2.1
+Version: 0.2.3
 Summary: 安全的本地 TOTP 双因素认证管理工具
 License-Expression: GPL-3.0-or-later
 Project-URL: Homepage, https://github.com/Duroxi/Mini2FA

{mini2fa-0.2.1 → mini2fa-0.2.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "mini2fa"
-version = "0.2.1"
+version = "0.2.3"
 description = "安全的本地 TOTP 双因素认证管理工具"
 readme = "README.md"
 license = "GPL-3.0-or-later"

{mini2fa-0.2.1 → mini2fa-0.2.3}/src/mini2fa/__init__.py

@@ -6,4 +6,4 @@ from .storage import StorageManager
 from .scanner import scan_qrcode, scan_qrcode_from_raw_data, parse_otp_uri, preprocess_image, UnsupportedOTPTypeError
 from .models import Account, OTPAccountInfo
 
-__version__ = "0.2.1"
+__version__ = "0.2.3"

{mini2fa-0.2.1 → mini2fa-0.2.3}/src/mini2fa/_cli.py

@@ -156,6 +156,7 @@ def handle_add_account(storage: StorageManager):
 
     if not os.path.exists(image_path):
         ui.print_line("✗ 文件不存在！请检查路径。")
+        ui.prompt("\n按 Enter 返回主菜单 ")
         return
 
     ui.print_line("正在扫描二维码...")
@@ -164,14 +165,17 @@ def handle_add_account(storage: StorageManager):
         result = scan_qrcode(image_path)
     except UnsupportedOTPTypeError as e:
         ui.print_line(f"✗ 该二维码是 {e.otp_type.upper()} 协议，本工具仅支持 TOTP。")
+        ui.prompt("\n按 Enter 返回主菜单 ")
         return
     except Exception as e:
         ui.print_line(f"✗ 扫描失败: {e}")
+        ui.prompt("\n按 Enter 返回主菜单 ")
         return
 
     if not result:
         ui.print_line("✗ 未识别到有效的二维码！")
         ui.print_line("  提示：请确保图片包含清晰的 OTP 二维码。")
+        ui.prompt("\n按 Enter 返回主菜单 ")
         return
 
     # 查重：已存在则直接提示，不进入填字段流程
@@ -180,6 +184,7 @@ def handle_add_account(storage: StorageManager):
         ui.print_line(f"✗ 账号已存在: {result.issuer} - {result.account}")
         ui.print_line("  该账号已在库中（分类: {0}, ID: {1}）。".format(
             existing.category, existing.id))
+        ui.prompt("\n按 Enter 返回主菜单 ")
         return
 
     # 显示识别结果
@@ -296,6 +301,7 @@ def handle_view_code(storage: StorageManager):
             idx = int(raw)
             if idx < 1 or idx > len(accounts):
                 ui.print_line("✗ 无效的选择！")
+                ui.prompt("\n按 Enter 继续 ")
                 continue
             account = accounts[idx - 1]
         else:
@@ -310,6 +316,7 @@ def handle_view_code(storage: StorageManager):
             searching = True
             if not accounts:
                 ui.print_line("✗ 未找到匹配账号，已返回全量列表。")
+                ui.prompt("\n按 Enter 继续 ")
                 accounts = all_accounts
                 searching = False
             continue
@@ -317,6 +324,7 @@ def handle_view_code(storage: StorageManager):
             secret = storage.get_secret(account.id)
         except Exception as e:
             ui.print_line(f"✗ 获取密钥失败: {e}")
+            ui.prompt("\n按 Enter 继续 ")
             continue
 
         # 进入详情页
@@ -353,9 +361,11 @@ def handle_edit_account(storage: StorageManager):
         idx = int(idx)
         if idx < 1 or idx > len(accounts):
             ui.print_line("✗ 无效的选择！")
+            ui.prompt("\n按 Enter 返回主菜单 ")
             return
     except ValueError:
         ui.print_line("✗ 请输入数字！")
+        ui.prompt("\n按 Enter 返回主菜单 ")
         return
 
     account = accounts[idx - 1]
@@ -421,9 +431,11 @@ def handle_delete_account(storage: StorageManager):
         idx = int(idx)
         if idx < 1 or idx > len(accounts):
             ui.print_line("✗ 无效的选择！")
+            ui.prompt("\n按 Enter 返回主菜单 ")
             return
     except ValueError:
         ui.print_line("✗ 请输入数字！")
+        ui.prompt("\n按 Enter 返回主菜单 ")
         return
 
     account = accounts[idx - 1]
@@ -474,6 +486,7 @@ def handle_export(storage: StorageManager):
     invalid_chars = set('*?<>|"')
     if user_path and any(c in os.path.basename(backup_path) for c in invalid_chars):
         ui.print_line("✗ 文件名包含非法字符（* ? < > | \"），请重新输入。")
+        ui.prompt("\n按 Enter 返回主菜单 ")
         return
 
     # 目标文件已存在时确认覆盖
@@ -513,6 +526,7 @@ def handle_change_password(crypto: CryptoManager):
             ui.print_line(f"  剩余 {2 - attempt} 次尝试。")
     else:
         ui.print_line("✗ 验证失败次数过多，已返回主菜单。")
+        ui.prompt("\n按 Enter 返回主菜单 ")
         return
 
     ui.print_line("✓ 旧密码验证通过！")
@@ -538,6 +552,7 @@ def handle_change_password(crypto: CryptoManager):
     else:
         ui.print_line(f"\n✗ 新密码设置失败：已尝试 {MAX_PASSWORD_ATTEMPTS} 次。")
         ui.print_line("  请重新运行 mini2fa 重试。")
+        ui.prompt("\n按 Enter 返回主菜单 ")
         return
 
     # 3. 重设密保提示（回车保留原提示）
@@ -571,6 +586,7 @@ def handle_import(storage: StorageManager):
 
     if not input_path:
         ui.print_line("✗ 路径不能为空！")
+        ui.prompt("\n按 Enter 返回主菜单 ")
         return
 
     # 移除路径两端的引号
@@ -583,10 +599,12 @@ def handle_import(storage: StorageManager):
     invalid_chars = set('*?<>|"')
     if any(c in os.path.basename(input_path) for c in invalid_chars):
         ui.print_line("✗ 文件名包含非法字符（* ? < > | \"），请重新输入。")
+        ui.prompt("\n按 Enter 返回主菜单 ")
         return
 
     if not os.path.exists(input_path):
         ui.print_line("✗ 文件不存在！")
+        ui.prompt("\n按 Enter 返回主菜单 ")
         return
 
     try:
@@ -604,9 +622,11 @@ def handle_import(storage: StorageManager):
                 external_key = storage.crypto.load_external_key(embedded_key, bk_pwd)
             except ValueError as e:
                 ui.print_line(f"✗ {e}")
+                ui.prompt("\n按 Enter 返回主菜单 ")
                 return
             if external_key is None:
                 ui.print_line("✗ 备份主密码错误！")
+                ui.prompt("\n按 Enter 返回主菜单 ")
                 return
             ui.print_line("✓ 备份主密码验证通过！")
 
@@ -615,6 +635,7 @@ def handle_import(storage: StorageManager):
 
         if preview['total'] == 0:
             ui.print_line("✗ 备份文件中没有账号数据。")
+            ui.prompt("\n按 Enter 返回主菜单 ")
             return
 
         ui.print_line("\n📋 备份文件预览：")
@@ -678,12 +699,14 @@ def handle_import(storage: StorageManager):
             _print_import_result(result)
     except json.JSONDecodeError:
         ui.print_line("✗ 导入失败：备份文件格式错误，不是有效的 JSON 文件。")
+        ui.prompt("\n按 Enter 返回主菜单 ")
     except Exception as e:
         err_msg = str(e)
         if 'InvalidTag' in err_msg or 'decrypt' in err_msg.lower():
             ui.print_line("✗ 导入失败：无法解密备份文件，可能是主密码不匹配或文件已损坏。")
         else:
             ui.print_line(f"✗ 导入失败: {e}")
+        ui.prompt("\n按 Enter 返回主菜单 ")
 
 
 def _build_parser() -> argparse.ArgumentParser:
@@ -701,6 +724,25 @@ def _build_parser() -> argparse.ArgumentParser:
         '--version', action='version', version=f'mini2fa {__version__}',
         help='显示版本号并退出',
     )
+
+    # 子命令
+    sub = parser.add_subparsers(dest='command', help='子命令')
+
+    # list 子命令（-p 在子命令后面）
+    list_cmd = sub.add_parser('list', help='列出所有账号（ID + issuer + account）')
+    list_cmd.add_argument(
+        '-p', '--password', metavar='PASSWORD', required=True,
+        help='主密码',
+    )
+
+    # get 子命令（-p 在子命令后面）
+    get_cmd = sub.add_parser('get', help='获取验证码')
+    get_cmd.add_argument(
+        '-p', '--password', metavar='PASSWORD', required=True,
+        help='主密码',
+    )
+    get_cmd.add_argument('id', type=int, help='账号 ID（从 list 获取）')
+
     return parser
 
 
@@ -709,11 +751,22 @@ def main(argv=None):
 
     Args:
         argv: 命令行参数列表（None 时用 sys.argv[1:]）。
-              由 argparse 解析，支持 --no-tui、--version。
+              由 argparse 解析，支持 --no-tui、-p、子命令 list。
 
     Ctrl-C 在任意阶段（含主密码输入）都被捕获，优雅退出。
     """
     args = _build_parser().parse_args(argv)
+
+    # 子命令模式：直接执行，不进 TUI
+    if args.command == 'list':
+        _handle_list(args.password)
+        return
+
+    if args.command == 'get':
+        _handle_get(args.password, args.id)
+        return
+
+    # 交互模式（TUI）
     ui.configure(no_tui=args.no_tui)
     ui.enter()
     try:
@@ -726,6 +779,88 @@ def main(argv=None):
         ui.leave()
 
 
+def _handle_list(password: str):
+    """子命令 list：列出所有账号（ID + issuer + account）
+
+    零交互：密码通过参数传入，输出纯文本。
+    """
+    # 初始化数据目录（确保存在）
+    init_data_dir()
+
+    db_path = str(get_db_path())
+    key_path = str(get_key_path())
+
+    # 检查主密钥文件是否存在
+    if not os.path.exists(key_path):
+        print("错误: 首次使用请先运行 mini2fa 设置主密码", file=sys.stderr)
+        sys.exit(1)
+
+    crypto = CryptoManager(key_path)
+
+    # 验证密码（不污染会话）
+    if not crypto.initialize(password):
+        print("密码错误", file=sys.stderr)
+        sys.exit(1)
+
+    storage = StorageManager(db_path, crypto)
+    accounts = storage.get_all_accounts()
+
+    if not accounts:
+        print("暂无账号", file=sys.stderr)
+        sys.exit(0)
+
+    # 按分类分组，输出 ID + issuer + account
+    groups = {}
+    for acc in accounts:
+        groups.setdefault(acc.category, []).append(acc)
+
+    categories = sorted(groups.keys(), key=lambda c: (c != 'default', c))
+
+    for category in categories:
+        print(f"\n  [{category}]")
+        for acc in groups[category]:
+            print(f"    {acc.id:3d}. {acc.issuer} - {acc.account}")
+
+
+def _handle_get(password: str, account_id: int):
+    """子命令 get：用 ID 获取验证码
+
+    零交互：密码通过参数传入，输出简洁一行式。
+    """
+    # 初始化数据目录（确保存在）
+    init_data_dir()
+
+    db_path = str(get_db_path())
+    key_path = str(get_key_path())
+
+    # 检查主密钥文件是否存在
+    if not os.path.exists(key_path):
+        print("错误: 首次使用请先运行 mini2fa 设置主密码", file=sys.stderr)
+        sys.exit(1)
+
+    crypto = CryptoManager(key_path)
+
+    # 验证密码
+    if not crypto.initialize(password):
+        print("密码错误", file=sys.stderr)
+        sys.exit(1)
+
+    storage = StorageManager(db_path, crypto)
+    account = storage.get_account(account_id)
+
+    if not account:
+        print(f"账号不存在: {account_id}", file=sys.stderr)
+        sys.exit(1)
+
+    # 解密密钥并生成验证码
+    secret = storage.get_secret(account_id)
+    code = generate_totp(secret, account.algorithm, account.digits, account.period)
+    remaining = get_remaining_seconds(account.period)
+
+    # 简洁一行式：issuer - account: code (remaining)
+    print(f"{account.issuer} - {account.account}: {code} ({remaining}s)")
+
+
 def _main_inner():
     """程序主体（不含 TUI 生命周期）"""
     ui.clear()  # 启动：清屏后显示 banner，保证顶部干净
@@ -768,6 +903,8 @@ def _main_inner():
         else:
             ui.print_line(f"\n✗ 主密码设置失败：已尝试 {MAX_PASSWORD_ATTEMPTS} 次。")
             ui.print_line("  请重新运行 mini2fa 重新设置。")
+            import time
+            time.sleep(1.5)
             sys.exit(1)
 
         # 设置密保提示
@@ -777,6 +914,8 @@ def _main_inner():
             ui.print_line("✓ 主密码已设置成功！")
         else:
             ui.print_line("✗ 初始化失败！")
+            import time
+            time.sleep(1.5)
             sys.exit(1)
     else:
         # 获取密保提示
@@ -794,7 +933,9 @@ def _main_inner():
                 if hint:
                     ui.print_line(f"💡 密码提示：{hint}")
         else:
-            ui.print_line("✗ 密码错误")
+            ui.print_line("✗ 密码错误次数过多，程序将退出")
+            import time
+            time.sleep(1.5)
             sys.exit(1)
 
     # 初始化存储管理器
@@ -846,6 +987,8 @@ def _run():
     except StorageCorruptedError as e:
         ui.print_line(f"\n✗ {e}")
         ui.print_line("  请检查数据文件是否完整，或从备份恢复。")
+        import time
+        time.sleep(1.5)
         sys.exit(1)
 
 

{mini2fa-0.2.1 → mini2fa-0.2.3/src/mini2fa.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mini2fa
-Version: 0.2.1
+Version: 0.2.3
 Summary: 安全的本地 TOTP 双因素认证管理工具
 License-Expression: GPL-3.0-or-later
 Project-URL: Homepage, https://github.com/Duroxi/Mini2FA

{mini2fa-0.2.1 → mini2fa-0.2.3}/tests/test_cli.py

@@ -197,7 +197,8 @@ class TestChangePasswordCLI:
             # getpass 3次都是错误旧密码
             inputs = iter(['wrong1', 'wrong2', 'wrong3'])
             with patch('getpass.getpass', side_effect=lambda prompt: next(inputs)):
-                handle_change_password(crypto)
+                with patch('builtins.input', return_value=''):
+                    handle_change_password(crypto)
 
             from mini2fa.crypto import CryptoManager
             reload = CryptoManager(key)
@@ -318,7 +319,8 @@ class TestAddAccountFlow:
         storage.add_account('Google', 'u@gmail.com', 'JBSWY3DPEHPK3PXP')
         try:
             # 若流程错误进入填字段, next() 会 StopIteration 崩溃
-            inputs = iter(['fake.png'])
+            # 需要两个 input: 图片路径 + 按 Enter 返回主菜单
+            inputs = iter(['fake.png', ''])
             with patch('mini2fa._cli.os.path.exists', return_value=True):
                 with patch('mini2fa._cli.scan_qrcode', return_value=info):
                     with patch('builtins.input', side_effect=lambda prompt: next(inputs)):