154f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum"""MH interface -- purely object-oriented (well, almost) 254f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum 354f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van RossumExecutive summary: 454f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum 554f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossumimport mhlib 654f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum 754f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossummh = mhlib.MH() # use default mailbox directory and profile 854f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossummh = mhlib.MH(mailbox) # override mailbox location (default from profile) 954f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossummh = mhlib.MH(mailbox, profile) # override mailbox and profile 1054f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum 1154f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossummh.error(format, ...) # print error message -- can be overridden 1254f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossums = mh.getprofile(key) # profile entry (None if not set) 1354f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossumpath = mh.getpath() # mailbox pathname 1454f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossumname = mh.getcontext() # name of current folder 1554f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossummh.setcontext(name) # set name of current folder 1654f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum 1754f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossumlist = mh.listfolders() # names of top-level folders 1854f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossumlist = mh.listallfolders() # names of all folders, including subfolders 1954f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossumlist = mh.listsubfolders(name) # direct subfolders of given folder 2054f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossumlist = mh.listallsubfolders(name) # all subfolders of given folder 2154f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum 2254f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossummh.makefolder(name) # create new folder 2354f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossummh.deletefolder(name) # delete folder -- must have no subfolders 2454f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum 2554f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossumf = mh.openfolder(name) # new open folder object 2654f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum 2754f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossumf.error(format, ...) # same as mh.error(format, ...) 2854f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossumpath = f.getfullname() # folder's full pathname 2954f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossumpath = f.getsequencesfilename() # full pathname of folder's sequences file 3054f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossumpath = f.getmessagefilename(n) # full pathname of message n in folder 3154f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum 3254f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossumlist = f.listmessages() # list of messages in folder (as numbers) 3354f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossumn = f.getcurrent() # get current message 3454f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossumf.setcurrent(n) # set current message 3554f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossumlist = f.parsesequence(seq) # parse msgs syntax into list of messages 3654f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossumn = f.getlast() # get last message (0 if no messagse) 3754f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossumf.setlast(n) # set last message (internal use only) 3854f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum 3954f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossumdict = f.getsequences() # dictionary of sequences in folder {name: list} 4054f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossumf.putsequences(dict) # write sequences back to folder 4154f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum 4254f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossumf.createmessage(n, fp) # add message from file f as number n 4354f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossumf.removemessages(list) # remove messages in list from folder 4454f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossumf.refilemessages(list, tofolder) # move messages in list to other folder 4554f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossumf.movemessage(n, tofolder, ton) # move one message to a given destination 4654f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossumf.copymessage(n, tofolder, ton) # copy one message to a given destination 4754f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum 4854f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossumm = f.openmessage(n) # new open message object (costs a file descriptor) 4954f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossumm is a derived class of mimetools.Message(rfc822.Message), with: 5054f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossums = m.getheadertext() # text of message's headers 5154f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossums = m.getheadertext(pred) # text of message's headers, filtered by pred 5254f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossums = m.getbodytext() # text of message's body, decoded 5354f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossums = m.getbodytext(0) # text of message's body, not decoded 5454f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum""" 552a869138645a38bea2517d00133158567a39caafBrett Cannonfrom warnings import warnpy3k 562a869138645a38bea2517d00133158567a39caafBrett Cannonwarnpy3k("the mhlib module has been removed in Python 3.0; use the mailbox " 572a869138645a38bea2517d00133158567a39caafBrett Cannon "module instead", stacklevel=2) 582a869138645a38bea2517d00133158567a39caafBrett Cannondel warnpy3k 5954f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum 60560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum# XXX To do, functionality: 61560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum# - annotate messages 624fe6caaaf082429e778a09043f6dce78b253b701Guido van Rossum# - send messages 63560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum# 6440b2cfb3f39e3db4c5f04f0545f9af7d299c2f4fGuido van Rossum# XXX To do, organization: 65560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum# - move IntSet to separate file 66560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum# - move most Message functionality to module mimetools 67560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum 68560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum 69560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum# Customizable defaults 70560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum 71560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van RossumMH_PROFILE = '~/.mh_profile' 72560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van RossumPATH = '~/Mail' 73560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van RossumMH_SEQUENCES = '.mh_sequences' 74560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van RossumFOLDER_PROTECT = 0700 75560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum 76560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum 77560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum# Imported modules 78560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum 79560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossumimport os 80508a092e2e44332c2f1ac381bdb956e3b94e0cc2Guido van Rossumimport sys 819694fcab5332f27dc28b195ba1391e5491d2eaefGuido van Rossumimport re 82560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossumimport mimetools 83560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossumimport multifile 8440b2cfb3f39e3db4c5f04f0545f9af7d299c2f4fGuido van Rossumimport shutil 857cfd31ee8a1447e177f129c4e3f1bfd937528043Guido van Rossumfrom bisect import bisect 86560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum 8717ab123cf1f0597e7e257c1ce83a6e87b85ffd7bSkip Montanaro__all__ = ["MH","Error","Folder","Message"] 88560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum 89560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum# Exported constants 90560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum 91ffdc48f45c7f942192ad19c26766207826ba99fdFred Drakeclass Error(Exception): 92ffdc48f45c7f942192ad19c26766207826ba99fdFred Drake pass 93560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum 94560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum 95560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossumclass MH: 9654f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """Class representing a particular collection of folders. 9754f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum Optional constructor arguments are the pathname for the directory 9854f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum containing the collection, and the MH profile to use. 9954f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum If either is omitted or empty a default is used; the default 10054f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum directory is taken from the MH profile if it is specified there.""" 101560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum 1020c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def __init__(self, path = None, profile = None): 10354f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """Constructor.""" 10416e3c427f35589ac3b83e8c13a8ec6495ec6cfa1Raymond Hettinger if profile is None: profile = MH_PROFILE 10545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum self.profile = os.path.expanduser(profile) 10616e3c427f35589ac3b83e8c13a8ec6495ec6cfa1Raymond Hettinger if path is None: path = self.getprofile('Path') 10745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if not path: path = PATH 10845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if not os.path.isabs(path) and path[0] != '~': 10945e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum path = os.path.join('~', path) 11045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum path = os.path.expanduser(path) 11145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if not os.path.isdir(path): raise Error, 'MH() path not found' 11245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum self.path = path 1130c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 1140c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def __repr__(self): 11554f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """String representation.""" 11670a6b49821a3226f55e9716f32d802d06640cb89Walter Dörwald return 'MH(%r, %r)' % (self.path, self.profile) 1170c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 1180c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def error(self, msg, *args): 11954f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """Routine to print an error. May be overridden by a derived class.""" 12045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum sys.stderr.write('MH error: %s\n' % (msg % args)) 1210c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 1220c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def getprofile(self, key): 12354f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """Return a profile entry, None if not found.""" 12445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return pickline(self.profile, key) 1250c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 1260c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def getpath(self): 12754f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """Return the path (the name of the collection's directory).""" 12845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return self.path 1290c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 1300c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def getcontext(self): 13154f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """Return the name of the current folder.""" 13245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum context = pickline(os.path.join(self.getpath(), 'context'), 13345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum 'Current-Folder') 13445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if not context: context = 'inbox' 13545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return context 1360c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 1370c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def setcontext(self, context): 13854f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """Set the name of the current folder.""" 13945e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum fn = os.path.join(self.getpath(), 'context') 14045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum f = open(fn, "w") 14145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum f.write("Current-Folder: %s\n" % context) 14245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum f.close() 1430c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 1440c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def listfolders(self): 14554f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """Return the names of the top-level folders.""" 14645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum folders = [] 14745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum path = self.getpath() 14845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum for name in os.listdir(path): 14945e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum fullname = os.path.join(path, name) 15045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if os.path.isdir(fullname): 15145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum folders.append(name) 15245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum folders.sort() 15345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return folders 1540c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 1550c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def listsubfolders(self, name): 15654f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """Return the names of the subfolders in a given folder 15754f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum (prefixed with the given folder name).""" 15845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum fullname = os.path.join(self.path, name) 15945e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum # Get the link count so we can avoid listing folders 16045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum # that have no subfolders. 16132200aeac697fcb3f2b4528127a2fbf0a22a8f17Raymond Hettinger nlinks = os.stat(fullname).st_nlink 162c54b8ceb91306359d1b7eb2c4c4ae356a8a0a271Serhiy Storchaka if nlinks == 2: 16345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return [] 16445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum subfolders = [] 16545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum subnames = os.listdir(fullname) 16645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum for subname in subnames: 16745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum fullsubname = os.path.join(fullname, subname) 16845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if os.path.isdir(fullsubname): 16945e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum name_subname = os.path.join(name, subname) 17045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum subfolders.append(name_subname) 17145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum # Stop looking for subfolders when 17245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum # we've seen them all 17345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum nlinks = nlinks - 1 174c54b8ceb91306359d1b7eb2c4c4ae356a8a0a271Serhiy Storchaka if nlinks == 2: 17545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum break 17645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum subfolders.sort() 17745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return subfolders 1780c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 1790c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def listallfolders(self): 18054f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """Return the names of all folders and subfolders, recursively.""" 18145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return self.listallsubfolders('') 1820c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 1830c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def listallsubfolders(self, name): 18454f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """Return the names of subfolders in a given folder, recursively.""" 18545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum fullname = os.path.join(self.path, name) 18645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum # Get the link count so we can avoid listing folders 18745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum # that have no subfolders. 18832200aeac697fcb3f2b4528127a2fbf0a22a8f17Raymond Hettinger nlinks = os.stat(fullname).st_nlink 189c54b8ceb91306359d1b7eb2c4c4ae356a8a0a271Serhiy Storchaka if nlinks == 2: 19045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return [] 19145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum subfolders = [] 19245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum subnames = os.listdir(fullname) 19345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum for subname in subnames: 19445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if subname[0] == ',' or isnumeric(subname): continue 19545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum fullsubname = os.path.join(fullname, subname) 19645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if os.path.isdir(fullsubname): 19745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum name_subname = os.path.join(name, subname) 19845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum subfolders.append(name_subname) 19945e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if not os.path.islink(fullsubname): 20045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum subsubfolders = self.listallsubfolders( 20145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum name_subname) 20245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum subfolders = subfolders + subsubfolders 20345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum # Stop looking for subfolders when 20445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum # we've seen them all 20545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum nlinks = nlinks - 1 206c54b8ceb91306359d1b7eb2c4c4ae356a8a0a271Serhiy Storchaka if nlinks == 2: 20745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum break 20845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum subfolders.sort() 20945e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return subfolders 2100c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 2110c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def openfolder(self, name): 21254f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """Return a new Folder object for the named folder.""" 21345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return Folder(self, name) 2140c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 2150c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def makefolder(self, name): 21654f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """Create a new folder (or raise os.error if it cannot be created).""" 21745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum protect = pickline(self.profile, 'Folder-Protect') 21845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if protect and isnumeric(protect): 21966d9919cab903e152d206b7edb853799eca564d0Eric S. Raymond mode = int(protect, 8) 22045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum else: 22145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum mode = FOLDER_PROTECT 22245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum os.mkdir(os.path.join(self.getpath(), name), mode) 2230c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 2240c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def deletefolder(self, name): 22554f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """Delete a folder. This removes files in the folder but not 22654f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum subdirectories. Raise os.error if deleting the folder itself fails.""" 22745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum fullname = os.path.join(self.getpath(), name) 22845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum for subname in os.listdir(fullname): 22945e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum fullsubname = os.path.join(fullname, subname) 23045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum try: 23145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum os.unlink(fullsubname) 23245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum except os.error: 23345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum self.error('%s not deleted, continuing...' % 23445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum fullsubname) 23545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum os.rmdir(fullname) 236560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum 237560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum 2389694fcab5332f27dc28b195ba1391e5491d2eaefGuido van Rossumnumericprog = re.compile('^[1-9][0-9]*$') 239560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossumdef isnumeric(str): 2409694fcab5332f27dc28b195ba1391e5491d2eaefGuido van Rossum return numericprog.match(str) is not None 241560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum 242560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossumclass Folder: 24354f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """Class representing a particular folder.""" 244560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum 2450c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def __init__(self, mh, name): 24654f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """Constructor.""" 24745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum self.mh = mh 24845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum self.name = name 24945e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if not os.path.isdir(self.getfullname()): 25045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum raise Error, 'no folder %s' % name 2510c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 2520c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def __repr__(self): 25354f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """String representation.""" 25470a6b49821a3226f55e9716f32d802d06640cb89Walter Dörwald return 'Folder(%r, %r)' % (self.mh, self.name) 2550c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 2560c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def error(self, *args): 25754f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """Error message handler.""" 25868468eba635570400f607e140425a222018e56f9Guido van Rossum self.mh.error(*args) 2590c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 2600c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def getfullname(self): 26154f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """Return the full pathname of the folder.""" 26245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return os.path.join(self.mh.path, self.name) 2630c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 2640c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def getsequencesfilename(self): 26554f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """Return the full pathname of the folder's sequences file.""" 26645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return os.path.join(self.getfullname(), MH_SEQUENCES) 2670c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 2680c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def getmessagefilename(self, n): 26954f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """Return the full pathname of a message in the folder.""" 27045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return os.path.join(self.getfullname(), str(n)) 2710c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 2720c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def listsubfolders(self): 27354f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """Return list of direct subfolders.""" 27445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return self.mh.listsubfolders(self.name) 2750c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 2760c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def listallsubfolders(self): 27754f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """Return list of all subfolders.""" 27845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return self.mh.listallsubfolders(self.name) 2790c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 2800c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def listmessages(self): 28154f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """Return the list of messages currently present in the folder. 28254f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum As a side effect, set self.last to the last message (or 0).""" 28345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum messages = [] 28445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum match = numericprog.match 28545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum append = messages.append 28645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum for name in os.listdir(self.getfullname()): 287d9d2625dbd284f91df2beab22f385e0f0bf60cabGuido van Rossum if match(name): 28845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum append(name) 28966d9919cab903e152d206b7edb853799eca564d0Eric S. Raymond messages = map(int, messages) 29045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum messages.sort() 29145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if messages: 29245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum self.last = messages[-1] 29345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum else: 29445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum self.last = 0 29545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return messages 296560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum 2970c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def getsequences(self): 29854f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """Return the set of sequences for the folder.""" 29945e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum sequences = {} 30045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum fullname = self.getsequencesfilename() 30145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum try: 30245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum f = open(fullname, 'r') 30345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum except IOError: 30445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return sequences 30545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum while 1: 30645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum line = f.readline() 30745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if not line: break 30866d9919cab903e152d206b7edb853799eca564d0Eric S. Raymond fields = line.split(':') 3098152d32375c40bba9ccbe43b780ebe96d9617781Fred Drake if len(fields) != 2: 31045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum self.error('bad sequence in %s: %s' % 31166d9919cab903e152d206b7edb853799eca564d0Eric S. Raymond (fullname, line.strip())) 31266d9919cab903e152d206b7edb853799eca564d0Eric S. Raymond key = fields[0].strip() 31366d9919cab903e152d206b7edb853799eca564d0Eric S. Raymond value = IntSet(fields[1].strip(), ' ').tolist() 31445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum sequences[key] = value 31545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return sequences 3160c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 3170c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def putsequences(self, sequences): 31854f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """Write the set of sequences back to the folder.""" 31945e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum fullname = self.getsequencesfilename() 32045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum f = None 32188f72ff95592730d3931d90028d0df934fa1a13dRaymond Hettinger for key, seq in sequences.iteritems(): 32245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum s = IntSet('', ' ') 32388f72ff95592730d3931d90028d0df934fa1a13dRaymond Hettinger s.fromlist(seq) 32445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if not f: f = open(fullname, 'w') 32545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum f.write('%s: %s\n' % (key, s.tostring())) 32645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if not f: 32745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum try: 32845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum os.unlink(fullname) 32945e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum except os.error: 33045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum pass 33145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum else: 33245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum f.close() 333560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum 3340c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def getcurrent(self): 335ffdc48f45c7f942192ad19c26766207826ba99fdFred Drake """Return the current message. Raise Error when there is none.""" 33645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum seqs = self.getsequences() 33745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum try: 33845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return max(seqs['cur']) 33945e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum except (ValueError, KeyError): 34045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum raise Error, "no cur message" 3410c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 3420c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def setcurrent(self, n): 34354f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """Set the current message.""" 34445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum updateline(self.getsequencesfilename(), 'cur', str(n), 0) 3450c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 3460c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def parsesequence(self, seq): 34754f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """Parse an MH sequence specification into a message list. 34854f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum Attempt to mimic mh-sequence(5) as close as possible. 34954f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum Also attempt to mimic observed behavior regarding which 35054f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum conditions cause which error messages.""" 35145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum # XXX Still not complete (see mh-format(5)). 35245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum # Missing are: 35345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum # - 'prev', 'next' as count 35445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum # - Sequence-Negation option 35545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum all = self.listmessages() 35645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum # Observed behavior: test for empty folder is done first 35745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if not all: 35845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum raise Error, "no messages in %s" % self.name 35945e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum # Common case first: all is frequently the default 36045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if seq == 'all': 36145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return all 36245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum # Test for X:Y before X-Y because 'seq:-n' matches both 36366d9919cab903e152d206b7edb853799eca564d0Eric S. Raymond i = seq.find(':') 36445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if i >= 0: 36545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum head, dir, tail = seq[:i], '', seq[i+1:] 36645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if tail[:1] in '-+': 36745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum dir, tail = tail[:1], tail[1:] 36845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if not isnumeric(tail): 36945e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum raise Error, "bad message list %s" % seq 37045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum try: 37166d9919cab903e152d206b7edb853799eca564d0Eric S. Raymond count = int(tail) 37245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum except (ValueError, OverflowError): 37345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum # Can't use sys.maxint because of i+count below 37445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum count = len(all) 37545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum try: 37645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum anchor = self._parseindex(head, all) 37745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum except Error, msg: 37845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum seqs = self.getsequences() 37954f0222547b1e92cd018ef132307a6f793dc9505Raymond Hettinger if not head in seqs: 38045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if not msg: 38145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum msg = "bad message list %s" % seq 38245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum raise Error, msg, sys.exc_info()[2] 38345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum msgs = seqs[head] 38445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if not msgs: 38545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum raise Error, "sequence %s empty" % head 38645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if dir == '-': 38745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return msgs[-count:] 38845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum else: 38945e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return msgs[:count] 39045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum else: 39145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if not dir: 39245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if head in ('prev', 'last'): 39345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum dir = '-' 39445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if dir == '-': 39545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum i = bisect(all, anchor) 39645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return all[max(0, i-count):i] 39745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum else: 39845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum i = bisect(all, anchor-1) 39945e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return all[i:i+count] 40045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum # Test for X-Y next 40166d9919cab903e152d206b7edb853799eca564d0Eric S. Raymond i = seq.find('-') 40245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if i >= 0: 40345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum begin = self._parseindex(seq[:i], all) 40445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum end = self._parseindex(seq[i+1:], all) 40545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum i = bisect(all, begin-1) 40645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum j = bisect(all, end) 40745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum r = all[i:j] 40845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if not r: 40945e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum raise Error, "bad message list %s" % seq 41045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return r 41145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum # Neither X:Y nor X-Y; must be a number or a (pseudo-)sequence 41245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum try: 41345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum n = self._parseindex(seq, all) 41445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum except Error, msg: 41545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum seqs = self.getsequences() 41654f0222547b1e92cd018ef132307a6f793dc9505Raymond Hettinger if not seq in seqs: 41745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if not msg: 41845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum msg = "bad message list %s" % seq 41945e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum raise Error, msg 42045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return seqs[seq] 42145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum else: 42245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if n not in all: 42345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if isnumeric(seq): 42445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum raise Error, "message %d doesn't exist" % n 42545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum else: 42645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum raise Error, "no %s message" % seq 42745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum else: 42845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return [n] 4290c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 4300c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def _parseindex(self, seq, all): 43154f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """Internal: parse a message number (or cur, first, etc.).""" 43245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if isnumeric(seq): 43345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum try: 43466d9919cab903e152d206b7edb853799eca564d0Eric S. Raymond return int(seq) 43545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum except (OverflowError, ValueError): 43645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return sys.maxint 43745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if seq in ('cur', '.'): 43845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return self.getcurrent() 43945e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if seq == 'first': 44045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return all[0] 44145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if seq == 'last': 44245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return all[-1] 44345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if seq == 'next': 44445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum n = self.getcurrent() 44545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum i = bisect(all, n) 44645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum try: 44745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return all[i] 44845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum except IndexError: 44945e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum raise Error, "no next message" 45045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if seq == 'prev': 45145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum n = self.getcurrent() 45245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum i = bisect(all, n-1) 45345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if i == 0: 45445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum raise Error, "no prev message" 45545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum try: 45645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return all[i-1] 45745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum except IndexError: 45845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum raise Error, "no prev message" 45945e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum raise Error, None 4600c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 4610c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def openmessage(self, n): 46254f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """Open a message -- returns a Message object.""" 46345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return Message(self, n) 4640c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 4650c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def removemessages(self, list): 46654f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """Remove one or more messages -- may raise os.error.""" 46745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum errors = [] 46845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum deleted = [] 46945e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum for n in list: 47045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum path = self.getmessagefilename(n) 47145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum commapath = self.getmessagefilename(',' + str(n)) 47245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum try: 47345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum os.unlink(commapath) 47445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum except os.error: 47545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum pass 47645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum try: 47745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum os.rename(path, commapath) 47845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum except os.error, msg: 47945e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum errors.append(msg) 48045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum else: 48145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum deleted.append(n) 48245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if deleted: 48345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum self.removefromallsequences(deleted) 48445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if errors: 48545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if len(errors) == 1: 48645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum raise os.error, errors[0] 48745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum else: 48845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum raise os.error, ('multiple errors:', errors) 4890c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 4900c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def refilemessages(self, list, tofolder, keepsequences=0): 49154f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """Refile one or more messages -- may raise os.error. 49254f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum 'tofolder' is an open folder object.""" 49345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum errors = [] 49445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum refiled = {} 49545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum for n in list: 49645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum ton = tofolder.getlast() + 1 49745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum path = self.getmessagefilename(n) 49845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum topath = tofolder.getmessagefilename(ton) 49945e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum try: 50045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum os.rename(path, topath) 50145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum except os.error: 50245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum # Try copying 50345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum try: 50445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum shutil.copy2(path, topath) 50545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum os.unlink(path) 50645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum except (IOError, os.error), msg: 50745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum errors.append(msg) 50845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum try: 50945e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum os.unlink(topath) 51045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum except os.error: 51145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum pass 51245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum continue 51345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum tofolder.setlast(ton) 51445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum refiled[n] = ton 51545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if refiled: 51645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if keepsequences: 51745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum tofolder._copysequences(self, refiled.items()) 51845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum self.removefromallsequences(refiled.keys()) 51945e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if errors: 52045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if len(errors) == 1: 52145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum raise os.error, errors[0] 52245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum else: 52345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum raise os.error, ('multiple errors:', errors) 5240c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 5250c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def _copysequences(self, fromfolder, refileditems): 52654f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """Helper for refilemessages() to copy sequences.""" 52745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum fromsequences = fromfolder.getsequences() 52845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum tosequences = self.getsequences() 52945e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum changed = 0 53045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum for name, seq in fromsequences.items(): 53145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum try: 53245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum toseq = tosequences[name] 53345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum new = 0 5343db163aa1952a53712d402f8f460228dde1192ceunknown except KeyError: 53545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum toseq = [] 53645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum new = 1 53745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum for fromn, ton in refileditems: 53845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if fromn in seq: 53945e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum toseq.append(ton) 54045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum changed = 1 54145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if new and toseq: 54245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum tosequences[name] = toseq 54345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if changed: 54445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum self.putsequences(tosequences) 5450c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 5460c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def movemessage(self, n, tofolder, ton): 54754f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """Move one message over a specific destination message, 54854f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum which may or may not already exist.""" 54945e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum path = self.getmessagefilename(n) 55045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum # Open it to check that it exists 55145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum f = open(path) 55245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum f.close() 55345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum del f 55445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum topath = tofolder.getmessagefilename(ton) 55545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum backuptopath = tofolder.getmessagefilename(',%d' % ton) 55645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum try: 55745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum os.rename(topath, backuptopath) 55845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum except os.error: 55945e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum pass 56045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum try: 56145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum os.rename(path, topath) 56245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum except os.error: 56345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum # Try copying 56445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum ok = 0 56545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum try: 56645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum tofolder.setlast(None) 56745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum shutil.copy2(path, topath) 56845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum ok = 1 56945e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum finally: 57045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if not ok: 57145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum try: 57245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum os.unlink(topath) 57345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum except os.error: 57445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum pass 57545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum os.unlink(path) 57645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum self.removefromallsequences([n]) 5770c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 5780c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def copymessage(self, n, tofolder, ton): 57954f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """Copy one message over a specific destination message, 58054f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum which may or may not already exist.""" 58145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum path = self.getmessagefilename(n) 58245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum # Open it to check that it exists 58345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum f = open(path) 58445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum f.close() 58545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum del f 58645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum topath = tofolder.getmessagefilename(ton) 58745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum backuptopath = tofolder.getmessagefilename(',%d' % ton) 58845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum try: 58945e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum os.rename(topath, backuptopath) 59045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum except os.error: 59145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum pass 59245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum ok = 0 59345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum try: 59445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum tofolder.setlast(None) 59545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum shutil.copy2(path, topath) 59645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum ok = 1 59745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum finally: 59845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if not ok: 59945e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum try: 60045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum os.unlink(topath) 60145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum except os.error: 60245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum pass 6030c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 6044e5cbcf5afa411d654983cd4cf8c99739b935b26Guido van Rossum def createmessage(self, n, txt): 60554f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """Create a message, with text from the open file txt.""" 60645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum path = self.getmessagefilename(n) 60745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum backuppath = self.getmessagefilename(',%d' % n) 60845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum try: 60945e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum os.rename(path, backuppath) 61045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum except os.error: 61145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum pass 61245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum ok = 0 61345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum BUFSIZE = 16*1024 61445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum try: 61545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum f = open(path, "w") 61645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum while 1: 61745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum buf = txt.read(BUFSIZE) 61845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if not buf: 61945e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum break 62045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum f.write(buf) 62145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum f.close() 62245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum ok = 1 62345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum finally: 62445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if not ok: 62545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum try: 62645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum os.unlink(path) 62745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum except os.error: 62845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum pass 6294e5cbcf5afa411d654983cd4cf8c99739b935b26Guido van Rossum 6300c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def removefromallsequences(self, list): 6317e47402264cf87b9bbb61fc9ff610af08add7c7bThomas Wouters """Remove one or more messages from all sequences (including last) 63254f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum -- but not from 'cur'!!!""" 63345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if hasattr(self, 'last') and self.last in list: 63445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum del self.last 63545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum sequences = self.getsequences() 63645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum changed = 0 63745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum for name, seq in sequences.items(): 63845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if name == 'cur': 63945e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum continue 64045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum for n in list: 64145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if n in seq: 64245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum seq.remove(n) 64345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum changed = 1 64445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if not seq: 64545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum del sequences[name] 64645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if changed: 64745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum self.putsequences(sequences) 6480c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 6490c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def getlast(self): 65054f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """Return the last message number.""" 65145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if not hasattr(self, 'last'): 652f93befc2094f5adea7a7b7aa8b979d52241b2062Guido van Rossum self.listmessages() # Set self.last 65345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return self.last 6540c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 6550c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def setlast(self, last): 65654f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """Set the last message number.""" 65745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if last is None: 65845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if hasattr(self, 'last'): 65945e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum del self.last 66045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum else: 66145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum self.last = last 662560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum 663560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossumclass Message(mimetools.Message): 664560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum 6650c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def __init__(self, f, n, fp = None): 66654f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """Constructor.""" 66745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum self.folder = f 66845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum self.number = n 66916e3c427f35589ac3b83e8c13a8ec6495ec6cfa1Raymond Hettinger if fp is None: 67045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum path = f.getmessagefilename(n) 67145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum fp = open(path, 'r') 67245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum mimetools.Message.__init__(self, fp) 6730c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 6740c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def __repr__(self): 67554f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """String representation.""" 67645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return 'Message(%s, %s)' % (repr(self.folder), self.number) 6770c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 6780c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def getheadertext(self, pred = None): 67954f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """Return the message's header text as a string. If an 68054f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum argument is specified, it is used as a filter predicate to 68154f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum decide which headers to return (its argument is the header 68254f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum name converted to lower case).""" 68316e3c427f35589ac3b83e8c13a8ec6495ec6cfa1Raymond Hettinger if pred is None: 68466d9919cab903e152d206b7edb853799eca564d0Eric S. Raymond return ''.join(self.headers) 68545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum headers = [] 68645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum hit = 0 68745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum for line in self.headers: 6886e025bcde84ec358734a116cefcc4acc357ee0b1Eric S. Raymond if not line[0].isspace(): 68966d9919cab903e152d206b7edb853799eca564d0Eric S. Raymond i = line.find(':') 69045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if i > 0: 69166d9919cab903e152d206b7edb853799eca564d0Eric S. Raymond hit = pred(line[:i].lower()) 69245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if hit: headers.append(line) 693c9838f9fcbf138b454606d606f7b53a5be21b5aeEric S. Raymond return ''.join(headers) 6940c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 6950c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def getbodytext(self, decode = 1): 69654f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """Return the message's body text as string. This undoes a 69754f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum Content-Transfer-Encoding, but does not interpret other MIME 69854f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum features (e.g. multipart messages). To suppress decoding, 69954f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum pass 0 as an argument.""" 70045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum self.fp.seek(self.startofbody) 70145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum encoding = self.getencoding() 7024fe6caaaf082429e778a09043f6dce78b253b701Guido van Rossum if not decode or encoding in ('', '7bit', '8bit', 'binary'): 70345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return self.fp.read() 704a617271dbd150549b021a5161d009869ad62f7b2Raymond Hettinger try: 705a617271dbd150549b021a5161d009869ad62f7b2Raymond Hettinger from cStringIO import StringIO 706a617271dbd150549b021a5161d009869ad62f7b2Raymond Hettinger except ImportError: 707a617271dbd150549b021a5161d009869ad62f7b2Raymond Hettinger from StringIO import StringIO 70845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum output = StringIO() 70945e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum mimetools.decode(self.fp, output, encoding) 71045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return output.getvalue() 7110c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 7120c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def getbodyparts(self): 71354f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """Only for multipart messages: return the message's body as a 71454f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum list of SubMessage objects. Each submessage object behaves 71554f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum (almost) as a Message object.""" 71645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if self.getmaintype() != 'multipart': 71745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum raise Error, 'Content-Type is not multipart/*' 71845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum bdry = self.getparam('boundary') 71945e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if not bdry: 72045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum raise Error, 'multipart/* without boundary param' 72145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum self.fp.seek(self.startofbody) 72245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum mf = multifile.MultiFile(self.fp) 72345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum mf.push(bdry) 72445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum parts = [] 72545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum while mf.next(): 72670a6b49821a3226f55e9716f32d802d06640cb89Walter Dörwald n = "%s.%r" % (self.number, 1 + len(parts)) 72745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum part = SubMessage(self.folder, n, mf) 72845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum parts.append(part) 72945e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum mf.pop() 73045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return parts 7310c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 7320c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def getbody(self): 73354f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """Return body, either a string or a list of messages.""" 73445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if self.getmaintype() == 'multipart': 73545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return self.getbodyparts() 73645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum else: 73745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return self.getbodytext() 738560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum 739560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum 740560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossumclass SubMessage(Message): 741560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum 7420c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def __init__(self, f, n, fp): 74354f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """Constructor.""" 74445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum Message.__init__(self, f, n, fp) 74545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if self.getmaintype() == 'multipart': 74645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum self.body = Message.getbodyparts(self) 74745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum else: 74845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum self.body = Message.getbodytext(self) 7494fe6caaaf082429e778a09043f6dce78b253b701Guido van Rossum self.bodyencoded = Message.getbodytext(self, decode=0) 75045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum # XXX If this is big, should remember file pointers 751560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum 7520c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def __repr__(self): 75354f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """String representation.""" 75445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum f, n, fp = self.folder, self.number, self.fp 75545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return 'SubMessage(%s, %s, %s)' % (f, n, fp) 756560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum 7574fe6caaaf082429e778a09043f6dce78b253b701Guido van Rossum def getbodytext(self, decode = 1): 7584fe6caaaf082429e778a09043f6dce78b253b701Guido van Rossum if not decode: 7594fe6caaaf082429e778a09043f6dce78b253b701Guido van Rossum return self.bodyencoded 76045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if type(self.body) == type(''): 76145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return self.body 762560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum 7630c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def getbodyparts(self): 76445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if type(self.body) == type([]): 76545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return self.body 766560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum 7670c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def getbody(self): 76845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return self.body 769560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum 770560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum 771560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossumclass IntSet: 77254f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """Class implementing sets of integers. 77354f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum 77454f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum This is an efficient representation for sets consisting of several 77554f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum continuous ranges, e.g. 1-100,200-400,402-1000 is represented 77654f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum internally as a list of three pairs: [(1,100), (200,400), 77754f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum (402,1000)]. The internal representation is always kept normalized. 77854f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum 77954f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum The constructor has up to three arguments: 78054f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum - the string used to initialize the set (default ''), 78154f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum - the separator between ranges (default ',') 78254f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum - the separator between begin and end of a range (default '-') 78354f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum The separators must be strings (not regexprs) and should be different. 78454f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum 78554f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum The tostring() function yields a string that can be passed to another 78654f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum IntSet constructor; __repr__() is a valid IntSet constructor itself. 78754f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum """ 78854f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum 78954f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum # XXX The default begin/end separator means that negative numbers are 79054f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum # not supported very well. 79154f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum # 79254f22ed30bab2e64909ba2d79205cb4b87c69db2Guido van Rossum # XXX There are currently no operations to remove set elements. 793560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum 7940c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def __init__(self, data = None, sep = ',', rng = '-'): 79545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum self.pairs = [] 79645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum self.sep = sep 79745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum self.rng = rng 7981ccccc08c2913b04fe9563b283dd1339d079a9bdMichael W. Hudson if data: self.fromstring(data) 7990c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 8000c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def reset(self): 80145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum self.pairs = [] 8020c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 8030c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def __cmp__(self, other): 80445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return cmp(self.pairs, other.pairs) 8050c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 8060c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def __hash__(self): 80745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return hash(self.pairs) 8080c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 8090c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def __repr__(self): 81070a6b49821a3226f55e9716f32d802d06640cb89Walter Dörwald return 'IntSet(%r, %r, %r)' % (self.tostring(), self.sep, self.rng) 8110c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 8120c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def normalize(self): 81345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum self.pairs.sort() 81445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum i = 1 81545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum while i < len(self.pairs): 81645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum alo, ahi = self.pairs[i-1] 81745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum blo, bhi = self.pairs[i] 81845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if ahi >= blo-1: 81945e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum self.pairs[i-1:i+1] = [(alo, max(ahi, bhi))] 82045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum else: 82145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum i = i+1 8220c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 8230c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def tostring(self): 82445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum s = '' 82545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum for lo, hi in self.pairs: 82670a6b49821a3226f55e9716f32d802d06640cb89Walter Dörwald if lo == hi: t = repr(lo) 82770a6b49821a3226f55e9716f32d802d06640cb89Walter Dörwald else: t = repr(lo) + self.rng + repr(hi) 82845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if s: s = s + (self.sep + t) 82945e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum else: s = t 83045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return s 8310c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 8320c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def tolist(self): 83345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum l = [] 83445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum for lo, hi in self.pairs: 83545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum m = range(lo, hi+1) 83645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum l = l + m 83745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return l 8380c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 8390c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def fromlist(self, list): 84045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum for i in list: 84145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum self.append(i) 8420c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 8430c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def clone(self): 84445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum new = IntSet() 84545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum new.pairs = self.pairs[:] 84645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return new 8470c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 8480c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def min(self): 84945e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return self.pairs[0][0] 8500c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 8510c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def max(self): 85245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return self.pairs[-1][-1] 8530c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 8540c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def contains(self, x): 85545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum for lo, hi in self.pairs: 856bc0e9108261693b6278687f4fb4709ff76c2e543Tim Peters if lo <= x <= hi: return True 857bc0e9108261693b6278687f4fb4709ff76c2e543Tim Peters return False 8580c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 8590c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def append(self, x): 86045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum for i in range(len(self.pairs)): 86145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum lo, hi = self.pairs[i] 86245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if x < lo: # Need to insert before 86345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if x+1 == lo: 86445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum self.pairs[i] = (x, hi) 86545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum else: 86645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum self.pairs.insert(i, (x, x)) 86745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if i > 0 and x-1 == self.pairs[i-1][1]: 86845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum # Merge with previous 86945e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum self.pairs[i-1:i+1] = [ 87045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum (self.pairs[i-1][0], 87145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum self.pairs[i][1]) 87245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum ] 87345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return 87445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if x <= hi: # Already in set 87545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return 87645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum i = len(self.pairs) - 1 87745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if i >= 0: 87845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum lo, hi = self.pairs[i] 87945e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if x-1 == hi: 88045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum self.pairs[i] = lo, x 88145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return 88245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum self.pairs.append((x, x)) 8830c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 8840c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def addpair(self, xlo, xhi): 88545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if xlo > xhi: return 88645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum self.pairs.append((xlo, xhi)) 88745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum self.normalize() 8880c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum 8890c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def fromstring(self, data): 89045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum new = [] 89166d9919cab903e152d206b7edb853799eca564d0Eric S. Raymond for part in data.split(self.sep): 89245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum list = [] 89366d9919cab903e152d206b7edb853799eca564d0Eric S. Raymond for subp in part.split(self.rng): 89466d9919cab903e152d206b7edb853799eca564d0Eric S. Raymond s = subp.strip() 89566d9919cab903e152d206b7edb853799eca564d0Eric S. Raymond list.append(int(s)) 89645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if len(list) == 1: 89745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum new.append((list[0], list[0])) 89845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum elif len(list) == 2 and list[0] <= list[1]: 89945e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum new.append((list[0], list[1])) 90045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum else: 90145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum raise ValueError, 'bad data passed to IntSet' 90245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum self.pairs = self.pairs + new 90345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum self.normalize() 904560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum 905560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum 906560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum# Subroutines to read/write entries in .mh_profile and .mh_sequences 907560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum 908560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossumdef pickline(file, key, casefold = 1): 9090c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum try: 91045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum f = open(file, 'r') 9110c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum except IOError: 91245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum return None 9139694fcab5332f27dc28b195ba1391e5491d2eaefGuido van Rossum pat = re.escape(key) + ':' 9149694fcab5332f27dc28b195ba1391e5491d2eaefGuido van Rossum prog = re.compile(pat, casefold and re.IGNORECASE) 9150c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum while 1: 91645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum line = f.readline() 91745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if not line: break 91845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if prog.match(line): 91945e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum text = line[len(key)+1:] 92045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum while 1: 92145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum line = f.readline() 9226e025bcde84ec358734a116cefcc4acc357ee0b1Eric S. Raymond if not line or not line[0].isspace(): 92345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum break 92445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum text = text + line 92566d9919cab903e152d206b7edb853799eca564d0Eric S. Raymond return text.strip() 9260c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum return None 927560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum 928560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossumdef updateline(file, key, value, casefold = 1): 9290c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum try: 93045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum f = open(file, 'r') 93145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum lines = f.readlines() 93245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum f.close() 9330c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum except IOError: 93445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum lines = [] 9359694fcab5332f27dc28b195ba1391e5491d2eaefGuido van Rossum pat = re.escape(key) + ':(.*)\n' 9369694fcab5332f27dc28b195ba1391e5491d2eaefGuido van Rossum prog = re.compile(pat, casefold and re.IGNORECASE) 9370c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum if value is None: 93845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum newline = None 9390c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum else: 94045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum newline = '%s: %s\n' % (key, value) 9410c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum for i in range(len(lines)): 94245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum line = lines[i] 94345e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if prog.match(line): 94445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if newline is None: 94545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum del lines[i] 94645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum else: 94745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum lines[i] = newline 94845e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum break 9490c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum else: 95045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum if newline is not None: 95145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum lines.append(newline) 9520c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum tempfile = file + "~" 9530c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum f = open(tempfile, 'w') 9540c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum for line in lines: 95545e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum f.write(line) 9560c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum f.close() 9570c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum os.rename(tempfile, file) 958560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum 959560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum 960560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum# Test program 961560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum 962560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossumdef test(): 9630c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum global mh, f 9640c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum os.system('rm -rf $HOME/Mail/@test') 9650c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum mh = MH() 9660c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum def do(s): print s; print eval(s) 9670c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum do('mh.listfolders()') 9680c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum do('mh.listallfolders()') 9690c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum testfolders = ['@test', '@test/test1', '@test/test2', 97045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum '@test/test1/test11', '@test/test1/test12', 97145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum '@test/test1/test11/test111'] 97270a6b49821a3226f55e9716f32d802d06640cb89Walter Dörwald for t in testfolders: do('mh.makefolder(%r)' % (t,)) 9730c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum do('mh.listsubfolders(\'@test\')') 9740c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum do('mh.listallsubfolders(\'@test\')') 9750c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum f = mh.openfolder('@test') 9760c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum do('f.listsubfolders()') 9770c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum do('f.listallsubfolders()') 9780c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum do('f.getsequences()') 9790c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum seqs = f.getsequences() 9800c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum seqs['foo'] = IntSet('1-10 12-20', ' ').tolist() 9810c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum print seqs 9820c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum f.putsequences(seqs) 9830c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum do('f.getsequences()') 98470a6b49821a3226f55e9716f32d802d06640cb89Walter Dörwald for t in reversed(testfolders): do('mh.deletefolder(%r)' % (t,)) 9850c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum do('mh.getcontext()') 9860c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum context = mh.getcontext() 9870c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum f = mh.openfolder(context) 9880c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum do('f.getcurrent()') 989dbecd93b7203cd187c1978de1207c29d3a9a686cRaymond Hettinger for seq in ('first', 'last', 'cur', '.', 'prev', 'next', 99045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum 'first:3', 'last:3', 'cur:3', 'cur:-3', 99145e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum 'prev:3', 'next:3', 99245e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum '1:3', '1:-3', '100:3', '100:-3', '10000:3', '10000:-3', 993dbecd93b7203cd187c1978de1207c29d3a9a686cRaymond Hettinger 'all'): 99445e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum try: 99570a6b49821a3226f55e9716f32d802d06640cb89Walter Dörwald do('f.parsesequence(%r)' % (seq,)) 99645e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum except Error, msg: 99745e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum print "Error:", msg 99870a6b49821a3226f55e9716f32d802d06640cb89Walter Dörwald stuff = os.popen("pick %r 2>/dev/null" % (seq,)).read() 99966d9919cab903e152d206b7edb853799eca564d0Eric S. Raymond list = map(int, stuff.split()) 100045e2fbc2e70ef28b1f0327207f33dab3a4e825c5Guido van Rossum print list, "<-- pick" 10010c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum do('f.listmessages()') 1002560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum 1003560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossum 1004560131328c9ddfa25555048b1bf4d760c02e1dd7Guido van Rossumif __name__ == '__main__': 10050c5e049c751c45aca40e26e3f1f76597798b86d9Guido van Rossum test() 1006