使用 Python 条件和 Django 返回 DRY

Question

我要分享的是我的项目中的一个常见模式。有些案例比其他案例更长，但模式或多或少是相同的，我想知道我怎样才能更干。

if status == 'ACTIVE' and p.result != "0":

    messages.add_message(request, messages.ERROR, 
        _('Active Message... bla bla bla'))

    return HttpResponseRedirect(reverse('billing_update'))

if status == 'TOO MANY FAILURES':

    messages.add_message(request, messages.ERROR, 
        _('Failed. Foooobarrrrr'))

    return HttpResponseRedirect(reverse('billing_update'))

if status == 'DEACTIVATED BY MERCHANT':

    messages.add_message(request, messages.ERROR, 
        _('Deactivated derpa derp'))

    return HttpResponseRedirect(reverse('billing_update'))

这里唯一发生的事情是根据状态设置消息。必须有一种方法只调用一次 return，甚至可以收紧条件。

想法？

Answer 1

我注意到您的最后一次返回是有条件的，并且根据我自己的类似代码猜测，如果没有任何检查触发，您想继续并返回其他内容，甚至可能不是重定向...

我认为我会这样做：

def checkForMessage():
    if status == 'ACTIVE' and p.result != "0":
        return 'Active Message... bla bla bla'
    if status == 'TOO MANY FAILURES':
        return 'Failed. Foooobarrrrr'
    if status == 'DEACTIVATED BY MERCHANT':
        return 'Deactivated derpa derp'
    return None

msg = checkForMessage()
if msg:
    messages.add_message(request, messages.ERROR, _(msg))

    return HttpResponseRedirect(reverse('billing_update'))

// ... go on and do some other stuff ...
return direct_to_template('some/template.html', {some: stuff})

请注意，函数 checkForMessage 是在视图函数内部定义的，因此我们不必将测试中涉及的所有内容作为参数传递给它（如果 tets 很多且不同，如果它们只需要状态和一些“p”变量，它也可以在视图函数之外声明并获取这些参数）。

主要的一点是，我们根本不想添加任何消息的“失败”替代方案可以通过在 check 方法中返回 None 并在 view 方法中检查是否存在消息来处理。

Answer 2

可能是这样的：

库模块

def add_error_message(request, status, result=0):
    if status == 'ACTIVE' and not result:
        msg = 'Active Message... bla bla bla'
    elif status == 'TOO MANY FAILURES':
        msg = 'Failed. Foooobarrrrr'
    elif status == 'DEACTIVATED BY MERCHANT':
        msg = 'Deactivated derpa derp'
    messages.add_message(request, messages.ERROR, _(msg))

如果不是'p.result'条件，字典会更好：

MESSAGES = {
    'ACTIVE': 'Active Message... bla bla bla',
    'TOO MANY FAILURES': 'Failed. Foooobarrrrr',
    'DEACTIVATED BY MERCHANT': 'Deactivated derpa derp',
}
def add_error_message(request, status):
    try:
        messages.add_message(request, messages.ERROR, _(MESSAGES[status]))
    except KeyError:
        pass

查看模块

from lib import add_error_message

def your_view(request, *args, **kwargs):
    # ...
    add_error_message(request, status, p.result if p else 0)
    return HttpResponseRedirect(reverse('billing_update'))