def connectToServers(self, profile):
recv = profile['config']['receiver']['protocol']
send = profile['config']['sender']['protocol']
if recv == "POP3":
self.mailbox = MailboxPOP3(profile)
elif recv == "IMAP":
self.mailbox = MailboxIMAP(profile)
if send == "SMTP":
self.sendbox = SendboxSMTP(profile)
def connectToServers (self, profile):
recv = profile['config']['receiver']['protocol']
send = profile['config']['sender']['protocol']
if recv == "POP3":
self.mailbox = MailboxPOP3(profile)
elif recv == "IMAP":
self.mailbox = MailboxIMAP(profile)
if send == "SMTP":
self.sendbox = SendboxSMTP(profile)
class MainWindow(QMainWindow):
def __init__(self):
QMainWindow.__init__(self)
dirname = os.path.dirname(os.path.abspath(__file__))
uic.loadUi(os.path.join(dirname, "mainwindow.ui"), self)
self.profiles = []
self.action_New.triggered.connect(self.newProfileDialog)
self.action_Edit.triggered.connect(self.editProfile)
self.action_Exit.triggered.connect(self.close)
self.action_About.triggered.connect(self.aboutDialog)
self.newMailButton.clicked.connect(self.newMailDialog)
self.answerButton.clicked.connect(self.answerMail)
self.syncButton.clicked.connect(self.syncMailbox)
self.deleteButton.clicked.connect(self.deleteMail)
self.webView.page().setLinkDelegationPolicy(QWebPage.DelegateAllLinks)
self.webView.linkClicked.connect(self.openLinkInBrowser)
self.mail_Subject.setTextInteractionFlags(Qt.TextSelectableByMouse)
self.mail_From.setTextInteractionFlags(Qt.TextSelectableByMouse)
self.profileDir = 'profile'
self.currentProfile = None
self.currentProfileAction = None
self.mailbox = None
self.sendbox = None
self.currentMailNo = None
self.updateAllProfiles()
def updateAllProfiles(self):
files = [
os.path.join(self.profileDir, path)
for path in os.listdir(self.profileDir)
if os.path.isfile(os.path.join(self.profileDir, path))
and path.endswith('.yaml')
]
for path in files:
profile = None
with open(path, 'r') as stream:
profile = yaml.load(stream)
if profile:
self.appendProfile(profile, os.path.basename(path))
def newProfileDialog(self):
dialog = Dialog_NewProfile(self)
dialog.exec()
def aboutDialog(self):
dialog = Dialog_About(self)
dialog.exec()
def newMailDialog(self, **kwargs):
if self.sendbox:
dialog = Dialog_NewMail(self, **kwargs)
dialog.exec()
else:
QMessageBox(
QMessageBox.Critical, "Error",
"You have update the Sender configuration to be able to send mails."
).exec_()
def appendProfile(self, profile, filename):
self.action_No_profile_known.setVisible(False)
action = QtGui.QAction(self)
action.setText(profile['name'])
action.setObjectName('action_' + filename)
action.triggered.connect(partial(self.selectProfile, action, profile))
self.profiles.append(profile)
self.menuProfile.addAction(action)
return action
def getProfileID(self):
files = os.listdir(self.profileDir)
profile_id = 1
used_ids = []
for name in files:
try:
used_ids.append(int(name.split('-')[0]))
except Exception:
pass
while profile_id in used_ids:
profile_id += 1
return profile_id
def getProfileFile(self, profile_id, name):
safe_name = re.sub(r"[^A-Za-z0-9-]", "-", name)
return os.path.join(self.profileDir,
"%03d-%s.yaml" % (profile_id, safe_name))
def registerProfile(self, name, config, edit_id=None):
"""
config.receiver.host: string
config.receiver.port: integer
config.receiver.protocol: "POP3" | "IMAP"
config.receiver.username: string
config.receiver.password: string
config.receiver.passRemember: boolean
config.receiver.ssl: boolean
config.sender.host: string
config.sender.port: integer
config.sender.protocol: "SMTP"
config.sender.auth: boolean
config.sender.username: string
config.sender.password: string
config.sender.passRemember: boolean
config.sender.ssl: boolean
config.sender.tls: boolean
config.email: string
"""
profile_id = self.getProfileID() if not edit_id else edit_id
filename = self.getProfileFile(profile_id, name)
profile = {'id': profile_id, 'name': name, 'config': config}
with open(filename, 'w') as stream:
stream.write(yaml.dump(profile))
action = self.appendProfile(profile, filename)
self.selectProfile(action, profile)
def editProfile(self):
dialog = Dialog_NewProfile(self)
dialog.load(self.currentProfile)
dialog.exec()
def connectToServers(self, profile):
recv = profile['config']['receiver']['protocol']
send = profile['config']['sender']['protocol']
if recv == "POP3":
self.mailbox = MailboxPOP3(profile)
elif recv == "IMAP":
self.mailbox = MailboxIMAP(profile)
if send == "SMTP":
self.sendbox = SendboxSMTP(profile)
def selectProfile(self, action, profile):
if not self.currentProfile or profile['id'] != self.currentProfile[
'id']:
self.logout()
try:
self.connectToServers(profile)
self.mailbox.sync()
self.refreshMailList()
except Exception:
QMessageBox(
QMessageBox.Critical, "Error",
"Failure while trying to join the servers.").exec_()
if self.currentProfileAction:
font = QtGui.QFont()
font.setBold(False)
self.currentProfileAction.setFont(font)
self.currentProfileAction = action
self.action_Edit.setEnabled(True)
font = QtGui.QFont()
font.setBold(True)
self.currentProfileAction.setFont(font)
self.currentProfile = profile
def showMail(self, item):
try:
mailno = item.data(Qt.UserRole)
from_, subject, date, content = self.mailbox.get(mailno)
self.mail_From.setText(from_)
except Exception:
QMessageBox(
QMessageBox.Critical, "Error",
"Failure while trying to retriview the email.").exec_()
try:
# converting date to local-date
_date = date
date_tuple = email.utils.parsedate_tz(date)
if date_tuple:
local_date = datetime.datetime.fromtimestamp(
email.utils.mktime_tz(date_tuple))
_date = local_date.strftime("%a, %d %b %Y %H:%M:%S")
except Exception:
pass
else:
date = _date
try:
# cutting the subject to force word wrapping
_subject = subject
for i in range(int(len(_subject) / 100)):
_subject = _subject[:i * 100] + ' ' + _subject[i * 100:]
except Exception:
pass
else:
subject = _subject
try:
self.mail_Subject.setText(subject)
except Exception:
pass
try:
self.mail_Date.setText(date)
except Exception:
pass
try:
self.webView.setHtml(content)
except Exception:
pass
self.currentMailNo = mailno
def refreshMailList(self):
while self.mailList.count() > 0:
self.mailList.takeItem(0)
for message in reversed(self.mailbox.list()):
subject = self.mailbox.fdecode(message['header']['Subject'])
subject = subject if isinstance(subject, str) else '<No subject>'
item = QListWidgetItem(subject)
item.setData(Qt.UserRole, message['id'])
self.mailList.addItem(item)
self.mailList.itemClicked.connect(self.showMail)
def syncMailbox(self):
try:
self.mailbox.sync()
self.refreshMailList()
except Exception:
QMessageBox(QMessageBox.Critical, "Error",
"Failure while trying to join the servers.").exec_()
def openLinkInBrowser(self, url):
webbrowser.open(url.toString())
def deleteMail(self):
if self.currentMailNo != None: # currentMailNo can be 0
self.mailbox.delete(self.currentMailNo)
if self.mailList.currentRow() + 1 <= self.mailList.count():
next_row = self.mailList.currentRow()
elif self.mailList.currentRow() - 1 <= self.mailList.count():
next_row = self.mailList.currentRow() - 1
else:
next_row = 0
self.mailList.takeItem(self.mailList.currentRow())
self.mailList.setCurrentRow(next_row)
self.showMail(self.mailList.currentItem())
else:
QMessageBox(QMessageBox.Critical, "Error",
"You have to select an email before.").exec_()
def answerMail(self):
if self.currentMailNo != None:
_from, subject, date, content = self.mailbox.get(
self.currentMailNo)
dialog = self.newMailDialog(to=_from, subject='RE: ' + subject)
else:
QMessageBox(QMessageBox.Critical, "Error",
"You have to select an email before.").exec_()
def logout(self):
self.mail_From.setText('')
self.mail_Subject.setText('')
self.mail_Date.setText('')
self.webView.setHtml('')
self.currentMailNo = None
if self.mailbox:
self.mailbox.logout()
self.mailbox = None
if self.sendbox:
self.sendbox.logout()
self.sendbox = None
def closeEvent(self, event):
if self.mailbox:
self.logout()
event.accept()
class MainWindow (QMainWindow):
def __init__ (self):
QMainWindow.__init__(self)
dirname = os.path.dirname(os.path.abspath(__file__))
uic.loadUi(os.path.join(dirname, "mainwindow.ui"), self)
self.profiles = []
self.action_New.triggered.connect(self.newProfileDialog)
self.action_Edit.triggered.connect(self.editProfile)
self.action_Exit.triggered.connect(self.close)
self.action_About.triggered.connect(self.aboutDialog)
self.newMailButton.clicked.connect(self.newMailDialog)
self.answerButton.clicked.connect(self.answerMail)
self.syncButton.clicked.connect(self.syncMailbox)
self.deleteButton.clicked.connect(self.deleteMail)
self.webView.page().setLinkDelegationPolicy(QWebPage.DelegateAllLinks)
self.webView.linkClicked.connect(self.openLinkInBrowser)
self.mail_Subject.setTextInteractionFlags(Qt.TextSelectableByMouse)
self.mail_From.setTextInteractionFlags(Qt.TextSelectableByMouse)
self.profileDir = 'profile'
self.currentProfile = None
self.currentProfileAction = None
self.mailbox = None
self.sendbox = None
self.currentMailNo = None
self.updateAllProfiles()
def updateAllProfiles (self):
files = [os.path.join(self.profileDir, path)
for path
in os.listdir(self.profileDir)
if os.path.isfile(os.path.join(self.profileDir, path)) and path.endswith('.yaml')]
for path in files:
profile = None
with open(path, 'r') as stream:
profile = yaml.load(stream)
if profile:
self.appendProfile(profile, os.path.basename(path))
def newProfileDialog (self):
dialog = Dialog_NewProfile(self)
dialog.exec()
def aboutDialog (self):
dialog = Dialog_About(self)
dialog.exec()
def newMailDialog (self, **kwargs):
if self.sendbox:
dialog = Dialog_NewMail(self, **kwargs)
dialog.exec()
else:
QMessageBox(QMessageBox.Critical, "Error", "You have update the Sender configuration to be able to send mails.").exec_()
def appendProfile (self, profile, filename):
self.action_No_profile_known.setVisible(False)
action = QtGui.QAction(self)
action.setText(profile['name'])
action.setObjectName('action_' + filename)
action.triggered.connect(partial(self.selectProfile, action, profile))
self.profiles.append(profile)
self.menuProfile.addAction(action)
return action
def getProfileID (self):
files = os.listdir(self.profileDir)
profile_id = 1
used_ids = []
for name in files:
try:
used_ids.append(int(name.split('-')[0]))
except Exception:
pass
while profile_id in used_ids:
profile_id += 1
return profile_id
def getProfileFile (self, profile_id, name):
safe_name = re.sub(r"[^A-Za-z0-9-]", "-", name)
return os.path.join(
self.profileDir,
"%03d-%s.yaml" % (profile_id, safe_name))
def registerProfile (self, name, config, edit_id = None):
"""
config.receiver.host: string
config.receiver.port: integer
config.receiver.protocol: "POP3" | "IMAP"
config.receiver.username: string
config.receiver.password: string
config.receiver.passRemember: boolean
config.receiver.ssl: boolean
config.sender.host: string
config.sender.port: integer
config.sender.protocol: "SMTP"
config.sender.auth: boolean
config.sender.username: string
config.sender.password: string
config.sender.passRemember: boolean
config.sender.ssl: boolean
config.sender.tls: boolean
config.email: string
"""
profile_id = self.getProfileID() if not edit_id else edit_id
filename = self.getProfileFile(profile_id, name)
profile = {
'id': profile_id,
'name': name,
'config': config
}
with open(filename, 'w') as stream:
stream.write(yaml.dump(profile))
action = self.appendProfile(profile, filename)
self.selectProfile(action, profile)
def editProfile (self):
dialog = Dialog_NewProfile(self)
dialog.load(self.currentProfile)
dialog.exec()
def connectToServers (self, profile):
recv = profile['config']['receiver']['protocol']
send = profile['config']['sender']['protocol']
if recv == "POP3":
self.mailbox = MailboxPOP3(profile)
elif recv == "IMAP":
self.mailbox = MailboxIMAP(profile)
if send == "SMTP":
self.sendbox = SendboxSMTP(profile)
def selectProfile (self, action, profile):
if not self.currentProfile or profile['id'] != self.currentProfile['id']:
self.logout()
try:
self.connectToServers(profile)
self.mailbox.sync()
self.refreshMailList()
except Exception:
QMessageBox(QMessageBox.Critical, "Error", "Failure while trying to join the servers.").exec_()
if self.currentProfileAction:
font = QtGui.QFont()
font.setBold(False)
self.currentProfileAction.setFont(font)
self.currentProfileAction = action
self.action_Edit.setEnabled(True)
font = QtGui.QFont()
font.setBold(True)
self.currentProfileAction.setFont(font)
self.currentProfile = profile
def showMail (self, item):
try:
mailno = item.data(Qt.UserRole)
from_, subject, date, content = self.mailbox.get(mailno)
self.mail_From.setText(from_)
except Exception:
QMessageBox(QMessageBox.Critical, "Error", "Failure while trying to retriview the email.").exec_()
try:
# converting date to local-date
_date = date
date_tuple = email.utils.parsedate_tz(date)
if date_tuple:
local_date = datetime.datetime.fromtimestamp(email.utils.mktime_tz(date_tuple))
_date = local_date.strftime("%a, %d %b %Y %H:%M:%S")
except Exception: pass
else: date = _date
try:
# cutting the subject to force word wrapping
_subject = subject
for i in range(int(len(_subject) / 100)):
_subject = _subject[:i*100] + ' ' + _subject[i*100:]
except Exception: pass
else: subject = _subject
try: self.mail_Subject.setText(subject)
except Exception: pass
try: self.mail_Date.setText(date)
except Exception: pass
try: self.webView.setHtml(content)
except Exception: pass
self.currentMailNo = mailno
def refreshMailList (self):
while self.mailList.count() > 0:
self.mailList.takeItem(0)
for message in reversed(self.mailbox.list()):
subject = self.mailbox.fdecode(message['header']['Subject'])
subject = subject if isinstance(subject, str) else '<No subject>'
item = QListWidgetItem(subject)
item.setData(Qt.UserRole, message['id']);
self.mailList.addItem(item)
self.mailList.itemClicked.connect(self.showMail)
def syncMailbox (self):
try:
self.mailbox.sync()
self.refreshMailList()
except Exception:
QMessageBox(QMessageBox.Critical, "Error", "Failure while trying to join the servers.").exec_()
def openLinkInBrowser(self, url):
webbrowser.open(url.toString())
def deleteMail (self):
if self.currentMailNo != None: # currentMailNo can be 0
self.mailbox.delete(self.currentMailNo)
if self.mailList.currentRow() + 1 <= self.mailList.count():
next_row = self.mailList.currentRow()
elif self.mailList.currentRow() - 1 <= self.mailList.count():
next_row = self.mailList.currentRow() - 1
else:
next_row = 0
self.mailList.takeItem(self.mailList.currentRow())
self.mailList.setCurrentRow(next_row)
self.showMail(self.mailList.currentItem())
else:
QMessageBox(QMessageBox.Critical, "Error", "You have to select an email before.").exec_()
def answerMail (self):
if self.currentMailNo != None:
_from, subject, date, content = self.mailbox.get(self.currentMailNo)
dialog = self.newMailDialog(to=_from, subject='RE: ' + subject)
else:
QMessageBox(QMessageBox.Critical, "Error", "You have to select an email before.").exec_()
def logout (self):
self.mail_From.setText('')
self.mail_Subject.setText('')
self.mail_Date.setText('')
self.webView.setHtml('')
self.currentMailNo = None
if self.mailbox:
self.mailbox.logout()
self.mailbox = None
if self.sendbox:
self.sendbox.logout()
self.sendbox = None
def closeEvent (self, event):
if self.mailbox:
self.logout()
event.accept()