# Send email reminders to users having sizable homedirs.
##
# Copyright (c) 2013 Philipp Kern <pkern@debian.org>
-# Copyright (c) 2013 Peter Palfrader <peter@palfrader.org>
+# Copyright (c) 2013, 2014 Peter Palfrader <peter@palfrader.org>
# Copyright (c) 2013 Luca Filipozzi <lfilipoz@debian.org>
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# THE SOFTWARE.
from collections import defaultdict
+from dsa_mq.connection import Connection
+from dsa_mq.config import Config
import email
import email.mime.text
import glob
import logging
+from optparse import OptionParser
import os.path
import platform
import pwd
# avoid base64 encoding for utf-8
email.charset.add_charset('utf-8', email.charset.SHORTEST, email.charset.QP)
-DRYRUN = False
-
-if DRYRUN:
+parser = OptionParser()
+parser.add_option("-D", "--dryrun",
+ action="store_true", default=False,
+ help="Dry run mode")
+
+parser.add_option("-d", "--debug",
+ action="store_true", default=False,
+ help="Enable debug output")
+
+(options, args) = parser.parse_args()
+options.section = 'dsa-homedirs'
+options.config = '/etc/dsa/pubsub.conf'
+config = Config(options)
+mq_conf = {
+ 'rabbit_userid': config.username,
+ 'rabbit_password': config.password,
+ 'rabbit_virtual_host': config.vhost,
+ 'rabbit_hosts': ['pubsub02.debian.org', 'pubsub01.debian.org'],
+ 'use_ssl': False
+}
+
+if options.dryrun:
SENDMAIL_COMMAND = ['/bin/cat']
RM_COMMAND = ['/bin/echo', 'Would remove']
else:
RM_COMMAND = ['/bin/rm', '-rf']
CRITERIA = [
- { 'size': 10240, 'notifyafter': 5}, #, 'deleteafter': 40 },
- { 'size': 1024, 'notifyafter': 10}, #, 'deleteafter': 50 },
- { 'size': 100, 'notifyafter': 30}, #, 'deleteafter': 90 },
- { 'size': 20, 'notifyafter': 90}, #, 'deleteafter': 150 },
- { 'size': 5, 'deleteafter': 700 }
+ { 'size': 10240, 'notifyafter': 5, 'deleteafter': 40 },
+ { 'size': 1024, 'notifyafter': 10, 'deleteafter': 50 },
+ { 'size': 100, 'notifyafter': 30, 'deleteafter': 90 },
+ { 'size': 20, 'notifyafter': 90, 'deleteafter': 150 },
+ { 'size': 5, 'deleteafter': 700 }
]
EXCLUDED_USERNAMES = ['lost+found', 'debian']
MAIL_FROM = 'debian-admin (via Cron) <bulk@admin.debian.org>'
def remove(self, **kwargs):
try:
pwinfo = pwd.getpwnam(kwargs.get('username'))
- return
except KeyError:
return
def run(self):
current_time = time.time()
+ try:
+ data = {}
+ for user in set(self.homedir_sizes.keys()) +
+ set(self.lastlog_times.keys()):
+ data[user] = {
+ 'homedir': self.homedir_sizes.get(user, 0),
+ 'lastlog': self.lastlog_times.get(user, 0),
+ }
+
+ msg = {
+ 'timestamp': current_time,
+ 'data': data,
+ 'host': platform.node(),
+ }
+ conn = Connection(conf=mq_conf)
+ conn.topic_send(config.topic,
+ msg,
+ exchange_name=config.exchange,
+ timeout=5)
+ except Exception, e:
+ LOG.error("Error sending: %s" % e)
+ finally:
+ if conn:
+ conn.close()
for username, homedir_size in self.homedir_sizes.iteritems():
try:
self.notify(**kwargs)
if __name__ == '__main__':
- logging.basicConfig()
- # DEBUG for debugging, ERROR for production.
- logging.getLogger().setLevel(logging.ERROR)
+ lvl = logging.ERROR
+ if options.debug:
+ lvl = logging.DEBUG
+ logging.basicConfig(level=lvl)
HomedirReminder().run()