def main(category_name, imap_folder_name): cat = Category.get(category_name) with config.IMAP_connection() as im: for msg_num, msg in each_imap_message(im, imap_folder_name): print ('Training "%s"' % msg.get('Subject', '(none)').replace('\n',' ')) cat.train(str(msg)) return 'not implemented'
def ret(self, cat, *a, **kw): if not cat and self.current_category: cat = self.current_category return fn(self, cat, *a, **kw) elif cat and Category.exists(cat): cat = Category.get(cat) return fn(self, cat, *a, **kw) elif cat and not Category.exists(cat): print "category %s not found. try 'categories'?" % cat else: print "must specify a category. try 'categories'?"