История изменений

# -*- coding: utf-8 -*-

from email.parser import BytesParser, Parser
from email.policy import default
import mailbox
import MySQLdb as mdb
import quopri
import string
import uuid
import base64
import re
import time

con = mdb.connect('localhost', '', '', '', use_unicode=True, charset="utf8");

def encoded_words_to_text(encoded_words):
	encoded_word_regex = r'=\?{1}(.+)\?{1}([B|Q])\?{1}(.+)\?{1}='
	try:
		charset, encoding, encoded_text = re.match(encoded_word_regex, encoded_words).groups()
	except Exception as e:
		return None
	if encoding is 'B':
		byte_string = base64.b64decode(encoded_text)
	elif encoding is 'Q':
		byte_string = quopri.decodestring(encoded_text)
	return byte_string.decode(charset)

with con:
	for message in mailbox.mbox('/var/mail/mailbot'):
		attachs = list()
		headers = Parser(policy=default).parsestr(str(message))
		if message.is_multipart():
			# get body mail
			for body in message.get_payload():
				if body.get_content_type() == 'text/plain':
					bodytext = quopri.decodestring(str(message.get_payload()[1].get_payload())).decode('utf8')
				if body.get_content_type() == 'multipart/alternative':
					for body2 in body.get_payload():
						bodytext = quopri.decodestring(str(body2.get_payload())).decode('utf8')
					
		for part in message.walk():
			if part.get_content_maintype() == 'multipart': continue
			# if part.get_content_maintype() == 'text': continue
			if part.get('Content-Disposition') == 'inline': continue
			if part.get('Content-Disposition') is None: continue

			filename = part.get_filename()
			if filename != None:
				if not encoded_words_to_text(filename):
					filename=filename
				else :
					filename=encoded_words_to_text(filename)
			filename=str(uuid.uuid1())+'|'+filename
			open('/home/admin/web/***/public_html/attach/'+filename, 'wb').write(part.get_payload(decode=True))
			attachs.append(filename)

		cur = con.cursor()
		cur.execute("select id from mails where mailid = %s",(headers['Message-Id'],))
		mail = rows = cur.fetchone()
		if(mail == None):
			timestamp = int(time.time())
			cur.execute('INSERT INTO mails(`mailid`, `subject`, `to`, `from`, `body`, `time_receive`) VALUES(%s, %s, %s, %s, %s, %s)',(headers['Message-ID'], headers['subject'].encode('utf-8'), headers['to'], headers['from'], bodytext, timestamp))
			mailid = cur.lastrowid
		for at in attachs:
				cur.execute('INSERT INTO attach(`mailid`, `name`) VALUES(%s, %s)', (mailid, at))

# -*- coding: utf-8 -*-

from email.parser import BytesParser, Parser
from email.policy import default
import mailbox
import MySQLdb as mdb
import quopri
import string
import uuid
import base64
import re
import time

con = mdb.connect('localhost', '', '', '', use_unicode=True, charset="utf8");

def encoded_words_to_text(encoded_words):
	encoded_word_regex = r'=\?{1}(.+)\?{1}([B|Q])\?{1}(.+)\?{1}='
	try:
		charset, encoding, encoded_text = re.match(encoded_word_regex, encoded_words).groups()
	except Exception as e:
		return None
	if encoding is 'B':
		byte_string = base64.b64decode(encoded_text)
	elif encoding is 'Q':
		byte_string = quopri.decodestring(encoded_text)
	return byte_string.decode(charset)

with con:
	for message in mailbox.mbox('/var/mail/mailbot'):
		attachs = list()
		headers = Parser(policy=default).parsestr(str(message))
		if message.is_multipart():
			# get body mail
			for body in message.get_payload():
				if body.get_content_type() == 'text/plain':
					bodytext = quopri.decodestring(str(message.get_payload()[1].get_payload())).decode('utf8')
				if body.get_content_type() == 'multipart/alternative':
					for body2 in body.get_payload():
						bodytext = quopri.decodestring(str(body2.get_payload())).decode('utf8')
					
		for part in message.walk():
			if part.get_content_maintype() == 'multipart': continue
			# if part.get_content_maintype() == 'text': continue
			if part.get('Content-Disposition') == 'inline': continue
			if part.get('Content-Disposition') is None: continue

			filename = part.get_filename()
			if filename != None:
				if not encoded_words_to_text(filename):
					filename=filename
				else :
					filename=encoded_words_to_text(filename)
			filename=str(uuid.uuid1())+'|'+filename
			open('/home/admin/web/***/public_html/attach/'+filename, 'wb').write(part.get_payload(decode=True))
			attachs.append(filename)

		cur = con.cursor()
		cur.execute("select id from mails where mailid = %s",(headers['Message-Id'],))
		mail = rows = cur.fetchone()
		if(mail == None):
			timestamp = int(time.time())
			cur.execute('INSERT INTO mails(`mailid`, `subject`, `to`, `from`, `body`, `time_receive`) VALUES(%s, %s, %s, %s, %s, %s)',(headers['Message-ID'], headers['subject'].encode('utf-8'), headers['to'], headers['from'], bodytext, timestamp))
			mailid = cur.lastrowid
			cur.execute('SELECT `count_receive` from settings')
			counter = cur.fetchone()
			for at in attachs:
				cur.execute('INSERT INTO attach(`mailid`, `name`) VALUES(%s, %s)', (mailid, at))