diff options
Diffstat (limited to 'generator.py')
| -rw-r--r-- | generator.py | 127 |
1 files changed, 127 insertions, 0 deletions
diff --git a/generator.py b/generator.py new file mode 100644 index 0000000..793eb28 --- /dev/null +++ b/generator.py @@ -0,0 +1,127 @@ +import os +import dotenv +from markdown_parser import * +from datetime import datetime + +def md2html(filename, env_vars): + """ + Create the html webpage from template and markdown content + filename: file to transform into HTML + env_vars: dictionnary of env variables + return: a dictionnary containing title, metadata, local path, content for HTML + """ + # Getting parsed content of markdown file & page template + data = parsemd(filename, env_vars) + template = open(env_vars['template_page'], 'r').read() + + # Generating the HTML page + output = open(env_vars['pages_path'] + '/' + filename.split('.')[0] + '.html', 'w') + output.write(template.replace("$CONTENT", data['content']).replace("$TITLE", data['title']). + replace("$DATE", data['date']).replace("$DESC", data['description'])) + output.close() + + return data + + +def generatePageXML(data, env_vars): + """ + Generate a RSS / Atom post for the page + data: dictionnary generated by the markdown parser + env_vars: dictionnary of env variables + return: RSS / Atom post + """ + 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", + data['content']).replace("$URL", env_vars['website_url'] + data['filepath']) + + +def generateAtomFeed(posts, env_vars): + """ + Generate a RSS / Atom feed + posts: list of data get from markdown pages + env_vars: dictionnary of env variables + """ + # Generate RSS / Atom post + atom_content = [] + for post in posts: + 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.close() + + +def generateIndex(data, env_vars): + """ + Generate the main page + data: list of data get from markdown pages + env_vars: dictionnary of env variables + """ + # Create the index content + index_content = "<ul>" + for page in data: + index_content += '<li><a href="' + page['filepath'] + '">' + page['title'] + '</a><p>' + + page['date'] + '</p></li>\n' + index_content += "</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.close() + + +if __name__=="__main__": + # Load .env file into python environment + dotenv.load_dotenv() + + # Color for print + color = { 'red': '\033[1;31m', 'green' : '\033[1;32m', 'end' : '\033[0m'} + + # Checking if all environment variable are present & setup + env = ['PARENT_PATH', 'PAGES_PATH', 'MARKDOWN_PATH', 'TEMPLATE_PAGE', 'TEMPLATE_ATOM_POST', + 'TEMPLATE_ATOM_FEED', 'WEBSITE_URL', 'TEMPLATE_INDEX'] + for variable in env: + if variable not in os.environ: + print(f"{color['red']}{variable} isn't present in the .env file, please fix this {color['end']}") + quit() + + if (os.environ.get(variable) or '') == '': + print(f"{color['red']}{variable} isn't setup in the .env file, please fix this {color['end']}") + quit() + + # 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') + , '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 + if os.path.exists(env_vars['pages_path']): + for file in os.listdir(env_vars['pages_path']): + os.remove(env_vars['pages_path'] + '/' + file) + os.remove(env_vars['parent_path'] + '/atom.xml') + os.remove(env_vars['parent_path'] + '/index.html') + else: + os.mkdir(env_vars['pages_path']) + + # 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')) + + # 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)
\ No newline at end of file |