diff --git a/modules/primonialreim/module.py b/modules/primonialreim/module.py index c47546f4d74a0d185da66c4c104aefcb8deb46ba..b8a1e0fa84106c026d3d26ee55a6571a74c7faee 100644 --- a/modules/primonialreim/module.py +++ b/modules/primonialreim/module.py @@ -67,6 +67,9 @@ def get_subscription(self, id): def iter_documents(self, subscription): return self.browser.iter_documents() + def get_document(self, id): + return find_object(self.iter_documents(None), id=id, error=DocumentNotFound) + def download_document(self, document): if isinstance(document, str): document = find_object(self.iter_documents(None), id=document, error=DocumentNotFound) diff --git a/modules/primonialreim/pages.py b/modules/primonialreim/pages.py index bd030583a7abe471e0ab85d0a701846c532051c4..069b9896b0a79e3e84fe9548c182f3fdb34a2622 100644 --- a/modules/primonialreim/pages.py +++ b/modules/primonialreim/pages.py @@ -74,6 +74,8 @@ class item(ItemElement): klass = Document obj_type = DocumentTypes.NOTICE + obj_format = "pdf" + obj_url = AbsoluteLink(".") obj_id = Regexp(obj_url, r"/([^/]+)\.pdf") diff --git a/modules/primonialreim/test.py b/modules/primonialreim/test.py index 2d556aa3609f7c68e8b86f2d6997e631b7b9dddf..c97e7895f89d2d371ecab06db91588df657ebe3a 100644 --- a/modules/primonialreim/test.py +++ b/modules/primonialreim/test.py @@ -17,8 +17,9 @@ # You should have received a copy of the GNU Lesser General Public License # along with this weboob module. If not, see . -from __future__ import unicode_literals +# flake8: compatible +from __future__ import unicode_literals from weboob.tools.test import BackendTest