Changeset 91:cd7e6ad6d118 for imapclient/imapclient.py

Timestamp:

28/12/09 18:37:09 (2 years ago)

Author:

msmits@…

Branch:

default

Message:

Backslash escaped double quotes in folder names now handled correctly

Also extracted the common code for list_folders and
list_sub_folders. The tests act on that directly.

Files:

• imapclient/imapclient.py (modified) (3 diffs)

imapclient/imapclient.py

@@ -64 +64 @@
 
     re_sep = re.compile('^\(\("[^"]*" "([^"]+)"\)\)')
-    re_folder = re.compile('\([^)]*\) "[^"]+" "?([^"]+)"?')
+#     re_folder = re.compile('\([^)]*\) "[^"]+" "?([^"]+)"?')
+    re_folder = re.compile(r'\([^)]*\) "[^"]+" (?P<qqq>"?)(?P<folder>.+)(?P=qqq)')
     re_status = re.compile(r'^\s*"?(?P<folder>[^"]+)"?\s+'
                            r'\((?P<status_items>.*)\)$')
@@ -160 +161 @@
         typ, data = self._imap.list(directory, pattern)
         self._checkok('list', typ, data)
-
+        return self._proc_folder_list(data)
+
+
+    def list_sub_folders(self, directory="", pattern="*"):
+        """Get a listing of subscribed folders on the server.
+
+        The default behaviour (no args) will list all subscribed folders for the
+        logged in user.
+
+        @param directory: The base directory to look for folders from.
+        @param pattern: A pattern to match against folder names. Only folder
+            names matching this pattern will be returned. Wildcards accepted.
+        @return: A list of folder names. As per the return of list_folders().
+        """
+        typ, data = self._imap.lsub(directory, pattern)
+        self._checkok('lsub', typ, data)
+        return self._proc_folder_list(data)
+
+
+    def _proc_folder_list(self, folder_data):
         folders = []
-        for line in data:
+        for line in folder_data:
             #TODO can the FetchParser code be adapted for use here?
             folder_text = None
@@ -170 +190 @@
                 match = self.re_folder.match(line)
                 if match:
-                    folder_text = match.group(1)
+                    folder_text = match.group('folder')
+                    folder_text = folder_text.replace(r'\"', '"')
             if folder_text is not None:
                 folders.append(self._decode_folder_name(folder_text))
         return folders
-
-
-    def list_sub_folders(self, directory="", pattern="*"):
-        """Get a listing of subscribed folders on the server.
-
-        The default behaviour (no args) will list all subscribed folders for the
-        logged in user.
-
-        @param directory: The base directory to look for folders from.
-        @param pattern: A pattern to match against folder names. Only folder
-            names matching this pattern will be returned. Wildcards accepted.
-        @return: A list of folder names. As per the return of list_folders().
-        """
-        typ, data = self._imap.lsub(directory, pattern)
-        self._checkok('lsub', typ, data)
-
-        folders = []
-        for line in data:
-            if line:
-                m = self.re_folder.match(line)
-                if m:
-                    folders.append(self._decode_folder_name(m.group(1)))
-        return folders
-
+        
 
     def select_folder(self, folder):