2iccnY5V

1import re
2import sqlite3
3
4
5
6conn = sqlite3.connect('emaildb.sqlite')
7cur = conn.cursor()
8
9cur.execute('''
10DROP TABLE IF EXISTS Counts''')
11
12cur.execute('''
13CREATE TABLE Counts (org TEXT, count INTEGER)''')
14
15fname = raw_input('Enter file name: ')
16if ( len(fname) < 1 ) : fname = 'mbox.txt'
17fh = open(fname)
18for line in fh:
19	if not line.startswith('From: ') : continue
20	pieces = line.split()[1]
21	org = pieces.split('@')[1]
22	
23	
24	print org
25	cur.execute('SELECT count FROM Counts WHERE org = ? ', (org, ))
26	row = cur.fetchone()
27	if row is None:
28		cur.execute('''INSERT INTO Counts (org, count) 
29				VALUES ( ?, 1 )''', ( org, ) )
30	else : 
31		cur.execute('UPDATE Counts SET count=count+1 WHERE org = ?', 
32			(org, ))
33
34conn.commit()
35
36
37sqlstr = 'SELECT org, count FROM Counts ORDER BY count DESC LIMIT 10'
38
39print
40print "Counts:"
41for row in cur.execute(sqlstr) :
42    print str(row[0]), row[1]
43
44cur.close()