Use six to reload module

This commit is contained in:
Peter Inglesby 2014-03-06 23:51:02 +00:00
parent 2353878951
commit 29f5ce7aeb

View File

@ -1,4 +1,5 @@
from contextlib import contextmanager from contextlib import contextmanager
from rest_framework.compat import six
from rest_framework.settings import api_settings from rest_framework.settings import api_settings
@ -14,11 +15,11 @@ def temporary_setting(setting, value, module=None):
setattr(api_settings, setting, value) setattr(api_settings, setting, value)
if module is not None: if module is not None:
reload(module) six.moves.reload_module(module)
yield yield
setattr(api_settings, setting, original_value) setattr(api_settings, setting, original_value)
if module is not None: if module is not None:
reload(module) six.moves.reload_module(module)