Changeset 126:4319998fc9b5 for imapclient

Timestamp:

01/25/10 13:26:30 (2 years ago)

Author:

Menno Smits <menno@…>

Branch:

default

Message:

select_folder() now returns the full parsed SELECT response (#24)

Location:

imapclient

Files:

• imapclient.py (modified) (2 diffs)
• response_parser.py (modified) (2 diffs)

imapclient/imapclient.py

@@ -15 +15 @@
     'RECENT']
 
-from response_parser import parse_fetch_response
+from response_parser import parse_response, parse_fetch_response
+
 
 # System flags
@@ -205 +206 @@
 
         @param folder: The folder name.
-        @return: Number of messages in the folder.
-        @rtype: long int
+        @return: A dictionary containing the SELECT response
+          values. At least the EXISTS, FLAGS and RECENT keys are
+          guaranteed to exist. Example:
+            {'EXISTS': 3,
+             'FLAGS': ('\\Answered', '\\Flagged', '\\Deleted', ... ),
+             'RECENT': 0,
+             'PERMANENTFLAGS': ('\\Answered', '\\Flagged', '\\Deleted', ... ),
+             'READ-WRITE': True,
+             'UIDNEXT': 11,
+             'UIDVALIDITY': 1239278212}
         """
         typ, data = self._imap.select(self._encode_folder_name(folder))
         self._checkok('select', typ, data)
-        return long(data[0])
+        return self._process_select_response(self._imap.untagged_responses)
+
+
+    def _process_select_response(self, resp):
+        out = {}
+        for key, value in resp.iteritems():
+            key = key.upper()
+            if key == 'OK':
+                continue
+            elif key in ('EXISTS', 'RECENT', 'UIDNEXT', 'UIDVALIDITY'):
+                value = int(value[0])
+            elif key in ('FLAGS', 'PERMANENTFLAGS'):
+                value = parse_response(value)[0]
+            elif key == 'READ-WRITE':
+                value = True
+                                 
+            out[key] = value
+        return out
 
 

imapclient/response_parser.py

@@ -162 +162 @@
 
     CTRL_CHARS = ''.join([chr(ch) for ch in range(32)])
-    ATOM_SPECIALS = r'()%*"' + CTRL_CHARS
+    SPECIALS = r'()%"' + CTRL_CHARS
     ALL_CHARS = [chr(ch) for ch in range(256)]
-    ATOM_NON_SPECIALS = [ch for ch in ALL_CHARS if ch not in ATOM_SPECIALS]
+    NON_SPECIALS = [ch for ch in ALL_CHARS if ch not in SPECIALS]
 
     def __init__(self, resp_chunks):
@@ -174 +174 @@
         self.lex.quotes = '"'
         self.lex.commenters = ''
-        self.lex.wordchars = self.ATOM_NON_SPECIALS
+        self.lex.wordchars = self.NON_SPECIALS
 
     def __iter__(self):