datatracker/ietf/mailtrigger/utils.py

154 lines
4.9 KiB
Python

# Copyright The IETF Trust 2015-2023, All Rights Reserved
from collections import namedtuple
import debug # pyflakes:ignore
from ietf.mailtrigger.models import MailTrigger, Recipient
from ietf.submit.models import Submission
from ietf.utils.mail import excludeaddrs
class AddrLists(namedtuple("AddrLists", ["to", "cc"])):
__slots__ = ()
def as_strings(self, compact=True):
separator = ", " if compact else ",\n "
to_string = separator.join(self.to)
cc_string = separator.join(self.cc)
return namedtuple("AddrListsAsStrings", ["to", "cc"])(
to=to_string, cc=cc_string
)
def gather_address_lists(
slug,
skipped_recipients=None,
create_from_slug_if_not_exists=None,
desc_if_not_exists=None,
**kwargs
):
mailtrigger = get_mailtrigger(
slug, create_from_slug_if_not_exists, desc_if_not_exists
)
to = set()
for recipient in mailtrigger.to.all():
to.update(recipient.gather(**kwargs))
to.discard("")
if skipped_recipients:
to = excludeaddrs(to, skipped_recipients)
cc = set()
for recipient in mailtrigger.cc.all():
cc.update(recipient.gather(**kwargs))
cc.discard("")
if skipped_recipients:
cc = excludeaddrs(cc, skipped_recipients)
return AddrLists(to=sorted(list(to)), cc=sorted(list(cc)))
def get_mailtrigger(slug, create_from_slug_if_not_exists, desc_if_not_exists):
try:
mailtrigger = MailTrigger.objects.get(slug=slug)
except MailTrigger.DoesNotExist:
if create_from_slug_if_not_exists and desc_if_not_exists:
template = MailTrigger.objects.get(slug=create_from_slug_if_not_exists)
mailtrigger = MailTrigger.objects.create(slug=slug, desc=desc_if_not_exists)
mailtrigger.to.set(template.to.all())
mailtrigger.cc.set(template.cc.all())
if slug.startswith("review_completed") and slug.endswith("early"):
mailtrigger.cc.remove("ietf_last_call")
else:
raise
return mailtrigger
def gather_relevant_expansions(**kwargs):
def starts_with(prefix):
return MailTrigger.objects.filter(slug__startswith=prefix).values_list(
"slug", flat=True
)
relevant = set()
if "doc" in kwargs:
doc = kwargs["doc"]
relevant.add("doc_state_edited")
if not doc.type_id in ["bofreq", "statement", "rfc"]:
relevant.update(
["doc_telechat_details_changed", "ballot_deferred", "iesg_ballot_saved"]
)
if doc.type_id in ["draft", "statchg"]:
relevant.update(starts_with("last_call_"))
if doc.type_id == "rfc":
relevant.update(
[
"doc_added_comment",
"doc_external_resource_change_requested",
"doc_state_edited",
"ipr_posted_on_doc",
]
)
if doc.type_id == "draft":
relevant.update(starts_with("doc_"))
relevant.update(starts_with("resurrection_"))
relevant.update(
[
"ipr_posted_on_doc",
]
)
if doc.stream_id == "ietf":
relevant.update(["ballot_approved_ietf_stream", "pubreq_iesg"])
else:
relevant.update(["pubreq_rfced"])
last_submission = (
Submission.objects.filter(name=doc.name, state="posted")
.order_by("-rev")
.first()
)
if last_submission and "submission" not in kwargs:
kwargs["submission"] = last_submission
if doc.type_id == "conflrev":
relevant.update(["conflrev_requested", "ballot_approved_conflrev"])
if doc.type_id == "charter":
relevant.update(["charter_external_review", "ballot_approved_charter"])
if doc.type_id == "bofreq":
relevant.update(starts_with("bofreq"))
if "group" in kwargs:
relevant.update(starts_with("group_"))
relevant.update(starts_with("milestones_"))
group = kwargs["group"]
if group.features.acts_like_wg:
relevant.update(starts_with("session_"))
if group.features.has_chartering_process:
relevant.update(
[
"charter_external_review",
]
)
if "submission" in kwargs:
relevant.update(starts_with("sub_"))
rule_list = []
for mailtrigger in MailTrigger.objects.filter(slug__in=relevant):
addrs = gather_address_lists(mailtrigger.slug, **kwargs)
if addrs.to or addrs.cc:
rule_list.append((mailtrigger.slug, mailtrigger.desc, addrs.to, addrs.cc))
return sorted(rule_list)
def get_base_ipr_request_address():
return Recipient.objects.get(slug="ipr_requests").gather()[0]