Python 中 dataclass 风格定义异常类,正确吗?

148 天前
 gosky

python 源码中有下面这样的异常定义:

class OSError(Exception):
    errno: int
    strerror: str
    # filename, filename2 are actually str | bytes | None
    filename: Any
    filename2: Any
    if sys.platform == "win32":
        winerror: int

因此我经常按上面的风格定义异常类,比如这次的:

class Error(Exception):
    code: str
    message: str

这样子写,一直工作良好。也许是因为我之前从没访问里面的属性? 这次,我写了下面的代码:

assert e.code in ("unauthorized", "permission")

收到很温馨的提示:

    assert e.code in ("unauthorized", "permission")
           ^^^^^^
AttributeError: 'Error' object has no attribute 'code'

问了 Claude ,回答为:

你提到的这种异常定义方法确实在 Python 源码中存在,特别是在类型提示和静态类型检查方面。这种定义方式主要用于类型注解和静态类型检查工具(如 mypy )。让我解释一下这种定义的目的和工作原理:

1. 类型注解的目的:
   这种定义主要是为了提供类型信息,而不是实际定义属性。它告诉类型检查器这个类预期会有哪些属性,以及这些属性的预期类型。

2. 运行时行为:
   这些注解在运行时不会创建实际的属性。实际的属性创建通常在类的 `__init__` 方法或其他方法中完成。

3. 兼容性和灵活性:
   这种方式允许在不改变运行时行为的情况下添加类型信息,这对于大型的、需要保持向后兼容性的项目(如 Python 标准库)非常有用。

4. 静态类型检查:
   静态类型检查工具可以使用这些注解来检查代码,即使这些属性可能在运行时动态创建。

5. 文档作用:
   这些注解也起到了文档的作用,告诉开发者这个类预期会有哪些属性。

在 `OSError` 的例子中,这些属性可能在异常被实际创建时(可能是在 C 代码中)被设置,而不是在 Python 的 `__init__` 方法中。

需要注意的是,这种定义方式主要用于 Python 的内置类型和标准库,在普通的 Python 代码中,我们通常会使用 `__init__` 方法来初始化属性,就像我在前面的回答中建议的那样。

对于你自己的代码,除非你有特殊需求(比如为了兼容性或与特定的类型检查工具配合),否则最好还是使用常规的方式定义和初始化属性。

你对这种定义方式还有其他疑问吗?或者你想了解更多关于 Python 类型注解的信息?
1421 次点击
所在节点    Python
6 条回复
retanoj
148 天前
你的 e.code 中的 e 是怎么初始化和赋值的呢?
huangyezhufeng
148 天前
第一句话就没理解,Python 源码也没找到你说的这个定义...
https://github.com/search?q=repo%3Apython%2Fcpython+class+OSError%28Exception%29&type=code
Vegetable
148 天前
你发的 OSError 是 buitilins.pyi 文件中的类型定义,根本不是 py 代码。

class 内部的`field: str`形式只是一种类型标注,dataclass 和 pydantic 之类的库支持这种形式的写法,但是 Exception 不支持,你需要自己在 init 方法中处理好。
Vegetable
148 天前
@huangyezhufeng https://github.com/python/typeshed/blob/9eae1ae1a5b47745c7d289bb2092f2b4340e7622/stdlib/builtins.pyi#L1890

他把编辑器(估计是 vscode )内置的 pyi 当作是 python 源码了
ruanimal
147 天前
因为你 except 的类型不对吧
lulaolu
147 天前
Claude 说得挺清楚:这种写法只是注解,并不真的生效。

你既然提到了 dataclass ,那可以用这个库优化下代码:

```python
>>> from dataclasses import dataclass
>>>
>>> @dataclass
... class Error(Exception):
... code: str
... message: str
...
>>>
>>> err = Error(code=1, message="2")
>>> err.code
1
```

这是一个专为移动设备优化的页面(即为了让你能够在 Google 搜索结果里秒开这个页面),如果你希望参与 V2EX 社区的讨论,你可以继续到 V2EX 上打开本讨论主题的完整版本。

https://www.v2ex.com/t/1060371

V2EX 是创意工作者们的社区,是一个分享自己正在做的有趣事物、交流想法,可以遇见新朋友甚至新机会的地方。

V2EX is a community of developers, designers and creative people.

© 2021 V2EX