diff --git a/imbox/imap.py b/imbox/imap.py index 568195b..627b8b5 100644 --- a/imbox/imap.py +++ b/imbox/imap.py @@ -2,29 +2,27 @@ from imaplib import IMAP4, IMAP4_SSL class ImapTransport(object): - - def __init__(self, hostname, port=None, ssl=False): - self.hostname = hostname - self.port = port - - if ssl: - self.transport = IMAP4_SSL - if not self.port: - self.port = 993 - else: - self.transport = IMAP4 - if not self.port: - self.port = 143 - self.server = self.transport(self.hostname, self.port) + def __init__(self, hostname, port=None, ssl=False): + self.hostname = hostname + self.port = port - def list_folders(self): - return self.server.list() + if ssl: + self.transport = IMAP4_SSL + if not self.port: + self.port = 993 + else: + self.transport = IMAP4 + if not self.port: + self.port = 143 - def connect(self, username, password): - self.server.login(username, password) - self.server.select() + self.server = self.transport(self.hostname, self.port) - return self.server - + def list_folders(self): + return self.server.list() + def connect(self, username, password): + self.server.login(username, password) + self.server.select() + + return self.server