altf4arnold/datatracker
1
0
Fork 0
Code Issues Pull requests Actions 6 Packages Projects Releases Wiki Activity

chore: black

Browse source
This commit is contained in:
Robert Sparks 2023-09-15 16:15:47 -05:00
parent edadd99044
commit bc177b08ea
No known key found for this signature in database
GPG key ID: 6E2A6A5775F91318
1 changed files with 87 additions and 64 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

151
ietf/mailtrigger/utils.py
View file

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