diff options
Diffstat (limited to 'generator.py')
-rw-r--r-- | generator.py | 44 |
1 files changed, 23 insertions, 21 deletions
diff --git a/generator.py b/generator.py index de029a2..4c45caf 100644 --- a/generator.py +++ b/generator.py @@ -32,7 +32,7 @@ def generatePageXML(data, env_vars): """ template = open(env_vars['template_atom_post'], 'r').read() date = datetime.strptime(data['date'],"%d-%m-%Y").isoformat() + "Z" - return template.replace("$TITLE", data['title']).replace("$DATE", data['date']).replace("$CONTENT", + return template.replace("$TITLE", data['title']).replace("$DATE", date).replace("$CONTENT", data['content']).replace("$URL", env_vars['website_url'] + data['filepath']) @@ -43,16 +43,16 @@ def generateAtomFeed(posts, env_vars): env_vars: dictionnary of env variables """ # Generate RSS / Atom post - atom_content = [] + atom_content = "" for post in posts: # Checking if there is metadata, if not we don't create a RSS / Atom post - if post['date'] != '': - atom_content.append(generatePageXML(post, env_vars)) + if post['date'] != '01-01-0001': + atom_content += generatePageXML(post, env_vars) # Generate RSS / atom feed template = open(env_vars['template_atom_feed'], 'r').read() output = open(env_vars['parent_path'] + '/atom.xml', 'w') - output.write(template.replace("$CONTENT", atom_content)) + output.write(template.replace('$CONTENT', atom_content).replace('$DATE', datetime.today().strftime("%d-%m-%Y"))) output.close() @@ -63,18 +63,19 @@ def generateIndex(data, env_vars): env_vars: dictionnary of env variables """ # Create the index content - index_content = "<ul>" + index_content = "<ul>\n" for page in data: # Checking if there is metadata, if not we don't add the page in the index - if page['date'] != '': - index_content += '<li><a href="' + page['filepath'] + '">' + page['title'] + '</a><p>' - + page['date'] + '</p></li>\n' - index_content += "</ul>" + if page['date'] != '01-01-0001': + print(page['date']) + index_content += ('\t\t\t\t<li><a href="' + page['filepath'] + '">' + page['title'] + '</a><p>' + + page['date'] + '</p></li>\n') + index_content += "\t\t\t</ul>" # Generate main page template = open(env_vars['template_index'], 'r').read() output = open(env_vars['parent_path'] + '/index.html', 'w') - output.write(template.replace("$CONTENT", index_content)) + output.write(template.replace('$CONTENT', index_content)) output.close() @@ -100,7 +101,7 @@ if __name__=="__main__": # Getting env variable env_vars = { 'parent_path' : os.environ.get('PARENT_PATH'), 'pages_path' : os.environ.get('PAGES_PATH') , 'markdown_path' : os.environ.get('MARKDOWN_PATH'), 'template_page' : os.environ.get('TEMPLATE_PAGE') - , 'template_atom_post' : os.environ.get('TEMPLATE_PAGE'), 'template_atom_feed' : os.environ.get('TEMPLATE_ATOM_FEED') + , 'template_atom_post' : os.environ.get('TEMPLATE_ATOM_POST'), 'template_atom_feed' : os.environ.get('TEMPLATE_ATOM_FEED') , 'website_url' : os.environ.get('WEBSITE_URL'), 'template_index' : os.environ.get('TEMPLATE_INDEX') } # Checking if generate folder exist to remove previouly generated content, if not create it @@ -112,20 +113,21 @@ if __name__=="__main__": else: os.mkdir(env_vars['pages_path']) + data = [] # A list for data generated by md2html + # Generate all markdown file for file in os.listdir(env_vars['markdown_path']): # Generating HTML page print(f"{color['green']}Generating file: {file} {color['end']}") - data = [] # A list for data generated by md2html data.append(md2html(file, env_vars)) - sorted_data = sorted(data, key=lambda x:datetime.strptime(x['date'], '%d/%m/%Y')) + sorted_data = sorted(data, key=lambda x:datetime.strptime(x['date'], '%d-%m-%Y')) - # Generating atom feed - print(f"{color['green']}Generating RSS / Atom feed {color['end']}") - generateAtomFeed(data, env_vars) - - # Generating index - print(f"{color['green']}Generating main page {color['end']}") - generateIndex(data, env_vars) + # Generating atom feed + print(f"{color['green']}Generating RSS / Atom feed {color['end']}") + generateAtomFeed(data, env_vars) + + # Generating index + print(f"{color['green']}Generating main page {color['end']}") + generateIndex(data, env_vars) |