Guest User

Untitled

a guest
Mar 15th, 2018
97
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.91 KB | None | 0 0
  1. #!/usr/bin/python
  2.  
  3. import sys
  4. import MySQLdb
  5. from email.parser import Parser
  6.  
  7. DB_HOST = 'derp.com'
  8. DB_USER = 'derp_user'
  9. DB_PASS = 'derpish'
  10. DB_NAME = 'some_db_name'
  11.  
  12. def db_insert(*args):
  13.     try:
  14.         con = MySQLdb.connect(host=DB_HOST, user=DB_USER, passwd=DB_PASS, db=DB_NAME)
  15.         cur = con.cursor()
  16.     except Exception, e:
  17.         print "Error connection to the DB: %s" % e
  18.         sys.exit(1)
  19.        
  20.     query = "INSERT INTO `table` (from, to, subject, body) VALUES(%s, %s, %s, %s)"
  21.     cur.execute(query % args)
  22.    
  23. def parse_headers(msg):
  24.     msg = msg.replace('\\r\\n', '\r\n')
  25.     try:
  26.         msg_obj = Parser().parsestr(msg)
  27.         to = msg_obj['To']
  28.         frm = msg_obj['From']
  29.         subject = msg_obj['Subject']
  30.         body = msg_obj.get_payload()
  31.     except Exception, e:
  32.         print "Error parsing email: %s" % e
  33.         sys.exit(1)
  34.    
  35.     db_insert(to, frm, subject, body)
  36.    
  37. if __name__ == '__main__':
  38.     f = sys.stdin
  39.     if f:
  40.         parse_headers(f.read())
Add Comment
Please, Sign In to add comment