diff --git a/modules/societegenerale/module.py b/modules/societegenerale/module.py index 7e5c46380f5ccd284cf0ebb4b061034170b85f73..f1b695359cb75e42cc0423b6a14a9c13323b4983 100644 --- a/modules/societegenerale/module.py +++ b/modules/societegenerale/module.py @@ -158,7 +158,16 @@ def iter_documents(self, subscription): return self.browser.iter_documents(subscription) def iter_documents_by_types(self, subscription, accepted_types): - return self.browser.iter_documents_by_types(subscription, accepted_types) + if not isinstance(subscription, Subscription): + subscription = self.get_subscription(subscription) + + if self.config['website'].get() not in ('ent', 'pro'): + for doc in self.browser.iter_documents_by_types(subscription, accepted_types): + yield doc + else: + for doc in self.browser.iter_documents(subscription): + if doc.type in accepted_types: + yield doc def download_document(self, document): if not isinstance(document, Document):