microsoft/pyright
1
1
Fork 0
mirror of https://github.com/microsoft/pyright.git synced 2024-09-11 07:55:56 +03:00
Code Issues Projects Releases Wiki Activity

Updated logic for dataclass_transform to handle recent clarification in PEP 681 for cases where one or more overloads are decorated with dataclass_transform.

Browse Source
This commit is contained in:
Eric Traut 2022-04-01 09:52:41 -07:00
parent 139139a9a2
commit dc153cb35d
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
packages/pyright-internal/src/analyzer/dataClasses.ts
View File

@ -738,7 +738,13 @@ export function getDataclassDecoratorBehaviors(type: Type): DataClassBehaviors |
if (isFunction(type)) {
functionType = type;
} else if (isOverloadedFunction(type)) {
functionType = type.overloads[0];
// Find the first overload or implementation that contains a
// dataclass_transform decorator. If more than one have such a decorator,
// only the first one will be honored, as per PEP 681.
functionType =
type.overloads.find((overload) => {
overload.details.decoratorDataClassBehaviors !== undefined;
}) ?? type.overloads[0];
}
if (!functionType) {