Limpieza selectiva de buzones IMAP4 (II)

 #!/usr/bin/env python3

 # (c) 2015-2016 Jesús Cea Avión - jcea@jcea.es
 # This code is licensed under AGPLv3.

 import sys, os, argparse
 import imaplib
 import time, locale
 from collections import defaultdict
 from itertools import chain

 from imapclient import imap_utf7

 # Para el tiempo
 locale.setlocale(locale.LC_ALL, 'en_US.utf8')

 imaplib._MAXLINE = 1000000000

 parser = argparse.ArgumentParser(description='''Procesa un buzón IMAP4 y
 conserva los hilos donde hemos escrito nosotros o que contienen mensajes
 de menos de un mes. El resto de mensajes son enviados a la salida
 estándar y borrados del servidor IMAP4.

 Dado que es una acción destructiva, por defecto se realiza el análisis
 y se muestra cuál sería el resultado si se aplicase realmente, pero no
 la acción no se realiza a menos que pasemos el parámetro "--do".''')
 parser.add_argument('--do', action='store_true', help='Realiza la acción')
 parser.add_argument('--all', action='store_true',
         help='No conservar hilos con mensajes propios o recientes')
 parser.add_argument('buzon', metavar='buzón', help='Buzón a procesar')
 args = parser.parse_args()
 do = args.do
 archive_all = args.all
 buzon = '"%s"' %imap_utf7.encode(args.buzon).decode('latin-1')

 imap = imaplib.IMAP4('127.0.0.1')
 imap.login('X', 'X')
 reply = imap.select(buzon)
 if reply[0] != 'OK' :
     raise RuntimeError(reply[0])

 threads = defaultdict(list)
 if archive_all :
     reply = imap.search(None, 'not deleted')
     if reply[0] != 'OK' :
         raise RuntimeError(reply[0])
     reply = reply[1][0].decode('latin-1')
     ids = list(map(int, reply.split()))
     threads[0] = ids
 else :
     reply = imap.thread('REFERENCES', 'utf-8', 'not deleted')
     if reply[0] != 'OK' :
         raise RuntimeError(reply[0])
     reply = reply[1][0].decode('latin-1')
     num_thread = 0
     ids = {}
     indent = 0
     token = ''
     for i in reply :
         if i == '(' :
                 indent += 1
                 if indent == 1 :
                     num_thread += 1
         elif i == ')' :
             indent -= 1
             if token :
                 token = int(token)
                 ids[token] = num_thread
                 threads[num_thread].append(token)
                 token = ''
         elif i == ' ' :
             token = int(token)
             ids[token] = num_thread
             threads[num_thread].append(token)
             token = ''
         else :
             token += i

 num_hilos_ignorados = 0
 num_mensajes_ignorados = 0
 if not archive_all :
     ventana = time.localtime(time.time()-86400*30)
     reply = imap.search('utf-8', 'OR', 'FROM', 'jcea@', 'SINCE',
             time.strftime('%d-%b-%Y', ventana))
     if reply[0] != 'OK' :
         raise RuntimeError(reply[0])
     mensajes_a_ignorar = set(reply[1][0].decode('latin-1').split())
     for msg in mensajes_a_ignorar :
         msg = int(msg)
         thread = ids[msg]
         if thread in threads :
             num_hilos_ignorados += 1
             num_mensajes_ignorados += len(threads[thread])
             del threads[thread]
 else :
     mensajes_a_ignorar = set()

 print("Mensajes en total: %d - Hilos a conservar: %d - Mensajes a conservar: %d"
         %(len(ids), num_hilos_ignorados, num_mensajes_ignorados),
         file=sys.stderr)

 if not do :
     print("Para realizar la operación, añade el parámetro '--do'",
             file=sys.stderr)
     sys.exit()

 if sys.stdout.isatty() :
     raise RuntimeError('Debes redirigir STDOUT')

 mensajes_a_volcar = sorted(chain.from_iterable(threads.values()))

 devnull = os.stat('/dev/null').st_rdev == \
              os.fstat(sys.stdout.fileno()).st_rdev

 if devnull :
     print("Como redirigimos a '/dev/null', no hacemos el volcado",
             file=sys.stderr)
 else :
     for i in mensajes_a_volcar :
         print('Volcando mensaje', i, end='\r', file=sys.stderr)
         #reply = imap.fetch(str(i), '(INTERNALDATE RFC822)')
         reply = imap.fetch(str(i), '(INTERNALDATE BODY.PEEK[])')
         if reply[0] != 'OK' :
             raise RuntimeError(reply[0])
         date = reply[1][0][0].decode('latin-1').split()
         t = time.strptime(date[2]+' '+date[3], '"%d-%b-%Y %H:%M:%S')
         separador = time.strftime("From - %a %b %d %H:%M:%S %Y\r\n", t)
         sys.stdout.buffer.write(separador.encode('latin-1'))
         msg = reply[1][0][1].replace(b'\nFrom ', b'\n>From ')
         sys.stdout.buffer.write(msg)

     sys.stdout.buffer.flush()
     fd = sys.stdout.buffer.fileno()
     try :
         os.fsync(fd)
     except OSError :  # En Linux un fsync sobre "/dev/null" falla
         pass

     print(' '*30, end='\r', file=sys.stderr)

 if len(mensajes_a_volcar) :
     runs = []
     runstart = runend = mensajes_a_volcar[0]
     for i in mensajes_a_volcar[1:] :
         if i == runend + 1 :
             runend += 1
             continue
         runs.append((runstart, runend))
         runstart = runend = i
     else :
         runs.append((runstart, runend))

     for runstart, runend in runs :
         if runstart == runend :
             msg_set = str(runstart)
         else :
             msg_set = "%d:%d" %(runstart, runend)


         print('BORRANDO MENSAJE(S) %s' %msg_set + ' '*30, \
                 end='\r', file=sys.stderr)

         reply = imap.store(msg_set, '+FLAGS', '\\Deleted')
         if reply[0] != 'OK' :
             raise RuntimeError(reply[0])
     imap.close()  # EXPUNGE implícito
     print(' '*60, end='\r', file=sys.stderr)

 imap.logout()