gosky 发表于 2026-7-10 16:51:48

哪位朋友帮忙用 mac 测试下我的 Python 脚本

把输出贴给我

如果有异常,如果能附带帮忙 fix 更好

import signal
import subprocess
from typing import OrderedDict

APP_NAME = "Hello"

class CalledProcessError(subprocess.CalledProcessError):
    """
    `subprocess.CalledProcessError` does not print `stderr` and `output`.
    """

    def __str__(self):
      if self.returncode and self.returncode < 0:
            try:
                return "Command '%s' died with %r." % (
                  self.cmd,
                  signal.Signals(-self.returncode),
                )
            except ValueError:
                return "Command '%s' died with unknown signal %d." % (
                  self.cmd,
                  -self.returncode,
                )
      else:
            return (
                "Command '%s' returned non-zero exit status %d. stderr: '%s'. output: '%s'"
                % (self.cmd, self.returncode, self.stderr.strip(), self.output.strip())
            )

class DecryptError(Exception): ...

class DarwinCryptex:
    scheme = "security"

    def encrypt(self, attrs: OrderedDict, plaintext: str) -> str:
      account = "-".join()
      cmd = [
            "security",
            "add-generic-password",
            "-a",
            account,
            "-s",
            APP_NAME,
            "-w",
            plaintext,
            "-U",
      ]
      r = subprocess.run(
            cmd,
            capture_output=True,
            encoding="utf-8",
            text=True,
            timeout=5,
      )
      if r.returncode != 0:
            raise CalledProcessError(
                r.returncode, cmd, output=r.stdout, stderr=r.stderr
            )
      return "******"

    def decrypt(self, attrs: OrderedDict, ciphertext: str) -> str:
      account = "-".join()
      cmd = [
            "security",
            "find-generic-password",
            "-a",
            account,
            "-s",
            APP_NAME,
            "-w",
      ]
      r = subprocess.run(
            cmd,
            capture_output=True,
            encoding="utf-8",
            text=True,
            timeout=5,
      )
      if r.returncode == 0:
            return r.stdout.strip()
      elif r.returncode == 44:
            raise DecryptError()
      else:
            raise CalledProcessError(
                r.returncode, cmd, output=r.stdout, stderr=r.stderr
            )

    def clean(self, attrs: OrderedDict):
      account = "-".join()
      cmd = ["security", "delete-generic-password", "-a", account, "-s", APP_NAME]
      r = subprocess.run(
            cmd, capture_output=True, encoding="utf-8", text=True, timeout=5
      )
      if r.returncode not in (0, 44):
            raise CalledProcessError(
                r.returncode, cmd, output=r.stdout, stderr=r.stderr
            )

cryptex = DarwinCryptex()
attrs: OrderedDict = OrderedDict(
    [
      ("app", "test"),
      ("section", "hello"),
      ("key", "你好"),
    ]
)
plaintext = "Hello world! 你好,世界!"

try:
    ciphertext = cryptex.encrypt(attrs, plaintext)
    decrypted = cryptex.decrypt(attrs, ciphertext)
    assert decrypted == plaintext

    mismatch_attrs = attrs.copy()
    mismatch_attrs["needless"] = "有毒"
    try:
      cryptex.decrypt(mismatch_attrs, ciphertext)
    except DecryptError:
      pass
    else:
      raise AssertionError("Expected DecryptError")
finally:
    cryptex.clean(attrs)

busln 发表于 2026-7-10 17:05:51

assert decrypted == plaintext

不成立报错了

且 encrypt 是不是没写完全呀

gosky 发表于 2026-7-10 17:14:17

@busln
感觉是写完了
把两个变量都打印瞧瞧?

busln 发表于 2026-7-10 17:26:10

程序有问题, 没写进去

gosky 发表于 2026-7-10 17:37:01

@busln 我用 claude review 了下,没大问题啊

zhhanging 发表于 2026-7-10 17:42:24

assert 失败,打印两个变量如下
decrypted: 48656c6c6f20776f726c642120e4bda0e5a5bdefbc8ce4b896e7958cefbc81
plaintext: Hello world! 你好,世界!
应该是遇到非 ascii 字符给他转成 hex 了

gosky 发表于 2026-7-10 17:48:42

@zhhanging
"Hello world! 你好,世界!".encode('utf-8').hex()
刚好是
'48656c6c6f20776f726c642120e4bda0e5a5bdefbc8ce4b896e7958cefbc81'

busln 发表于 2026-7-10 17:52:05

finally 给删了,漏看了
@gosky

busln 发表于 2026-7-10 17:53:57

bytes.fromhex(hex).decode("utf-8")
这样就能转换回来了

gosky 发表于 2026-7-10 17:56:44

@busln
把下面的中文字符,全换成英文字母,会怎么样?

attrs: OrderedDict = OrderedDict(
    [
      ("app", "test"),
      ("section", "hello"),
      ("key", "你好"),
    ]
)
plaintext = "Hello world! 你好,世界!"
页: [1] 2
查看完整版本: 哪位朋友帮忙用 mac 测试下我的 Python 脚本