1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
|
#! /usr/bin/env python3
""" Split a Unix-style mailbox into individual files
Written by Aquarius <aquarius@kryogenix.org>
Usage: splitmbox.py <mailbox-file> <directory>
This will create files numbered 01,02,03... in <directory>. The
number of prefixed zeroes will make all filenames in the
directory the same length. """
import mailbox,sys,getopt,string,os,subprocess
VERSION = '0.1'
USAGE = """Usage: splitmbox.py [ OPTION ] <mailbox-file> <directory>
-h, --help
Show this help text and exit
-v, --version
Show file version and exit
This will create files numbered 01,02,03... in <directory>. The
number of prefixed zeroes will make all filenames in the
directory the same length. """
try:
optlist, args = getopt.getopt(sys.argv[1:],'hv',['help','version'])
except getopt.error, error_text:
print error_text,'\n'
print USAGE
sys.exit(1)
for tuple in optlist:
if tuple[0] == '-h' or tuple[0] == '--help':
print USAGE
sys.exit(0)
if tuple[0] == '-v' or tuple[0] == '--version':
print VERSION
sys.exit(0)
if len(args) != 2:
print USAGE
mbox_fname, output_dir = args
if output_dir[-1] != '/':
output_dir = output_dir + '/'
# Make the output directory, if required
try:
os.mkdir(output_dir)
except os.error,ertxt:
if string.find(str(ertxt),'File exists') == -1:
print 'Failed to create or use directory',output_dir,'[',ertxt,']'
sys.exit(1)
try:
mbox_file = open(mbox_fname)
except:
print "Failed to open file",mbox_fname
sys.exit(1)
mbox = mailbox.UnixMailbox(mbox_file)
# Find out how many messages in the mailbox
count = 0
while 1:
msg = mbox.next()
if not msg:
break
count = count + 1
# Now do it again, outputting files
mbox_file.close()
mbox_file = open(mbox_fname)
mbox = mailbox.UnixMailbox(mbox_file)
digits = len(str(count))
#digits = 3
count = 0
while 1:
msg = mbox.next()
if not msg:
break
fname = output_dir+'msg.'+('0'*digits+str(count))[-digits:]
print 'Writing ', fname
outfile = open(fname,'w')
outfile.write('From foo@baz ');
outfile.write(subprocess.check_output('date'));
for s in msg.headers:
outfile.write(s)
outfile.write('\n')
for s in msg.fp.readlines():
outfile.write(s)
outfile.close()
count = count + 1
|
