Allow no posts (#55)

* Fix : Allow no post for sitemap --------- Co-authored-by: Clément <[email protected]>
algoo · Aug 6, 2024 · 2b617e5 · 2b617e5
1 parent e8e1ad6
commit 2b617e5
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/jssg/models.py b/jssg/models.py
@@ -373,8 +373,11 @@ def get_postlists(cls) :
 
     @classmethod
     def get_pages(cls) :
-        return [{"page": page} for page in range(1, cls().nb_pages+1)]
-
+        if len(list(Post.load_glob(all = True))) > 0 :
+            return [{"page": page} for page in range(1, cls().nb_pages+1)]
+        else : 
+            return []
+
     @property
     def posts(self) :
         posts = sorted(Post.load_glob(all=True), key=lambda p: p.timestamp, reverse=True)

diff --git a/jssg/sitemaps.py b/jssg/sitemaps.py
@@ -13,7 +13,10 @@ def get_urls(self, site=None, **kwargs):
 
 class ConstantUrlSitemap(MySitemap) :
     def items(self) :
-        return ["/", "/atom.xml", "/sitemap.xml"]
+        if len(list(Post.load_glob(all = True))) > 0 :
+            return ["/", "/atom.xml", "/sitemap.xml"]
+        else : 
+            return ["/", "/sitemap.xml"]
     def location(self, url) -> str:
         return url