Avoid KeyError when uid > max_uid_logged_in

[dsa-puppet.git] / modules / porterbox / files / mail-big-homedirs

diff --git a/modules/porterbox/files/mail-big-homedirs b/modules/porterbox/files/mail-big-homedirs
index f94e836fbd6e43e1b80248091b15f81a49e51271..4211805532d91a25e06f4f6db404eb60edf06071 100755 (executable)
--- a/modules/porterbox/files/mail-big-homedirs
+++ b/modules/porterbox/files/mail-big-homedirs
@@ -3,7 +3,7 @@
 # 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
@@ -25,10 +25,13 @@
 # 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
@@ -40,26 +43,42 @@ import StringIO
 # 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']
-else:
-  SENDMAIL_COMMAND = ['/usr/sbin/sendmail', '-t', '-oi']
-
-if DRYRUN:
   RM_COMMAND = ['/bin/echo', 'Would remove']
 else:
+  SENDMAIL_COMMAND = ['/usr/sbin/sendmail', '-t', '-oi']
   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': 450 }
+    { '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']
+EXCLUDED_USERNAMES = ['lost+found', 'debian']
 MAIL_FROM = 'debian-admin (via Cron) <bulk@admin.debian.org>'
 MAIL_TO = '{username}@{hostname}.debian.org'
 MAIL_CC = 'debian-admin (bulk sink) <bulk@admin.debian.org>'
@@ -167,7 +186,6 @@ class HomedirReminder(object):
   def remove(self, **kwargs):
     try:
       pwinfo = pwd.getpwnam(kwargs.get('username'))
-      return
     except KeyError:
       return
 
@@ -176,13 +194,38 @@ class HomedirReminder(object):
 
   def run(self):
     current_time = time.time()
+    conn = None
+    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:
+      logging.error("Error sending: %s" % e)
+    finally:
+      if conn:
+        conn.close()
 
     for username, homedir_size in self.homedir_sizes.iteritems():
       try:
         realname = pwd.getpwnam(username).pw_gecos.decode('utf-8').split(',', 1)[0]
       except:
         realname = username
-      lastlog_time = self.lastlog_times[username]
+      lastlog_time = self.lastlog_times.get(username, 0)
       days_ago = int( (current_time - lastlog_time) / 3600 / 24 )
       kwargs = {
           'hostname': platform.node(),
@@ -206,7 +249,8 @@ class HomedirReminder(object):
         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()