From 66ad4036ea3bfbe29dd99184c27a47d0a48102dd Mon Sep 17 00:00:00 2001 From: Serhiy Storchaka Date: Sun, 5 Dec 2021 22:42:50 +0200 Subject: [PATCH] bpo-45663: Fix is_dataclass() for dataclasses which are subclasses of types.GenericAlias (GH-29294) (cherry picked from commit 446be166861b2f08f87f74018113dd98ca5fca02) Co-authored-by: Serhiy Storchaka --- Lib/dataclasses.py | 2 +- Lib/test/test_dataclasses.py | 12 ++++++++++++ .../Library/2021-10-28-23-11-59.bpo-45663.J90N5R.rst | 2 ++ 3 files changed, 15 insertions(+), 1 deletion(-) create mode 100644 Misc/NEWS.d/next/Library/2021-10-28-23-11-59.bpo-45663.J90N5R.rst diff --git a/Lib/dataclasses.py b/Lib/dataclasses.py index b3a9194d9d0680..157e6d07ad2014 100644 --- a/Lib/dataclasses.py +++ b/Lib/dataclasses.py @@ -1211,7 +1211,7 @@ def _is_dataclass_instance(obj): def is_dataclass(obj): """Returns True if obj is a dataclass or an instance of a dataclass.""" - cls = obj if isinstance(obj, type) else type(obj) + cls = obj if isinstance(obj, type) and not isinstance(obj, GenericAlias) else type(obj) return hasattr(cls, _FIELDS) diff --git a/Lib/test/test_dataclasses.py b/Lib/test/test_dataclasses.py index bcd004f4ec3aa2..d42b1167609b05 100644 --- a/Lib/test/test_dataclasses.py +++ b/Lib/test/test_dataclasses.py @@ -8,6 +8,7 @@ import pickle import inspect import builtins +import types import unittest from unittest.mock import Mock from typing import ClassVar, Any, List, Union, Tuple, Dict, Generic, TypeVar, Optional, Protocol @@ -1350,6 +1351,17 @@ class B: with self.assertRaisesRegex(TypeError, 'should be called on dataclass instances'): replace(obj, x=0) + def test_is_dataclass_genericalias(self): + @dataclass + class A(types.GenericAlias): + origin: type + args: type + self.assertTrue(is_dataclass(A)) + a = A(list, int) + self.assertTrue(is_dataclass(type(a))) + self.assertTrue(is_dataclass(a)) + + def test_helper_fields_with_class_instance(self): # Check that we can call fields() on either a class or instance, # and get back the same thing. diff --git a/Misc/NEWS.d/next/Library/2021-10-28-23-11-59.bpo-45663.J90N5R.rst b/Misc/NEWS.d/next/Library/2021-10-28-23-11-59.bpo-45663.J90N5R.rst new file mode 100644 index 00000000000000..f246f67cf80e54 --- /dev/null +++ b/Misc/NEWS.d/next/Library/2021-10-28-23-11-59.bpo-45663.J90N5R.rst @@ -0,0 +1,2 @@ +Fix :func:`dataclasses.is_dataclass` for dataclasses which are subclasses of +:class:`types.GenericAlias`.