mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-08-22 09:45:06 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 13

bpo-32630: Use contextvars in decimal (GH-5278)

Browse source
This commit is contained in:
Yury Selivanov 2018-01-27 13:46:46 -05:00 • committed by GitHub
parent bc4123b0b3
commit f13f12d8da
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 71 additions and 101 deletions
Download patch file Download diff file Expand all files Collapse all files

20
Lib/_pydecimal.py
View file

@ -433,13 +433,11 @@ _rounding_modes = (ROUND_DOWN, ROUND_HALF_UP, ROUND_HALF_EVEN, ROUND_CEILING,
# The getcontext() and setcontext() function manage access to a thread-local
# current context.
import threading
import contextvars
local = threading.local()
if hasattr(local, '__decimal_context__'):
del local.__decimal_context__
_current_context_var = contextvars.ContextVar('decimal_context')
def getcontext(_local=local):
def getcontext():
"""Returns this thread's context.
If this thread does not yet have a context, returns
@ -447,20 +445,20 @@ def getcontext(_local=local):
New contexts are copies of DefaultContext.
"""
try:
return _local.__decimal_context__
except AttributeError:
return _current_context_var.get()
except LookupError:
context = Context()
_local.__decimal_context__ = context
_current_context_var.set(context)
return context
def setcontext(context, _local=local):
def setcontext(context):
"""Set this thread's context to context."""
if context in (DefaultContext, BasicContext, ExtendedContext):
context = context.copy()
context.clear_flags()
_local.__decimal_context__ = context
_current_context_var.set(context)
del threading, local # Don't contaminate the namespace
del contextvars # Don't contaminate the namespace
def localcontext(ctx=None):
"""Return a context manager for a copy of the supplied context