From 95ccc5fab5e2d56413189a5b89f6721a7194a789 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Guilhem=20Faur=C3=A9?= <pro@gfaure.eu>
Date: Mon, 22 May 2023 14:45:43 +0200
Subject: [PATCH] fixes & typing

---
 spip2md/database.py |  1 +
 spip2md/items.py    | 13 +++++++------
 2 files changed, 8 insertions(+), 6 deletions(-)

diff --git a/spip2md/database.py b/spip2md/database.py
index 0ed303b..ec9fd47 100644
--- a/spip2md/database.py
+++ b/spip2md/database.py
@@ -1,4 +1,5 @@
 # pyright: basic
+# type: ignore
 from peewee import (
     SQL,
     BigAutoField,
diff --git a/spip2md/items.py b/spip2md/items.py
index c4b09e5..bc57de3 100644
--- a/spip2md/items.py
+++ b/spip2md/items.py
@@ -30,7 +30,7 @@ class Item:
         self.sector_id: int = item.id_secteur
         self.update: str = item.maj
         self.lang: str = item.lang
-        self.set_lang: bool = item.langue_choisie  # TODO Why ?
+        self.set_lang: bool = item.langue_choisie == "oui"  # TODO Why ?
         self.translation_key: int = item.id_trad
         self.extra: str = convert_body(item.extra)  # Probably unused
 
@@ -88,7 +88,7 @@ class Article(Item):
         self.ps: str = convert_body(article.ps)  # Probably unused
         self.update_2: str = article.date_modif  # Probably unused duplicate of maj
         self.creation: str = article.date_redac
-        self.forum: bool = article.accepter_forum  # TODO Why ?
+        self.forum: bool = article.accepter_forum == "oui"  # TODO Why ?
         self.sitename: str = article.nom_site  # Probably useless
         self.virtual: str = article.virtuel  # TODO Why ?
         self.microblog: str = article.microblog  # Probably unused
@@ -162,10 +162,10 @@ class Document:
         self.date: str = document.date
         self.description: str = document.descriptif
         self.file: str = document.fichier
-        self.draft: bool = item.statut == "publie"
-        self.creation: str = item.date
-        self.publication: str = item.date_publication
-        self.update: str = item.maj
+        self.draft: bool = document.statut == "publie"
+        self.creation: str = document.date
+        self.publication: str = document.date_publication
+        self.update: str = document.maj
 
     def get_slug(self, date: bool = False) -> str:
         return slugify((self.publication + "-" if date else "") + self.title)
@@ -246,6 +246,7 @@ class Documents(Iterator):
                 SpipDocumentsLiens,
                 on=(SpipDocuments.id_document == SpipDocumentsLiens.id_document),
             )
+            .where(SpipDocumentsLiens.id_objet == object_id)
             .order_by(SpipArticles.date.desc())
         )
         super().__init__()