我有一个在 Django 服务上运行的大型 python 应用程序。我需要关闭某些操作的权限测试,因此我创建了这个上下文管理器:
class OverrideTests(object):
def __init__(self):
self.override = 0
def __enter__(self):
self.override += 1
# noinspection PyUnusedLocal
def __exit__(self, exc_type, exc_val, exc_tb):
self.override -= 1
assert not self.override < 0
@property
def overriding(self):
return self.override > 0
override_tests = OverrideTests()
然后,应用程序的各个部分可以使用上下文管理器覆盖测试:
with override_tests:
do stuff
...
在 do stuff 中,上述上下文管理器可以在不同的函数中多次使用。计数器的使用可以控制这种情况,并且看起来工作正常......直到线程介入。
一旦涉及到线程,全局上下文管理器就会被重新使用,因此,测试可能会被错误地覆盖。
这是一个简单的测试用例 - 如果thread.start_new_thread(do_id, ())
线被替换为简单的do_it
但严重失败,如下所示:
def stat(k, expected):
x = '.' if override_tests.overriding == expected else '*'
sys.stdout.write('{0}{1}'.format(k, x))
def do_it_inner():
with override_tests:
stat(2, True)
stat(3, True) # outer with context makes this true
def do_it():
with override_tests:
stat(1, True)
do_it_inner()
stat(4, False)
def do_it_lots(ntimes=10):
for i in range(ntimes):
thread.start_new_thread(do_it, ())
如何使这个上下文管理器线程安全,以便在每个 Python 线程中,即使它是可重入的,也能一致地使用它?