1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
|
#!/usr/bin/python -S
#
# Copyright 2009 Canonical Ltd. This software is licensed under the
# GNU Affero General Public License version 3 (see the file LICENSE).
# pylint: disable-msg=C0103,W0403
"""Flag expired team memberships and warn about impending expiration."""
import _pythonpath
import pytz
from datetime import datetime, timedelta
from zope.component import getUtility
from canonical.config import config
from canonical.launchpad.interfaces import (
DAYS_BEFORE_EXPIRATION_WARNING_IS_SENT, ILaunchpadCelebrities,
ITeamMembershipSet)
from lp.services.scripts.base import (
LaunchpadCronScript, LaunchpadScriptFailure)
class ExpireMemberships(LaunchpadCronScript):
"""A script for expired team memberships."""
def flag_expired_memberships_and_send_warnings(self):
"""Flag expired team memberships and warn about impending expiration.
Flag expired team memberships and send warnings for members whose
memberships are going to expire in one week (or less) from now.
"""
membershipset = getUtility(ITeamMembershipSet)
self.txn.begin()
reviewer = getUtility(ILaunchpadCelebrities).janitor
membershipset.handleMembershipsExpiringToday(reviewer)
self.txn.commit()
min_date_for_warning = datetime.now(pytz.timezone('UTC')) + timedelta(
days=DAYS_BEFORE_EXPIRATION_WARNING_IS_SENT)
self.txn.begin()
for membership in membershipset.getMembershipsToExpire(
min_date_for_warning, exclude_autorenewals=True):
membership.sendExpirationWarningEmail()
self.logger.debug("Sent warning email to %s in %s team."
% (membership.person.name, membership.team.name))
self.txn.commit()
def main(self):
"""Flag expired team memberships."""
if self.args:
raise LaunchpadScriptFailure(
"Unhandled arguments %s" % repr(self.args))
self.logger.info("Flagging expired team memberships.")
self.flag_expired_memberships_and_send_warnings()
self.logger.info("Finished flagging expired team memberships.")
if __name__ == '__main__':
script = ExpireMemberships('flag-expired-memberships',
dbuser=config.expiredmembershipsflagger.dbuser)
script.lock_and_run()
|