mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-08-04 08:59:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 11

gh-91291: Accept attributes as keyword arguments in decimal.localcontext (#32242)

Browse source 
Co-authored-by: Jelle Zijlstra <jelle.zijlstra@gmail.com>
This commit is contained in:
Sam Ezeh 2022-04-22 05:27:15 +01:00 • committed by GitHub
parent 5e130a8da4
commit bcf14ae433
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
6 changed files with 161 additions and 62 deletions
Download patch file Download diff file Expand all files Collapse all files

16
Lib/_pydecimal.py
View file

@ -441,6 +441,10 @@ import contextvars
_current_context_var = contextvars.ContextVar('decimal_context')
_context_attributes = frozenset(
['prec', 'Emin', 'Emax', 'capitals', 'clamp', 'rounding', 'flags', 'traps']
)
def getcontext():
"""Returns this thread's context.
@ -464,7 +468,7 @@ def setcontext(context):
del contextvars # Don't contaminate the namespace
def localcontext(ctx=None):
def localcontext(ctx=None, **kwargs):
"""Return a context manager for a copy of the supplied context
Uses a copy of the current context if no context is specified
@ -500,8 +504,14 @@ def localcontext(ctx=None):
>>> print(getcontext().prec)
28
"""
if ctx is None: ctx = getcontext()
return _ContextManager(ctx)
if ctx is None:
ctx = getcontext()
ctx_manager = _ContextManager(ctx)
for key, value in kwargs.items():
if key not in _context_attributes:
raise TypeError(f"'{key}' is an invalid keyword argument for this function")
setattr(ctx_manager.new_context, key, value)
return ctx_manager
##### Decimal class #######################################################