LCPQ/spip2md
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

more typing

Browse Source
This commit is contained in:
Guilhem Fauré 2023-05-22 11:40:12 +02:00
parent 1dc7d72987
commit 6fb1e9a29e
1 changed files with 10 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
spip2md/items.py
View File

@ -15,7 +15,7 @@ FILETYPE: str = "md"
class Item: class Item:
id: int id: int
def __init__(self, item) -> None: def __init__(self, item: SpipArticles | SpipRubriques) -> None:
self.title: str = convert_meta(item.titre) self.title: str = convert_meta(item.titre)
self.section_id: int = item.id_rubrique self.section_id: int = item.id_rubrique
self.description: str = convert_meta(item.descriptif) self.description: str = convert_meta(item.descriptif)
@ -74,7 +74,7 @@ class Item:
class Article(Item): class Article(Item):
def __init__(self, article) -> None: def __init__(self, article: SpipArticles) -> None:
super().__init__(article) super().__init__(article)
self.id: int = article.id_article self.id: int = article.id_article
self.surtitle: str = convert_meta(article.surtitre) # Probably unused self.surtitle: str = convert_meta(article.surtitre) # Probably unused
@ -93,7 +93,7 @@ class Article(Item):
# self.popularity: float = article.popularite # USELESS in static # self.popularity: float = article.popularite # USELESS in static
# self.version = article.id_version # USELESS # self.version = article.id_version # USELESS
def get_authors(self) -> tuple: def get_authors(self) -> list[SpipAuteurs]:
return ( return (
SpipAuteurs.select() SpipAuteurs.select()
.join( .join(
@ -103,7 +103,7 @@ class Article(Item):
.where(SpipAuteursLiens.id_objet == self.id) .where(SpipAuteursLiens.id_objet == self.id)
) )
def get_frontmatter(self) -> str: def get_frontmatter(self, append: Optional[dict[str, Any]] = None) -> str:
return super().get_frontmatter( return super().get_frontmatter(
{ {
"surtitle": self.surtitle, "surtitle": self.surtitle,
@ -114,7 +114,10 @@ class Article(Item):
"spip_id_rubrique": self.section_id, "spip_id_rubrique": self.section_id,
"spip_id_secteur": self.sector_id, "spip_id_secteur": self.sector_id,
"spip_chapo": self.caption, "spip_chapo": self.caption,
}, }
| append
if append is not None
else {},
) )
def get_body(self) -> str: def get_body(self) -> str:
@ -132,7 +135,7 @@ class Article(Item):
class Section(Item): class Section(Item):
def __init__(self, section) -> None: def __init__(self, section: SpipRubriques) -> None:
super().__init__(section) super().__init__(section)
self.id: int = section.id_rubrique self.id: int = section.id_rubrique
self.parent_id: int = section.id_parent self.parent_id: int = section.id_parent
@ -165,7 +168,7 @@ class LimitCounter:
class Iterator: class Iterator:
items: list items: list[Any]
def __init__(self) -> None: def __init__(self) -> None:
# Set a counter caped at the number of retrieved items # Set a counter caped at the number of retrieved items