| 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()
|
|---|
