python - 用于批量电子邮件选择退出的简单 Python 脚本

Question

我有一个 Python 脚本，可以让用户退出电子邮件。

optouts = [
"user1@example.com",
"user2@example.com",
"user3@example.com",
]

for email in optouts:
    user = User.objects.get(email=email)
    profile = user
    profile.allow_mass_mails = False
    profile.save()
    print email, "opted out."


print "done."

大约有 10,000 封电子邮件被选择退出。但是，每当它发现与已删除帐户关联的电子邮件时，它都会显示:

django.contrib.auth.models.DoesNotExist: User matching query does not exist.

如果用户配置文件不存在，我希望它显示“已跳过”(就像打印完成一样)。

我尝试添加“else print “skipped”，但这不起作用。我是 Python 新手，非常感谢任何帮助。

Answer 1

您可以使用 continue 捕获异常并跳过记录。通过在模型对象本身上使用异常，User.DoesNotExist:

for email in optouts:
    try:
        user = User.objects.get(email=email)
    except User.DoesNotExist:
        print email, "skipped."
        continue
    profile = user
    profile.allow_mass_mails = False
    profile.save()
    print email, "opted out."

或者通过显式导入基本异常，django.core.exceptions.ObjectDoesNotExist :

from django.core.exceptions import ObjectDoesNotExist

for email in optouts:
    try:
        user = User.objects.get(email=email)
    except ObjectDoesNotExist:
        print email, "skipped."
        continue
    profile = user
    profile.allow_mass_mails = False
    profile.save()
    print email, "opted out."