PEP343「與」上下文管理和Django的

Question

我做的和Django框架工作的一些應用程序的測試， 我有一個情況我測試，如果不活動的用戶可以登錄，我真的很喜歡這樣

self.testuser.is_active = False 
//DO testing 
self.testuser.is_active = True 
//Proceed 

我的問題是， 通過與PEP343 提供上下文管理使用我試圖這樣做，但我失敗了

with self.testuser.is_active = False : 
//code 

然後我試圖做

with self.settings(self.__set_attr(self.testuser.is_active = False)): 
//code 

它也失敗

是有辦法解決？或者我必須定義一個將is_active設置爲false的函數？

Answer 1

這裏是contextlib建立在一個更加通用的上下文管理。

from contextlib import contextmanager 

@contextmanager 
def temporary_changed_attr(object, attr, value): 
    if hasattr(object, attr): 
     old = getattr(object, attr) 
     setattr(object, attr, value) 
     yield 
     setattr(object, attr, old) 
    else: 
     setattr(object, attr, value) 
     yield 
     delattr(object, attr) 

# Example usage 
with temporary_changed_attr(self.testuser, 'is_active', False): 
    # self.testuser.is_active will be false in here 

Answer 2

你必須編寫自己的上下文管理器。這裏有一個對你的情況（使用contextlib）：

import contextlib 
@contextlib.contextmanager 
def toggle_active_user(user): 
    user.is_active = False 
    yield 
    user.is_active = True 

with toggle_active_user(self.testuser): 
    // Do testing 

更妙的是，將保存狀態之前，然後還原：

import contextlib 
@contextlib.contextmanager 
def toggle_active_user(user, new_value): 
    previous_is_active = user.is_active 
    user.is_active = new_value 
    yield 
    user.is_active = previous_is_active 

with toggle_active_user(self.testuser, False): 
    // Do testing