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
64
65
66
67
68
69
70
|
#!/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
from datetime import (
datetime,
timedelta,
)
import pytz
from zope.component import getUtility
from lp.app.interfaces.launchpad import ILaunchpadCelebrities
from lp.registry.interfaces.teammembership import (
DAYS_BEFORE_EXPIRATION_WARNING_IS_SENT,
ITeamMembershipSet,
)
from lp.services.config import config
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()
|