diff options
Diffstat (limited to 'astroid/brain/brain_dataclasses.py')
| -rw-r--r-- | astroid/brain/brain_dataclasses.py | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/astroid/brain/brain_dataclasses.py b/astroid/brain/brain_dataclasses.py index 6ea14511..a05c46f9 100644 --- a/astroid/brain/brain_dataclasses.py +++ b/astroid/brain/brain_dataclasses.py @@ -3,9 +3,12 @@ """ Astroid hook for the dataclasses library -Support both built-in dataclasses and pydantic.dataclasses. References: +Support built-in dataclasses, pydantic.dataclasses, and marshmallow_dataclass-annotated +dataclasses. References: - https://docs.python.org/3/library/dataclasses.html - https://pydantic-docs.helpmanual.io/usage/dataclasses/ +- https://lovasoa.github.io/marshmallow_dataclass/ + """ from typing import FrozenSet, Generator, List, Optional, Tuple @@ -35,7 +38,9 @@ from astroid.util import Uninferable DATACLASSES_DECORATORS = frozenset(("dataclass",)) FIELD_NAME = "field" -DATACLASS_MODULES = frozenset(("dataclasses", "pydantic.dataclasses")) +DATACLASS_MODULES = frozenset( + ("dataclasses", "marshmallow_dataclass", "pydantic.dataclasses") +) DEFAULT_FACTORY = "_HAS_DEFAULT_FACTORY" # based on typing.py |