From d7ef331e8b57d88d62fa323545f9ecd435db2e2d Mon Sep 17 00:00:00 2001
From: Vincent A <dev@indigo.re>
Date: Fri, 28 May 2021 16:10:01 +0200
Subject: [PATCH] [primonialreim] add missing get_document and Document.format

---
 modules/primonialreim/module.py | 3 +++
 modules/primonialreim/pages.py  | 2 ++
 modules/primonialreim/test.py   | 3 ++-
 3 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/modules/primonialreim/module.py b/modules/primonialreim/module.py
index c47546f4d7..b8a1e0fa84 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 bd030583a7..069b9896b0 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 2d556aa360..c97e7895f8 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 <http://www.gnu.org/licenses/>.
 
-from __future__ import unicode_literals
+# flake8: compatible
 
+from __future__ import unicode_literals
 
 from weboob.tools.test import BackendTest
 
-- 
GitLab