Restructured folder/IMAP code
In preparation for GMail label sync, we had split our some functionality that will be needed further into their own functions. This also permitted the code to look more compact and concise. Signed-off-by: Eygene Ryabinkin <rea@codelabs.ru>
This commit is contained in:
parent
5391476dfb
commit
9319ae212b
@ -66,6 +66,10 @@ class BaseFolder(object):
|
|||||||
self.ui.debug('', "Filtering out '%s'[%s] due to folderfilter" \
|
self.ui.debug('', "Filtering out '%s'[%s] due to folderfilter" \
|
||||||
% (self.ffilter_name, repository))
|
% (self.ffilter_name, repository))
|
||||||
|
|
||||||
|
# Passes for syncmessagesto
|
||||||
|
self.syncmessagesto_passes = [('copying messages' , self.__syncmessagesto_copy),
|
||||||
|
('deleting messages' , self.__syncmessagesto_delete),
|
||||||
|
('syncing flags' , self.__syncmessagesto_flags)]
|
||||||
|
|
||||||
def getname(self):
|
def getname(self):
|
||||||
"""Returns name"""
|
"""Returns name"""
|
||||||
@ -319,6 +323,103 @@ class BaseFolder(object):
|
|||||||
for uid in uidlist:
|
for uid in uidlist:
|
||||||
self.deletemessageflags(uid, flags)
|
self.deletemessageflags(uid, flags)
|
||||||
|
|
||||||
|
|
||||||
|
def addmessageheader(self, content, headername, headervalue):
|
||||||
|
self.ui.debug('',
|
||||||
|
'addmessageheader: called to add %s: %s' % (headername,
|
||||||
|
headervalue))
|
||||||
|
insertionpoint = content.find('\n\n')
|
||||||
|
self.ui.debug('', 'addmessageheader: insertionpoint = %d' % insertionpoint)
|
||||||
|
leader = content[0:insertionpoint]
|
||||||
|
self.ui.debug('', 'addmessageheader: leader = %s' % repr(leader))
|
||||||
|
if insertionpoint == 0 or insertionpoint == -1:
|
||||||
|
newline = ''
|
||||||
|
insertionpoint = 0
|
||||||
|
else:
|
||||||
|
newline = '\n'
|
||||||
|
newline += "%s: %s" % (headername, headervalue)
|
||||||
|
self.ui.debug('', 'addmessageheader: newline = ' + repr(newline))
|
||||||
|
trailer = content[insertionpoint:]
|
||||||
|
self.ui.debug('', 'addmessageheader: trailer = ' + repr(trailer))
|
||||||
|
return leader + newline + trailer
|
||||||
|
|
||||||
|
|
||||||
|
def __find_eoh(self, content):
|
||||||
|
"""
|
||||||
|
Searches for the point where mail headers end.
|
||||||
|
Either double '\n', or end of string.
|
||||||
|
|
||||||
|
Arguments:
|
||||||
|
- content: contents of the message to search in
|
||||||
|
Returns: position of the first non-header byte.
|
||||||
|
|
||||||
|
"""
|
||||||
|
eoh_cr = content.find('\n\n')
|
||||||
|
if eoh_cr == -1:
|
||||||
|
eoh_cr = len(content)
|
||||||
|
|
||||||
|
return eoh_cr
|
||||||
|
|
||||||
|
|
||||||
|
def getmessageheader(self, content, name):
|
||||||
|
"""
|
||||||
|
Searches for the given header and returns its value.
|
||||||
|
Arguments:
|
||||||
|
- contents: message itself
|
||||||
|
- name: name of the header to be searched
|
||||||
|
|
||||||
|
Returns: header value or None if no such header was found
|
||||||
|
|
||||||
|
"""
|
||||||
|
self.ui.debug('', 'getmessageheader: called to get %s' % name)
|
||||||
|
eoh = self.__find_eoh(content)
|
||||||
|
self.ui.debug('', 'getmessageheader: eoh = %d' % eoh)
|
||||||
|
headers = content[0:eoh]
|
||||||
|
self.ui.debug('', 'getmessageheader: headers = %s' % repr(headers))
|
||||||
|
|
||||||
|
m = re.search('^%s:(.*)$' % name, headers, flags = re.MULTILINE)
|
||||||
|
if m:
|
||||||
|
return m.group(1).strip()
|
||||||
|
else:
|
||||||
|
return None
|
||||||
|
|
||||||
|
|
||||||
|
def deletemessageheaders(self, content, header_list):
|
||||||
|
"""
|
||||||
|
Deletes headers in the given list from the message content.
|
||||||
|
|
||||||
|
Arguments:
|
||||||
|
- content: message itself
|
||||||
|
- header_list: list of headers to be deleted or just the header name
|
||||||
|
|
||||||
|
We expect our message to have '\n' as line endings.
|
||||||
|
|
||||||
|
"""
|
||||||
|
if type(header_list) != type([]):
|
||||||
|
header_list = [header_list]
|
||||||
|
self.ui.debug('', 'deletemessageheaders: called to delete %s' % (header_list))
|
||||||
|
|
||||||
|
if not len(header_list): return content
|
||||||
|
|
||||||
|
eoh = self.__find_eoh(content)
|
||||||
|
self.ui.debug('', 'deletemessageheaders: end of headers = %d' % eoh)
|
||||||
|
headers = content[0:eoh]
|
||||||
|
rest = content[eoh:]
|
||||||
|
self.ui.debug('', 'deletemessageheaders: headers = %s' % repr(headers))
|
||||||
|
new_headers = []
|
||||||
|
for h in headers.split('\n'):
|
||||||
|
keep_it = True
|
||||||
|
for trim_h in self.filterheaders:
|
||||||
|
if len(h) > len(trim_h) and h[0:len(trim_h)+1] == (trim_h + ":"):
|
||||||
|
keep_it = False
|
||||||
|
break
|
||||||
|
if keep_it: new_headers.append(h)
|
||||||
|
|
||||||
|
return ('\n'.join(new_headers) + rest)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
def change_message_uid(self, uid, new_uid):
|
def change_message_uid(self, uid, new_uid):
|
||||||
"""Change the message from existing uid to new_uid
|
"""Change the message from existing uid to new_uid
|
||||||
|
|
||||||
@ -564,14 +665,16 @@ class BaseFolder(object):
|
|||||||
deleted there), sync the flag change to both dstfolder and
|
deleted there), sync the flag change to both dstfolder and
|
||||||
statusfolder.
|
statusfolder.
|
||||||
|
|
||||||
|
Pass4: Synchronize label changes (Gmail only)
|
||||||
|
Compares label mismatches in self with those in statusfolder.
|
||||||
|
If msg has a valid UID and exists on dstfolder, syncs the labels
|
||||||
|
to both dstfolder and statusfolder.
|
||||||
|
|
||||||
:param dstfolder: Folderinstance to sync the msgs to.
|
:param dstfolder: Folderinstance to sync the msgs to.
|
||||||
:param statusfolder: LocalStatus instance to sync against.
|
:param statusfolder: LocalStatus instance to sync against.
|
||||||
"""
|
|
||||||
passes = [('copying messages' , self.__syncmessagesto_copy),
|
|
||||||
('deleting messages' , self.__syncmessagesto_delete),
|
|
||||||
('syncing flags' , self.__syncmessagesto_flags)]
|
|
||||||
|
|
||||||
for (passdesc, action) in passes:
|
"""
|
||||||
|
for (passdesc, action) in self.syncmessagesto_passes:
|
||||||
# bail out on CTRL-C or SIGTERM
|
# bail out on CTRL-C or SIGTERM
|
||||||
if offlineimap.accounts.Account.abort_NOW_signal.is_set():
|
if offlineimap.accounts.Account.abort_NOW_signal.is_set():
|
||||||
break
|
break
|
||||||
|
@ -30,6 +30,13 @@ from offlineimap.imaplib2 import MonthNames
|
|||||||
CRLF = '\r\n'
|
CRLF = '\r\n'
|
||||||
|
|
||||||
|
|
||||||
|
# NB: message returned from getmessage() will have '\n' all over the place,
|
||||||
|
# NB: there will be no CRLFs. Just before the sending stage of savemessage()
|
||||||
|
# NB: '\n' will be transformed back to CRLF. So, for the most parts of the
|
||||||
|
# NB: code the stored content will be clean of CRLF and one can rely that
|
||||||
|
# NB: line endings will be pure '\n'.
|
||||||
|
|
||||||
|
|
||||||
class IMAPFolder(BaseFolder):
|
class IMAPFolder(BaseFolder):
|
||||||
def __init__(self, imapserver, name, repository):
|
def __init__(self, imapserver, name, repository):
|
||||||
name = imaputil.dequote(name)
|
name = imaputil.dequote(name)
|
||||||
@ -41,6 +48,7 @@ class IMAPFolder(BaseFolder):
|
|||||||
self.messagelist = None
|
self.messagelist = None
|
||||||
self.randomgenerator = random.Random()
|
self.randomgenerator = random.Random()
|
||||||
#self.ui is set in BaseFolder
|
#self.ui is set in BaseFolder
|
||||||
|
self.imap_query = ['BODY.PEEK[]']
|
||||||
|
|
||||||
fh_conf = self.repository.account.getconf('filterheaders', '')
|
fh_conf = self.repository.account.getconf('filterheaders', '')
|
||||||
self.filterheaders = [h for h in re.split(r'\s*,\s*', fh_conf) if h]
|
self.filterheaders = [h for h in re.split(r'\s*,\s*', fh_conf) if h]
|
||||||
@ -131,23 +139,32 @@ class IMAPFolder(BaseFolder):
|
|||||||
return True
|
return True
|
||||||
return False
|
return False
|
||||||
|
|
||||||
# Interface from BaseFolder
|
|
||||||
def cachemessagelist(self):
|
def _msgs_to_fetch(self, imapobj):
|
||||||
|
"""
|
||||||
|
Determines UIDS of messages to be fetched
|
||||||
|
|
||||||
|
Arguments:
|
||||||
|
- imapobj: instance of IMAPlib
|
||||||
|
|
||||||
|
Returns: UID ranges for messages or None if no messages
|
||||||
|
are to be fetched.
|
||||||
|
|
||||||
|
"""
|
||||||
|
res_type, imapdata = imapobj.select(self.getfullname(), True, True)
|
||||||
|
if imapdata == [None] or imapdata[0] == '0':
|
||||||
|
# Empty folder, no need to populate message list
|
||||||
|
return None
|
||||||
|
|
||||||
|
# By default examine all UIDs in this folder
|
||||||
|
msgsToFetch = '1:*'
|
||||||
|
|
||||||
maxage = self.config.getdefaultint("Account %s" % self.accountname,
|
maxage = self.config.getdefaultint("Account %s" % self.accountname,
|
||||||
"maxage", -1)
|
"maxage", -1)
|
||||||
maxsize = self.config.getdefaultint("Account %s" % self.accountname,
|
maxsize = self.config.getdefaultint("Account %s" % self.accountname,
|
||||||
"maxsize", -1)
|
"maxsize", -1)
|
||||||
self.messagelist = {}
|
|
||||||
|
|
||||||
imapobj = self.imapserver.acquireconnection()
|
|
||||||
try:
|
|
||||||
res_type, imapdata = imapobj.select(self.getfullname(), True, True)
|
|
||||||
if imapdata == [None] or imapdata[0] == '0':
|
|
||||||
# Empty folder, no need to populate message list
|
|
||||||
return
|
|
||||||
# By default examine all UIDs in this folder
|
|
||||||
msgsToFetch = '1:*'
|
|
||||||
|
|
||||||
|
# Build search condition
|
||||||
if (maxage != -1) | (maxsize != -1):
|
if (maxage != -1) | (maxsize != -1):
|
||||||
search_cond = "(";
|
search_cond = "(";
|
||||||
|
|
||||||
@ -174,12 +191,22 @@ class IMAPFolder(BaseFolder):
|
|||||||
if res_type != 'OK':
|
if res_type != 'OK':
|
||||||
raise OfflineImapError("SEARCH in folder [%s]%s failed. "
|
raise OfflineImapError("SEARCH in folder [%s]%s failed. "
|
||||||
"Search string was '%s'. Server responded '[%s] %s'" % (
|
"Search string was '%s'. Server responded '[%s] %s'" % (
|
||||||
self.getrepository(), self,
|
self.getrepository(), self, search_cond, res_type, res_data),
|
||||||
search_cond, res_type, res_data),
|
|
||||||
OfflineImapError.ERROR.FOLDER)
|
OfflineImapError.ERROR.FOLDER)
|
||||||
|
|
||||||
# Result UIDs are seperated by space, coalesce into ranges
|
# Result UIDs are seperated by space, coalesce into ranges
|
||||||
msgsToFetch = imaputil.uid_sequence(res_data[0].split())
|
msgsToFetch = imaputil.uid_sequence(res_data[0].split())
|
||||||
|
|
||||||
|
return msgsToFetch
|
||||||
|
|
||||||
|
|
||||||
|
# Interface from BaseFolder
|
||||||
|
def cachemessagelist(self):
|
||||||
|
self.messagelist = {}
|
||||||
|
|
||||||
|
imapobj = self.imapserver.acquireconnection()
|
||||||
|
try:
|
||||||
|
msgsToFetch = self._msgs_to_fetch(imapobj)
|
||||||
if not msgsToFetch:
|
if not msgsToFetch:
|
||||||
return # No messages to sync
|
return # No messages to sync
|
||||||
|
|
||||||
@ -213,60 +240,42 @@ class IMAPFolder(BaseFolder):
|
|||||||
rtime = imaplibutil.Internaldate2epoch(messagestr)
|
rtime = imaplibutil.Internaldate2epoch(messagestr)
|
||||||
self.messagelist[uid] = {'uid': uid, 'flags': flags, 'time': rtime}
|
self.messagelist[uid] = {'uid': uid, 'flags': flags, 'time': rtime}
|
||||||
|
|
||||||
|
|
||||||
# Interface from BaseFolder
|
# Interface from BaseFolder
|
||||||
def getmessagelist(self):
|
def getmessagelist(self):
|
||||||
return self.messagelist
|
return self.messagelist
|
||||||
|
|
||||||
# Interface from BaseFolder
|
# Interface from BaseFolder
|
||||||
def getmessage(self, uid):
|
def getmessage(self, uid):
|
||||||
"""Retrieve message with UID from the IMAP server (incl body)
|
"""
|
||||||
|
Retrieve message with UID from the IMAP server (incl body)
|
||||||
|
|
||||||
|
After this function all CRLFs will be transformed to '\n'.
|
||||||
|
|
||||||
:returns: the message body or throws and OfflineImapError
|
:returns: the message body or throws and OfflineImapError
|
||||||
(probably severity MESSAGE) if e.g. no message with
|
(probably severity MESSAGE) if e.g. no message with
|
||||||
this UID could be found.
|
this UID could be found.
|
||||||
|
|
||||||
"""
|
"""
|
||||||
imapobj = self.imapserver.acquireconnection()
|
imapobj = self.imapserver.acquireconnection()
|
||||||
try:
|
try:
|
||||||
fails_left = 2 # retry on dropped connection
|
data = self._fetch_from_imap(imapobj, str(uid), 2)
|
||||||
while fails_left:
|
finally:
|
||||||
try:
|
self.imapserver.releaseconnection(imapobj)
|
||||||
imapobj.select(self.getfullname(), readonly = True)
|
|
||||||
res_type, data = imapobj.uid('fetch', str(uid),
|
|
||||||
'(BODY.PEEK[])')
|
|
||||||
fails_left = 0
|
|
||||||
except imapobj.abort as e:
|
|
||||||
# Release dropped connection, and get a new one
|
|
||||||
self.imapserver.releaseconnection(imapobj, True)
|
|
||||||
imapobj = self.imapserver.acquireconnection()
|
|
||||||
self.ui.error(e, exc_info()[2])
|
|
||||||
fails_left -= 1
|
|
||||||
if not fails_left:
|
|
||||||
raise e
|
|
||||||
if data == [None] or res_type != 'OK':
|
|
||||||
#IMAP server says bad request or UID does not exist
|
|
||||||
severity = OfflineImapError.ERROR.MESSAGE
|
|
||||||
reason = "IMAP server '%s' failed to fetch message UID '%d'."\
|
|
||||||
"Server responded: %s %s" % (self.getrepository(), uid,
|
|
||||||
res_type, data)
|
|
||||||
if data == [None]:
|
|
||||||
#IMAP server did not find a message with this UID
|
|
||||||
reason = "IMAP server '%s' does not have a message "\
|
|
||||||
"with UID '%s'" % (self.getrepository(), uid)
|
|
||||||
raise OfflineImapError(reason, severity)
|
|
||||||
# data looks now e.g. [('320 (UID 17061 BODY[]
|
# data looks now e.g. [('320 (UID 17061 BODY[]
|
||||||
# {2565}','msgbody....')] we only asked for one message,
|
# {2565}','msgbody....')] we only asked for one message,
|
||||||
# and that msg is in data[0]. msbody is in [0][1]
|
# and that msg is in data[0]. msbody is in [0][1]
|
||||||
data = data[0][1].replace(CRLF, "\n")
|
data = data[0][1].replace(CRLF, "\n")
|
||||||
|
|
||||||
if len(data)>200:
|
if len(data)>200:
|
||||||
dbg_output = "%s...%s" % (str(data)[:150],
|
dbg_output = "%s...%s" % (str(data)[:150], str(data)[-50:])
|
||||||
str(data)[-50:])
|
|
||||||
else:
|
else:
|
||||||
dbg_output = data
|
dbg_output = data
|
||||||
|
|
||||||
self.ui.debug('imap', "Returned object from fetching %d: '%s'" %
|
self.ui.debug('imap', "Returned object from fetching %d: '%s'" %
|
||||||
(uid, dbg_output))
|
(uid, dbg_output))
|
||||||
finally:
|
|
||||||
self.imapserver.releaseconnection(imapobj)
|
|
||||||
return data
|
return data
|
||||||
|
|
||||||
# Interface from BaseFolder
|
# Interface from BaseFolder
|
||||||
@ -304,63 +313,6 @@ class IMAPFolder(BaseFolder):
|
|||||||
return (headername, headervalue)
|
return (headername, headervalue)
|
||||||
|
|
||||||
|
|
||||||
def __savemessage_addheader(self, content, headername, headervalue):
|
|
||||||
self.ui.debug('imap',
|
|
||||||
'__savemessage_addheader: called to add %s: %s' % (headername,
|
|
||||||
headervalue))
|
|
||||||
insertionpoint = content.find(CRLF + CRLF)
|
|
||||||
self.ui.debug('imap', '__savemessage_addheader: insertionpoint = %d' % insertionpoint)
|
|
||||||
leader = content[0:insertionpoint]
|
|
||||||
self.ui.debug('imap', '__savemessage_addheader: leader = %s' % repr(leader))
|
|
||||||
if insertionpoint == 0 or insertionpoint == -1:
|
|
||||||
newline = ''
|
|
||||||
insertionpoint = 0
|
|
||||||
else:
|
|
||||||
newline = CRLF
|
|
||||||
newline += "%s: %s" % (headername, headervalue)
|
|
||||||
self.ui.debug('imap', '__savemessage_addheader: newline = ' + repr(newline))
|
|
||||||
trailer = content[insertionpoint:]
|
|
||||||
self.ui.debug('imap', '__savemessage_addheader: trailer = ' + repr(trailer))
|
|
||||||
return leader + newline + trailer
|
|
||||||
|
|
||||||
|
|
||||||
def __savemessage_delheaders(self, content, header_list):
|
|
||||||
"""
|
|
||||||
Deletes headers in the given list from the message content.
|
|
||||||
|
|
||||||
Arguments:
|
|
||||||
- content: message itself
|
|
||||||
- header_list: list of headers to be deleted or just the header name
|
|
||||||
|
|
||||||
We expect our message to have proper CRLF as line endings.
|
|
||||||
|
|
||||||
"""
|
|
||||||
if type(header_list) != type([]):
|
|
||||||
header_list = [header_list]
|
|
||||||
self.ui.debug('imap',
|
|
||||||
'__savemessage_delheaders: called to delete %s' % (header_list))
|
|
||||||
|
|
||||||
if not len(header_list): return content
|
|
||||||
|
|
||||||
eoh = content.find(CRLF + CRLF)
|
|
||||||
if eoh == -1:
|
|
||||||
eoh = len(content)
|
|
||||||
self.ui.debug('imap', '__savemessage_delheaders: end of headers = %d' % eoh)
|
|
||||||
headers = content[0:eoh]
|
|
||||||
rest = content[eoh:]
|
|
||||||
self.ui.debug('imap', '__savemessage_delheaders: headers = %s' % repr(headers))
|
|
||||||
new_headers = []
|
|
||||||
for h in headers.split(CRLF):
|
|
||||||
keep_it = True
|
|
||||||
for trim_h in self.filterheaders:
|
|
||||||
if len(h) > len(trim_h) and h[0:len(trim_h)+1] == (trim_h + ":"):
|
|
||||||
keep_it = False
|
|
||||||
break
|
|
||||||
if keep_it: new_headers.append(h)
|
|
||||||
|
|
||||||
return (CRLF.join(new_headers) + rest)
|
|
||||||
|
|
||||||
|
|
||||||
def __savemessage_searchforheader(self, imapobj, headername, headervalue):
|
def __savemessage_searchforheader(self, imapobj, headername, headervalue):
|
||||||
self.ui.debug('imap', '__savemessage_searchforheader called for %s: %s' % \
|
self.ui.debug('imap', '__savemessage_searchforheader called for %s: %s' % \
|
||||||
(headername, headervalue))
|
(headername, headervalue))
|
||||||
@ -554,14 +506,14 @@ class IMAPFolder(BaseFolder):
|
|||||||
self.savemessageflags(uid, flags)
|
self.savemessageflags(uid, flags)
|
||||||
return uid
|
return uid
|
||||||
|
|
||||||
|
content = self.deletemessageheaders(content, self.filterheaders)
|
||||||
|
|
||||||
# Use proper CRLF all over the message
|
# Use proper CRLF all over the message
|
||||||
content = re.sub("(?<!\r)\n", CRLF, content)
|
content = re.sub("(?<!\r)\n", CRLF, content)
|
||||||
|
|
||||||
# get the date of the message, so we can pass it to the server.
|
# get the date of the message, so we can pass it to the server.
|
||||||
date = self.__getmessageinternaldate(content, rtime)
|
date = self.__getmessageinternaldate(content, rtime)
|
||||||
|
|
||||||
content = self.__savemessage_delheaders(content, self.filterheaders)
|
|
||||||
|
|
||||||
retry_left = 2 # succeeded in APPENDING?
|
retry_left = 2 # succeeded in APPENDING?
|
||||||
imapobj = self.imapserver.acquireconnection()
|
imapobj = self.imapserver.acquireconnection()
|
||||||
# NB: in the finally clause for this try we will release
|
# NB: in the finally clause for this try we will release
|
||||||
@ -580,8 +532,8 @@ class IMAPFolder(BaseFolder):
|
|||||||
content)
|
content)
|
||||||
self.ui.debug('imap', 'savemessage: header is: %s: %s' %\
|
self.ui.debug('imap', 'savemessage: header is: %s: %s' %\
|
||||||
(headername, headervalue))
|
(headername, headervalue))
|
||||||
content = self.__savemessage_addheader(content, headername,
|
content = self.addmessageheader(content, headername, headervalue)
|
||||||
headervalue)
|
|
||||||
if len(content)>200:
|
if len(content)>200:
|
||||||
dbg_output = "%s...%s" % (content[:150], content[-50:])
|
dbg_output = "%s...%s" % (content[:150], content[-50:])
|
||||||
else:
|
else:
|
||||||
@ -685,6 +637,69 @@ class IMAPFolder(BaseFolder):
|
|||||||
self.ui.debug('imap', 'savemessage: returning new UID %d' % uid)
|
self.ui.debug('imap', 'savemessage: returning new UID %d' % uid)
|
||||||
return uid
|
return uid
|
||||||
|
|
||||||
|
|
||||||
|
def _fetch_from_imap(self, imapobj, uids, retry_num=1):
|
||||||
|
"""
|
||||||
|
Fetches data from IMAP server.
|
||||||
|
|
||||||
|
Arguments:
|
||||||
|
- imapobj: IMAPlib object
|
||||||
|
- uids: message UIDS
|
||||||
|
- retry_num: number of retries to make
|
||||||
|
|
||||||
|
Returns: data obtained by this query.
|
||||||
|
"""
|
||||||
|
query = "(%s)" % (" ".join(self.imap_query))
|
||||||
|
fails_left = retry_num # retry on dropped connection
|
||||||
|
while fails_left:
|
||||||
|
try:
|
||||||
|
imapobj.select(self.getfullname(), readonly = True)
|
||||||
|
res_type, data = imapobj.uid('fetch', uids, query)
|
||||||
|
fails_left = 0
|
||||||
|
except imapobj.abort as e:
|
||||||
|
# Release dropped connection, and get a new one
|
||||||
|
self.imapserver.releaseconnection(imapobj, True)
|
||||||
|
imapobj = self.imapserver.acquireconnection()
|
||||||
|
self.ui.error(e, exc_info()[2])
|
||||||
|
fails_left -= 1
|
||||||
|
if not fails_left:
|
||||||
|
raise e
|
||||||
|
if data == [None] or res_type != 'OK':
|
||||||
|
#IMAP server says bad request or UID does not exist
|
||||||
|
severity = OfflineImapError.ERROR.MESSAGE
|
||||||
|
reason = "IMAP server '%s' failed to fetch messages UID '%s'."\
|
||||||
|
"Server responded: %s %s" % (self.getrepository(), uids,
|
||||||
|
res_type, data)
|
||||||
|
if data == [None]:
|
||||||
|
#IMAP server did not find a message with this UID
|
||||||
|
reason = "IMAP server '%s' does not have a message "\
|
||||||
|
"with UID '%s'" % (self.getrepository(), uids)
|
||||||
|
raise OfflineImapError(reason, severity)
|
||||||
|
|
||||||
|
return data
|
||||||
|
|
||||||
|
|
||||||
|
def _store_to_imap(self, imapobj, uid, field, data):
|
||||||
|
"""
|
||||||
|
Stores data to IMAP server
|
||||||
|
|
||||||
|
Arguments:
|
||||||
|
- imapobj: instance of IMAPlib to use
|
||||||
|
- uid: message UID
|
||||||
|
- field: field name to be stored/updated
|
||||||
|
- data: field contents
|
||||||
|
|
||||||
|
"""
|
||||||
|
imapobj.select(self.getfullname())
|
||||||
|
res_type, retdata = imapobj.uid('store', uid, field, data)
|
||||||
|
if res_type != 'OK':
|
||||||
|
severity = OfflineImapError.ERROR.MESSAGE
|
||||||
|
reason = "IMAP server '%s' failed to store %s for message UID '%d'."\
|
||||||
|
"Server responded: %s %s" % (self.getrepository(), field, uid,
|
||||||
|
res_type, retdata)
|
||||||
|
raise OfflineImapError(reason, severity)
|
||||||
|
return retdata[0]
|
||||||
|
|
||||||
# Interface from BaseFolder
|
# Interface from BaseFolder
|
||||||
def savemessageflags(self, uid, flags):
|
def savemessageflags(self, uid, flags):
|
||||||
"""Change a message's flags to `flags`.
|
"""Change a message's flags to `flags`.
|
||||||
@ -694,17 +709,15 @@ class IMAPFolder(BaseFolder):
|
|||||||
dryrun mode."""
|
dryrun mode."""
|
||||||
imapobj = self.imapserver.acquireconnection()
|
imapobj = self.imapserver.acquireconnection()
|
||||||
try:
|
try:
|
||||||
try:
|
result = self._store_to_imap(imapobj, str(uid), 'FLAGS', imaputil.flagsmaildir2imap(flags))
|
||||||
imapobj.select(self.getfullname())
|
|
||||||
except imapobj.readonly:
|
except imapobj.readonly:
|
||||||
self.ui.flagstoreadonly(self, [uid], flags)
|
self.ui.flagstoreadonly(self, [uid], data)
|
||||||
return
|
return
|
||||||
result = imapobj.uid('store', '%d' % uid, 'FLAGS',
|
|
||||||
imaputil.flagsmaildir2imap(flags))
|
|
||||||
assert result[0] == 'OK', 'Error with store: ' + '. '.join(result[1])
|
|
||||||
finally:
|
finally:
|
||||||
self.imapserver.releaseconnection(imapobj)
|
self.imapserver.releaseconnection(imapobj)
|
||||||
result = result[1][0]
|
|
||||||
if not result:
|
if not result:
|
||||||
self.messagelist[uid]['flags'] = flags
|
self.messagelist[uid]['flags'] = flags
|
||||||
else:
|
else:
|
||||||
|
Loading…
Reference in New Issue
Block a user