Ajout du RSS/Atom & de la génération du fichier index

1 files changed, 39 insertions, 7 deletions

diff --git a/tools/page-generator.py b/tools/page-generator.py
index e5747db..bde61a0 100644
--- a/tools/page-generator.py
+++ b/tools/page-generator.py
@@ -2,10 +2,12 @@ import os
 
 articles_path = 'articles/'
 generate_path = 'archives/'
+atom_content = ""
+index_content = "<h2>Articles</h2><ul>"
 
 """Parse le fichier markdown et retourne le contenu à mettre dans notre page html"""
 def parsemd(filename):
-    content = {'content': '', 'title': '', 'date': '', 'description': ''}
+    content = {'content': '', 'title': '', 'date': '', 'description': '', 'filename': generate_path + filename.split('.')[0] + '.html'}
     inquote, inpre, inul = False, False, False
     for line in open('../' + articles_path + filename, "r"):
         line = line.strip()
@@ -89,22 +91,52 @@ def parsemd(filename):
 def md2html(filename):
     content = parsemd(filename) # Contenu parsé de notre fichier markdown
     template = open('page_template.html', 'r').read()
+
+    # Création du fichier html & ajout du contenu
     output = open('../' + generate_path + filename.split('.')[0] + '.html', 'w')
     output.write(template.replace("$CONTENT", content['content']).replace("$TITLE", content['title']).
      replace("$DATE", content['date']).replace("$DESC", content['description']))
+    output.close()
+
+    # Génération du post Atom et de l'index
+    generatePageXML(content)
+    global index_content
+    index_content += '<li><a href="' + content['filename'] + '">' + content['title'] + '</a></li>\n'
 
+""" Génère le post Atom"""
+def generatePageXML(data):
+    global atom_content
+    template = open('atom_post_template.xml', 'r').read()
+    atom_content += template.replace("$TITLE", data['title']).replace("$DATE", data['date']).replace("$CONTENT", 
+     data['content']).replace("$URL", "https://blog.arka.rocks/" + data['filename'])
 
-"""
-def generateRSS(data):
+""" Génère le fichier Atom"""
+def generateAtom():
+    template = open('atom_template.xml', 'r').read()
+    output = open('../atom.xml', 'w')
+    output.write(template.replace("$CONTENT", atom_content))
+    output.close()
+
+""" Génère l'index"""
+def updateIndex():
+    template = open('index_template.html', 'r').read()
+    output = open('../index.html', 'w')
+    output.write(template.replace("$CONTENT", index_content + '</ul>'))
+    output.close()
 
-def updateIndex(data):
-"""
 if __name__=="__main__":
     if not os.path.exists('../' + generate_path):
         os.mkdir('../' + generate_path)
-    else:
+    else: # Régénation du blog
         for file in os.listdir('../' + generate_path):
             os.remove('../' + generate_path + file)
+        os.remove('../atom.xml')
+        os.remove('../index.html')
+
     for file in os.listdir('../' + articles_path):
         print("Génération en cours du fichier: " + file)
-        md2html(file)
\ No newline at end of file
+        md2html(file)
+        print("Génération du fichier RSS / Atom")
+        generateAtom()
+        print("Génération de l'index")
+        updateIndex()
\ No newline at end of file