我应该如何重构一段存在于许多不同函数中但包含 return 语句的代码

Question

代码片段来自django视图代码，但这并不重要。

假设我有以下代码...

def unsubscribe(request):
  #start of block
  user = request.user
  sid = request.POST.get('subscription_id')
  try:
    sub = Subscription.objects.get(id=sid)
  except ObjectDoesNotExist:
    return ajax_response(False, [('subscription', 'Given subscription does not exist.')])
  if sub.user != user:
    return ajax_response(False, [('subscription', 'Invalid permission.')])
  #end of block.
  sub.is_active = False
  sub.save()
  return ajax_response(True)

并且说，我有另一个函数 resubscribe() ，它与上面的函数执行完全相同的操作，除了它执行 sub.is_active = True。

在这种情况下，组织代码的最佳方式是什么，以便 #block 和 #endblock 之间的代码不重复？我想一般来说，这个问题可以写成：

有一个代码块被复制并粘贴到许多不同的函数上。但是，此代码块包含 return 语句。在这种情况下，抽象出这段逻辑的最佳方法是什么？

编辑：固定代码片段。

EDIT2：实际上，解决这个问题的简单方法是创建一个函数，例如toggle_active_status，它接受 request 和一个布尔值。 （我在发布问题后就弄清楚了）。

但是，我想知道函数之间的不同逻辑超过 1 行的情况...例如，#block 和 #endlbock 之间的代码仅进行输入验证，并且可能会出现任何任意应用程序逻辑。

Answer 1

分解出重复的块并传入差异，创建传递差异的包装函数：

def alter_subscription(request, make_active):
  # start of block
  # (...)
  # end of block
  sub.is_active = make_active
  sub.save()
  return ajax_response(True)

def unsubscribe(request):
  return alter_subscription(request, False)

def resubscribe(request):
  return alter_subscription(request, True)

Answer 2

尚未显示的几个选项：

由于这是一个验证例程，因此使用异常来指示失败，并使用正常的 None（忽略）返回来指示成功：

def verify_permissions(request):
  user = request.user
  sid = request.POST.get('subscription_id')
  try:
    sub = Subscription.objects.get(id=sid)
  except ObjectDoesNotExist:
    raise PermissionError, 'Given subscription does not exist.'
  if sub.user != user:
    raise PermissionError, 'Invalid permission.'

def subscribe(request):
  try:
    verify_permissions(request)
    sub.save()
    return ajax_response(True)
  except PermissionError, why:
    return ajax_response(False, [('subscription', why)])

或者，由于 Python 是动态类型的，因此调用 < code>ajax_response 有几种不同的方式：返回用于构造响应的参数，并检查第一个参数的值。

def verify_permissions(request, purpose):
  user = request.user
  sid = request.POST.get('subscription_id')
  try:
    sub = Subscription.objects.get(id=sid)
  except ObjectDoesNotExist:
    return (False, [(purpose, 'Given subscription does not exist.')]) 
  if sub.user != user:
    return (False, [(purpose, 'Invalid permission.')])
  return (True,)

def subscribe(request):
  result = verify_permissions(request, 'subscription')
  if result[0]: sub.save()
  return ajax_response(*result)

Answer 3

一种方法是简单地创建一个包含重复功能的单独函数。您可以使用函数的参数来引入必要的差异。

如果代码片段返回某些内容，那么您可以使用约定，例如如果函数不应返回则返回 None ，否则返回对象。在调用函数中，只需测试返回的参数是否为 None ，如果不是 None 则返回返回的变量。