bpo-37868: Improve is_dataclass for instances. (GH-15325)

This commit is contained in:
Eric V. Smith 2019-08-20 01:40:28 -04:00 committed by GitHub
parent d3c8d73514
commit b0f4dab873
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 32 additions and 2 deletions

View File

@ -1015,13 +1015,14 @@ def fields(class_or_instance):
def _is_dataclass_instance(obj): def _is_dataclass_instance(obj):
"""Returns True if obj is an instance of a dataclass.""" """Returns True if obj is an instance of a dataclass."""
return not isinstance(obj, type) and hasattr(obj, _FIELDS) return hasattr(type(obj), _FIELDS)
def is_dataclass(obj): def is_dataclass(obj):
"""Returns True if obj is a dataclass or an instance of a """Returns True if obj is a dataclass or an instance of a
dataclass.""" dataclass."""
return hasattr(obj, _FIELDS) cls = obj if isinstance(obj, type) else type(obj)
return hasattr(cls, _FIELDS)
def asdict(obj, *, dict_factory=dict): def asdict(obj, *, dict_factory=dict):

View File

@ -1300,6 +1300,32 @@ class TestCase(unittest.TestCase):
self.assertTrue(is_dataclass(d.d)) self.assertTrue(is_dataclass(d.d))
self.assertFalse(is_dataclass(d.e)) self.assertFalse(is_dataclass(d.e))
def test_is_dataclass_when_getattr_always_returns(self):
# See bpo-37868.
class A:
def __getattr__(self, key):
return 0
self.assertFalse(is_dataclass(A))
a = A()
# Also test for an instance attribute.
class B:
pass
b = B()
b.__dataclass_fields__ = []
for obj in a, b:
with self.subTest(obj=obj):
self.assertFalse(is_dataclass(obj))
# Indirect tests for _is_dataclass_instance().
with self.assertRaisesRegex(TypeError, 'should be called on dataclass instances'):
asdict(obj)
with self.assertRaisesRegex(TypeError, 'should be called on dataclass instances'):
astuple(obj)
with self.assertRaisesRegex(TypeError, 'should be called on dataclass instances'):
replace(obj, x=0)
def test_helper_fields_with_class_instance(self): def test_helper_fields_with_class_instance(self):
# Check that we can call fields() on either a class or instance, # Check that we can call fields() on either a class or instance,
# and get back the same thing. # and get back the same thing.

View File

@ -0,0 +1,3 @@
Fix dataclasses.is_dataclass when given an instance that never raises
AttributeError in __getattr__. That is, an object that returns something
for __dataclass_fields__ even if it's not a dataclass.