zev/imbox
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

made Messages methods private, as well as uid_list.

Browse Source
This commit is contained in:
zevav
2018-07-25 12:11:59 -04:00
parent 8e26e92a39
commit 73fafcb368
1 changed files with 12 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
imbox/__init__.py
View File

@@ -85,25 +85,25 @@ class Messages:
self.connection = connection self.connection = connection
self.parser_policy = parser_policy self.parser_policy = parser_policy
self.kwargs = kwargs self.kwargs = kwargs
self.uid_list = self.query_uids(**kwargs) self._uid_list = self._query_uids(**kwargs)
logger.debug("Fetch all messages for UID in {}".format(self.uid_list)) logger.debug("Fetch all messages for UID in {}".format(self._uid_list))
def fetch_email(self, uid): def _fetch_email(self, uid):
return fetch_email_by_uid(uid=uid, return fetch_email_by_uid(uid=uid,
connection=self.connection, connection=self.connection,
parser_policy=self.parser_policy) parser_policy=self.parser_policy)
def query_uids(self, **kwargs): def _query_uids(self, **kwargs):
query_ = build_search_query(**kwargs) query_ = build_search_query(**kwargs)
message, data = self.connection.uid('search', None, query_) message, data = self.connection.uid('search', None, query_)
if data[0] is None: if data[0] is None:
return [] return []
return data[0].split() return data[0].split()
def fetch_email_list(self): def _fetch_email_list(self):
for uid in self.uid_list: for uid in self._uid_list:
yield uid, self.fetch_email(uid) yield uid, self._fetch_email(uid)
def __repr__(self): def __repr__(self):
if len(self.kwargs) > 0: if len(self.kwargs) > 0:
@@ -112,20 +112,20 @@ class Messages:
return 'Messages(ALL)' return 'Messages(ALL)'
def __iter__(self): def __iter__(self):
return self.fetch_email_list() return self._fetch_email_list()
def __next__(self): def __next__(self):
return self return self
def __len__(self): def __len__(self):
return len(self.uid_list) return len(self._uid_list)
def __getitem__(self, index): def __getitem__(self, index):
uids = self.uid_list[index] uids = self._uid_list[index]
if not isinstance(uids, list): if not isinstance(uids, list):
uid = uids uid = uids
return uid, self.fetch_email(uid) return uid, self._fetch_email(uid)
return [(uid, self.fetch_email(uid)) return [(uid, self._fetch_email(uid))
for uid in uids] for uid in uids]