Pick out all of the IP Addresses listed in Logwatch emails from within a .mbox MailBox

ip_finder.py

import mailbox, re, argparse
import cPickle as pickle

def gen_summary(filename):
	ip_store = dict()

	my_box = mailbox.mbox(filename)

	for message in my_box:
		subject = message['subject']       # Could possibly be None.
		if subject and 'Logwatch for' in subject:
			ip_addresses = ip_finder(message.get_payload())
			for x in ip_addresses:
				if x in ip_store:
					ip_store[x] = ip_store[x] +1
				else:
					ip_store[x] = 1
 # save the output in Pickle format for later use
	pickle.dump( ip_store, open( "ip_store.p", "wb" ) )				
	return ip_store

def ip_finder(message):
	ip = re.findall( r'\b(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\.(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\.(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\.(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\b', message)
	return ip


''' Take filename as input from the command-line '''
if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument("filename", help="path to MBOX file to be scraped; of type .mbox")
    args = parser.parse_args()

    print gen_summary(args.filename)