1 #! /usr/bin/env python

     2 

     3 # Scan MH folder, display results in window

     4 

     5 import os

     6 import sys

     7 import re

     8 import getopt

     9 import string

    10 import mhlib

    11 

    12 from Tkinter import *

    13 

    14 from dialog import dialog

    15 

    16 mailbox = os.environ['HOME'] + '/Mail'

    17 

    18 def main():

    19     global root, tk, top, mid, bot

    20     global folderbox, foldermenu, scanbox, scanmenu, viewer

    21     global folder, seq

    22     global mh, mhf

    23 

    24     # Parse command line options

    25 

    26     folder = 'inbox'

    27     seq = 'all'

    28     try:

    29         opts, args = getopt.getopt(sys.argv[1:], '')

    30     except getopt.error, msg:

    31         print msg

    32         sys.exit(2)

    33     for arg in args:

    34         if arg[:1] == '+':

    35             folder = arg[1:]

    36         else:

    37             seq = arg

    38 

    39     # Initialize MH

    40 

    41     mh = mhlib.MH()

    42     mhf = mh.openfolder(folder)

    43 

    44     # Build widget hierarchy

    45 

    46     root = Tk()

    47     tk = root.tk

    48 

    49     top = Frame(root)

    50     top.pack({'expand': 1, 'fill': 'both'})

    51 

    52     # Build right part: folder list

    53 

    54     right = Frame(top)

    55     right.pack({'fill': 'y', 'side': 'right'})

    56 

    57     folderbar = Scrollbar(right, {'relief': 'sunken', 'bd': 2})

    58     folderbar.pack({'fill': 'y', 'side': 'right'})

    59 

    60     folderbox = Listbox(right, {'exportselection': 0})

    61     folderbox.pack({'expand': 1, 'fill': 'both', 'side': 'left'})

    62 

    63     foldermenu = Menu(root)

    64     foldermenu.add('command',

    65                    {'label': 'Open Folder',

    66                     'command': open_folder})

    67     foldermenu.add('separator')

    68     foldermenu.add('command',

    69                    {'label': 'Quit',

    70                     'command': 'exit'})

    71     foldermenu.bind('<ButtonRelease-3>', folder_unpost)

    72 

    73     folderbox['yscrollcommand'] = (folderbar, 'set')

    74     folderbar['command'] = (folderbox, 'yview')

    75     folderbox.bind('<Double-1>', open_folder, 1)

    76     folderbox.bind('<3>', folder_post)

    77 

    78     # Build left part: scan list

    79 

    80     left = Frame(top)

    81     left.pack({'expand': 1, 'fill': 'both', 'side': 'left'})

    82 

    83     scanbar = Scrollbar(left, {'relief': 'sunken', 'bd': 2})

    84     scanbar.pack({'fill': 'y', 'side': 'right'})

    85 

    86     scanbox = Listbox(left, {'font': 'fixed'})

    87     scanbox.pack({'expand': 1, 'fill': 'both', 'side': 'left'})

    88 

    89     scanmenu = Menu(root)

    90     scanmenu.add('command',

    91                  {'label': 'Open Message',

    92                   'command': open_message})

    93     scanmenu.add('command',

    94                  {'label': 'Remove Message',

    95                   'command': remove_message})

    96     scanmenu.add('command',

    97                  {'label': 'Refile Message',

    98                   'command': refile_message})

    99     scanmenu.add('separator')

   100     scanmenu.add('command',

   101                  {'label': 'Quit',

   102                   'command': 'exit'})

   103     scanmenu.bind('<ButtonRelease-3>', scan_unpost)

   104 

   105     scanbox['yscrollcommand'] = (scanbar, 'set')

   106     scanbar['command'] = (scanbox, 'yview')

   107     scanbox.bind('<Double-1>', open_message)

   108     scanbox.bind('<3>', scan_post)

   109 

   110     # Separator between middle and bottom part

   111 

   112     rule2 = Frame(root, {'bg': 'black'})

   113     rule2.pack({'fill': 'x'})

   114 

   115     # Build bottom part: current message

   116 

   117     bot = Frame(root)

   118     bot.pack({'expand': 1, 'fill': 'both'})

   119     #

   120     viewer = None

   121 

   122     # Window manager commands

   123 

   124     root.minsize(800, 1) # Make window resizable

   125 

   126     # Fill folderbox with text

   127 

   128     setfolders()

   129 

   130     # Fill scanbox with text

   131 

   132     rescan()

   133 

   134     # Enter mainloop

   135 

   136     root.mainloop()

   137 

   138 def folder_post(e):

   139     x, y = e.x_root, e.y_root

   140     foldermenu.post(x - 10, y - 10)

   141     foldermenu.grab_set()

   142 

   143 def folder_unpost(e):

   144     tk.call('update', 'idletasks')

   145     foldermenu.grab_release()

   146     foldermenu.unpost()

   147     foldermenu.invoke('active')

   148 

   149 def scan_post(e):

   150     x, y = e.x_root, e.y_root

   151     scanmenu.post(x - 10, y - 10)

   152     scanmenu.grab_set()

   153 

   154 def scan_unpost(e):

   155     tk.call('update', 'idletasks')

   156     scanmenu.grab_release()

   157     scanmenu.unpost()

   158     scanmenu.invoke('active')

   159 

   160 scanparser = re.compile('^ *([0-9]+)')

   161 

   162 def open_folder(e=None):

   163     global folder, mhf

   164     sel = folderbox.curselection()

   165     if len(sel) != 1:

   166         if len(sel) > 1:

   167             msg = "Please open one folder at a time"

   168         else:

   169             msg = "Please select a folder to open"

   170         dialog(root, "Can't Open Folder", msg, "", 0, "OK")

   171         return

   172     i = sel[0]

   173     folder = folderbox.get(i)

   174     mhf = mh.openfolder(folder)

   175     rescan()

   176 

   177 def open_message(e=None):

   178     global viewer

   179     sel = scanbox.curselection()

   180     if len(sel) != 1:

   181         if len(sel) > 1:

   182             msg = "Please open one message at a time"

   183         else:

   184             msg = "Please select a message to open"

   185         dialog(root, "Can't Open Message", msg, "", 0, "OK")

   186         return

   187     cursor = scanbox['cursor']

   188     scanbox['cursor'] = 'watch'

   189     tk.call('update', 'idletasks')

   190     i = sel[0]

   191     line = scanbox.get(i)

   192     if scanparser.match(line) >= 0:

   193         num = string.atoi(scanparser.group(1))

   194         m = mhf.openmessage(num)

   195         if viewer: viewer.destroy()

   196         from MimeViewer import MimeViewer

   197         viewer = MimeViewer(bot, '+%s/%d' % (folder, num), m)

   198         viewer.pack()

   199         viewer.show()

   200     scanbox['cursor'] = cursor

   201 

   202 def interestingheader(header):

   203     return header != 'received'

   204 

   205 def remove_message(e=None):

   206     itop = scanbox.nearest(0)

   207     sel = scanbox.curselection()

   208     if not sel:

   209         dialog(root, "No Message To Remove",

   210                "Please select a message to remove", "", 0, "OK")

   211         return

   212     todo = []

   213     for i in sel:

   214         line = scanbox.get(i)

   215         if scanparser.match(line) >= 0:

   216             todo.append(string.atoi(scanparser.group(1)))

   217     mhf.removemessages(todo)

   218     rescan()

   219     fixfocus(min(todo), itop)

   220 

   221 lastrefile = ''

   222 tofolder = None

   223 def refile_message(e=None):

   224     global lastrefile, tofolder

   225     itop = scanbox.nearest(0)

   226     sel = scanbox.curselection()

   227     if not sel:

   228         dialog(root, "No Message To Refile",

   229                "Please select a message to refile", "", 0, "OK")

   230         return

   231     foldersel = folderbox.curselection()

   232     if len(foldersel) != 1:

   233         if not foldersel:

   234             msg = "Please select a folder to refile to"

   235         else:

   236             msg = "Please select exactly one folder to refile to"

   237         dialog(root, "No Folder To Refile", msg, "", 0, "OK")

   238         return

   239     refileto = folderbox.get(foldersel[0])

   240     todo = []

   241     for i in sel:

   242         line = scanbox.get(i)

   243         if scanparser.match(line) >= 0:

   244             todo.append(string.atoi(scanparser.group(1)))

   245     if lastrefile != refileto or not tofolder:

   246         lastrefile = refileto

   247         tofolder = None

   248         tofolder = mh.openfolder(lastrefile)

   249     mhf.refilemessages(todo, tofolder)

   250     rescan()

   251     fixfocus(min(todo), itop)

   252 

   253 def fixfocus(near, itop):

   254     n = scanbox.size()

   255     for i in range(n):

   256         line = scanbox.get(repr(i))

   257         if scanparser.match(line) >= 0:

   258             num = string.atoi(scanparser.group(1))

   259             if num >= near:

   260                 break

   261     else:

   262         i = 'end'

   263     scanbox.select_from(i)

   264     scanbox.yview(itop)

   265 

   266 def setfolders():

   267     folderbox.delete(0, 'end')

   268     for fn in mh.listallfolders():

   269         folderbox.insert('end', fn)

   270 

   271 def rescan():

   272     global viewer

   273     if viewer:

   274         viewer.destroy()

   275         viewer = None

   276     scanbox.delete(0, 'end')

   277     for line in scanfolder(folder, seq):

   278         scanbox.insert('end', line)

   279 

   280 def scanfolder(folder = 'inbox', sequence = 'all'):

   281     return map(

   282             lambda line: line[:-1],

   283             os.popen('scan +%s %s' % (folder, sequence), 'r').readlines())

   284 

   285 main()