mailt

1) #! /usr/bin/env python
2) # -*- coding: utf-8 -*-
3) 
4) # imapt -- Simple command-line tool to test IMAP server
5) # By: Benjamin RENARD <brenard@easter-eggs.com>
6) #
7) # Copyright (C) 2013 Easter-eggs

8) # http://git.zionetrix.net/mailt

9) #
10) # This file is part of MailT.
11) #
12) # MailT is free software; you can redistribute it and/or modify
13) # it under the terms of the GNU General Public License as published
14) # by the Free Software Foundation, either version 3 of the License,
15) # or (at your option) any later version.
16) #
17) # MailT is distributed in the hope that it will be useful,
18) # but WITHOUT ANY WARRANTY; without even the implied warranty of
19) # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
20) # GNU Affero General Public License for more details.
21) #
22) # You should have received a copy of the GNU General Public License
23) # along with this program.  If not, see <http://www.gnu.org/licenses/>.
24) 
25) from optparse import OptionParser
26) 
27) import os
28) import imaplib
29) 
30) import logging
31) import sys
32) 
33) parser = OptionParser()
34) 
35) 
36) parser.add_option('-H',

37) 		  '--host',
38) 		  action="store",
39) 		  type="string",
40) 		  dest="host",
41) 		  help="The IMAP host",
42) 		  default="127.0.0.1")

43) 
44) parser.add_option('-p',

45) 		  '--port',
46) 		  action="store",
47) 		  type="string",
48) 		  dest="port",
49) 		  help="The IMAP port",
50) 		  default=None)

51) 
52) parser.add_option('-S',

53) 		  '--ssl',
54) 		  action="store_true",
55) 		  dest="ssl",
56) 		  help="IMAP Over SSL (IMAPS)",
57) 		  default=False)

58) 
59) parser.add_option('-U',

60) 		  '--user',
61) 		  action="store",
62) 		  type="string",
63) 		  dest="user",
64) 		  help="User login",
65) 		  default=None)

66) 
67) parser.add_option('-P',

68) 		  '--password',
69) 		  action="store",
70) 		  type="string",
71) 		  dest="password",
72) 		  help="User password",
73) 		  default=None)

74) 
75) parser.add_option('--md5',

76) 		  action="store_true",
77) 		  dest="md5",
78) 		  help="Use CRAM-MD5 password for authentication",
79) 		  default=False)

80) 
81) parser.add_option('-m',

82) 		  '--mailbox',
83) 		  action="store",
84) 		  type="string",
85) 		  dest="mailbox",
86) 		  help="The mailbox directory to select",
87) 		  default='INBOX')

88) 
89) parser.add_option('-v',

90) 		  '--verbose',
91) 		  action="store_true",
92) 		  dest="verbose")

93) 

94) parser.add_option('-l',

95) 		  '--list',
96) 		  action="store_true",
97) 		  dest="list",
98) 		  help="List mailboxes",
99) 		  default=False)

100) 

101) parser.add_option('--expunge',

102) 		  action="store",
103) 		  type="string",
104) 		  dest="expunge",
105) 		  help="Expunge mailbox specify (ALL to expunge all mailboxes)",
106) 		  default=None)

107) 

108) (options, args) = parser.parse_args()
109) 
110) if options.verbose:
111) 	loglevel=logging.DEBUG
112) else:
113) 	loglevel=logging.INFO
114) 
115) logging.basicConfig(level=loglevel,format='%(asctime)s - %(levelname)s - %(message)s', datefmt='%m/%d/%Y %I:%M:%S %p')
116) 
117) if not options.user or not options.password:
118) 	logging.fatal('You must provide the user login and password')
119) 	sys.exit(1)
120) 
121) if options.port is None:
122) 	if options.ssl:
123) 		options.port=993
124) 	else:
125) 		options.port=143
126) 
127) if options.ssl:
128) 	logging.debug('Establish SSL connexion to server')
129) 	try:
130) 		server = imaplib.IMAP4_SSL(options.host,options.port)
131) 		logging.debug('Connected')
132) 	except Exception, e:
133) 		logging.critical("IMAP connection error : %s" % e)
134) 		sys.exit(2)
135) else:
136) 	logging.debug("Establish connexion to server")
137) 	try:
138) 		server = imaplib.IMAP4(options.host,options.port)
139) 		logging.debug('Connected')
140) 	except Exception, e:
141) 		logging.critical("IMAP connection error : %s" % e)
142) 		sys.exit(2)
143) 
144) if options.user:
145) 	try:
146) 		if options.md5:
147) 			logging.debug('Login into server as %s (with CRAM-MD5 password)' % options.user)
148) 			server.login_cram_md5(options.user,options.password)
149) 		else:
150) 			logging.debug('Login into server as %s' % options.user)
151) 			server.login(options.user,options.password)
152) 		logging.debug('Logged')
153) 	except Exception as e:
154) 		logging.critical("IMAP Auth error : %s" % e)
155) 		sys.exit(3)

156) 
157) def expunge(server,mailbox):
158) 	logging.info('Expunge %s mailbox' % mailbox)
159) 	try:
160) 		(st,c) = server.select(mailbox)
161) 		if st!='OK':
162) 			logging.error('Error selecting %s mailbox : %s' % (mailbox,st))
163) 			return
164) 		(st,ex) = server.expunge()
165) 		if st=='OK':
166) 			if len(ex)==1 and ex[0] is None:
167) 				logging.info('Mailbox %s expunged, no mail removed' % mailbox)
168) 			else:
169) 				logging.info('Mailbox %s expunged : %s mail(s) removed' % (mailbox,len(ex)))
170) 		else:
171) 			logging.info('Error expunging %s mailbox, return "%s"' % (mailbox,st))
172) 	except Exception as e:
173) 		logging.error('Error expunging %s mailbox : %s' % (mailbox,e))
174) 
175) 

176) try:

177) 	if options.list:
178) 		logging.info('List mailbox')
179) 		(status,l) = server.list()
180) 		for d in l:

181) 			logging.info('   "%s"' % d.split('"')[3])
182) 	elif options.expunge:
183) 		if options.expunge=='ALL':
184) 			logging.debug("Listing mailboxes")
185) 			(status,l) = server.list()
186) 			for d in l:
187) 				dirname=d.split('"')[3]
188) 				expunge(server,dirname)
189) 		else:
190) 			expunge(server,options.expunge)

191) 	else:

192) 		logging.debug("Select mailbox")
193) 		ret = server.select(options.mailbox)
194) 		if len(ret)>0 and ret[0]=='OK':
195) 			logging.info("Mailbox %s content %s message(s)" % (options.mailbox,ret[1][0]))
196) 		else:
197) 			logging.error("Selecting return incorrected : %s" % ret)
198) 		server.close()

199) except Exception as e:

200) 	logging.critical('Error selecting mailbox %s : %s' % (options.mailbox,e))