Thanks to visit codestin.com
Credit goes to github.com

Skip to content

gh-96151: Use a private name for passing builtins to dataclass #98143

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 8 commits into from
Oct 31, 2022
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 4 additions & 5 deletions Lib/dataclasses.py
Original file line number Diff line number Diff line change
Expand Up @@ -426,13 +426,11 @@ def wrapper(self):

def _create_fn(name, args, body, *, globals=None, locals=None,
return_type=MISSING):
# Note that we mutate locals when exec() is called. Caller
# beware! The only callers are internal to this module, so no
# Note that we may mutate locals. Callers beware!
# The only callers are internal to this module, so no
# worries about external callers.
if locals is None:
locals = {}
if 'BUILTINS' not in locals:
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I'd suggest using __BUILTINS__ because dunder names are supposed to be reserved for use by the stdlib.

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Before I commit this, I'd like to spend some time researching why this test is even present, instead of just unconditionally assigning to locals. At the very least it could use a comment.

Also, I'm not sure that exposing all of builtins in locals is a good idea, versus just exposing builtins.object.

Copy link
Contributor Author

@hauntsaninja hauntsaninja Oct 30, 2022

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Good catch, looks like the relevant history is:

That is, I think this check was made dead in #9518, but wasn't noticed in that PR

(Also note that the comment at the top of _create_fn is out of date: we do mutate locals, but not via exec)

Copy link
Contributor Author

@hauntsaninja hauntsaninja Oct 30, 2022

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I can do some double checking (locals should all be created within dataclasses.py) and clean that up + change the PR to only pass along object.

(I'll also note that there's still the inscrutable ().__class__.__base__ option on the table, in case we don't want to expose anything at all)

Copy link
Contributor Author

@hauntsaninja hauntsaninja Oct 31, 2022

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Okay, I audited all the call sites, it looks like there is one case where this check is not dead, but it's accidental.

Over here we reuse the same locals dict for two different _create_fn calls:

locals = {'cls': cls,

so the second time round we already have the entry for builtins in the dict. shrug

So my conclusion is:
a) It's safe to remove the check.
b) We should actually go a little further. Since we only need builtins for the frozen init, we should pass that in explicitly when creating __init__ and remove this from _create_fn

I've gone ahead and pushed this change to the PR

locals['BUILTINS'] = builtins
return_annotation = ''
if return_type is not MISSING:
locals['_return_type'] = return_type
Expand Down Expand Up @@ -462,7 +460,7 @@ def _field_assign(frozen, name, value, self_name):
# self_name is what "self" is called in this function: don't
# hard-code "self", since that might be a field name.
if frozen:
return f'BUILTINS.object.__setattr__({self_name},{name!r},{value})'
return f'__dataclass_builtins_object__.__setattr__({self_name},{name!r},{value})'
return f'{self_name}.{name}={value}'


Expand Down Expand Up @@ -569,6 +567,7 @@ def _init_fn(fields, std_fields, kw_only_fields, frozen, has_post_init,
locals.update({
'MISSING': MISSING,
'_HAS_DEFAULT_FACTORY': _HAS_DEFAULT_FACTORY,
'__dataclass_builtins_object__': object,
})

body_lines = []
Expand Down
8 changes: 8 additions & 0 deletions Lib/test/test_dataclasses.py
Original file line number Diff line number Diff line change
Expand Up @@ -257,6 +257,14 @@ class C:
c = C('foo')
self.assertEqual(c.object, 'foo')

def test_field_named_BUILTINS_frozen(self):
# gh-96151
@dataclass(frozen=True)
class C:
BUILTINS: int
c = C(5)
self.assertEqual(c.BUILTINS, 5)
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Please create a separate function for this, and don't mix it in with existing tests. Maybe test_field_named_BUILTINS or similar.

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks for the review, made the change :-)


def test_field_named_like_builtin(self):
# Attribute names can shadow built-in names
# since code generation is used.
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
Allow ``BUILTINS`` to be a valid field name for frozen dataclasses.