personal garden & website
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 

194 lines
7.2 KiB

import os
import markdown
import re
from jinja2 import Environment, FileSystemLoader
import yaml
import json
class Website:
def __init__(self, template_dir, public_dir):
self.template_dir_root = template_dir
self.public_dir = public_dir
self.output_dir_root = 'public/articles'
self.env = Environment(loader=FileSystemLoader(self.template_dir_root))
self.pages = []
self.tags = []
self.all_images = []
self.about_meta, self.about_content = self.fetch_page('content', 'about.md')
def build(self):
self.fetch_pages()
self.process_page()
self.fetch_tags()
self.create_list()
self.build_about()
self.fetch_all_images()
self.create_json('public/json/articles.json')
def fetch_pages(self):
for page in os.listdir('content/posts'):
if page.endswith('.md'):
with open(os.path.join('content/posts', page), 'r', encoding="utf8") as f:
content = f.read()
parts = content.split('---')
metadata = yaml.safe_load(parts[1])
md_content = ''.join(parts[2:])
md_content = re.sub(r'\(([^)]+)\)\[([^\]]+)\]', r'<a href="\2">\1</a>', md_content)
md_content = self.format_content(md_content)
html_content = markdown.markdown(md_content)
output_filename = os.path.splitext(page)[0] + '.html'
new_page = Page(metadata, html_content, output_filename)
self.pages.append(new_page)
def process_page(self):
template = self.env.get_template('article.html')
self.check_output_dir()
for page in self.pages:
metadata = page.get_metadata()
filename = page.get_filename()
html_content = page.get_content()
html_output = template.render(
title=metadata['title'],
year=metadata['year'],
date=metadata['date'],
tags=metadata.get('tags', []),
image=metadata['image'],
showcase=metadata['showcase'],
credits=metadata['credits'],
references=page.get_references(),
content=html_content)
with open(os.path.join(self.output_dir_root, filename), 'w', encoding='utf8') as output_file:
output_file.write(html_output)
def check_output_dir(self):
if not os.path.exists(self.output_dir_root):
os.makedirs(self.output_dir_root)
def fetch_page(self, dir, page):
for about in os.listdir(dir):
if about == page:
with open(os.path.join(dir, about), 'r', encoding='utf8') as f:
content = f.read()
parts = content.split('---')
metadata = yaml.safe_load(parts[1])
md_content = ''.join(parts[2:])
md_content = re.sub(r'\(([^)]+)\)\[([^\]]+)\]', r'<a href="\2">\1</a>', md_content)
html_content = markdown.markdown(md_content)
return metadata, html_content
def create_list(self):
template = self.env.get_template('list.html')
self.check_output_dir()
html_output = template.render(
tags=self.tags,
pages=self.pages
)
with open(os.path.join('public', 'list.html'), 'w', encoding='utf8') as output_file:
output_file.write(html_output)
def fetch_tags(self):
for page in self.pages:
page_tags = page.get_tags()
for tag in page_tags:
if tag not in self.tags:
self.tags.append(tag)
def build_about(self):
template = self.env.get_template('about.html')
html_output = template.render(
content=self.about_content,
socials=self.about_meta
)
with open(os.path.join('public', 'about.html'), 'w', encoding='utf8') as output_file:
output_file.write(html_output)
def format_content(self, content):
# convert all (link)(src) to <a> tags
content = re.sub(r'\(([^)]+)\)\[([^\]]+)\]', r'<a href="\2">\1</a>', content)
return content
def fetch_all_images(self):
dir = 'public/images'
template = self.env.get_template('gallery.html')
image_extensions = ('.png', '.jpg', '.jpeg', '.gif', '.webp')
for dirpath, _, filename in os.walk(dir):
for filename in filename:
if filename.lower().endswith(image_extensions):
relative_path = os.path.relpath(dirpath, dir).replace("\\", "/")
image_path = os.path.join('/images', relative_path, filename).replace("\\", "/")
if image_path.startswith("/images/."):
image_path = image_path.replace("/images/.", "/images")
self.all_images.append(image_path)
html_output = template.render(
images=self.all_images
)
with open(os.path.join('public', 'gallery.html'), 'w', encoding='utf8') as output_file:
output_file.write(html_output)
def create_json(self, json_file):
page_info_list = []
for page in self.pages:
page_info = {
'name': page.get_title(),
'filename': '/articles/' + page.get_filename(),
'image': page.get_image()
}
page_info_list.append(page_info)
with open(json_file, 'w', encoding='utf8') as f:
json.dump(page_info_list, f, ensure_ascii=False, indent=4)
class Page:
def __init__(self, metadata, md_content, filename):
self.metadata = metadata
self.content = md_content
self.filename = filename
self.title = self.metadata['title']
self.type = self.metadata['type']
self.year = self.metadata['year']
self.image_src = self.metadata['image']
self.tags = self.metadata['tags']
self.date = self.metadata['date']
self.showcase = self.metadata.get('showcase', [])
self.credits = self.metadata.get('credits', {})
self.references = self.metadata.get('references', [])
self.isDraft = self.metadata['draft']
def get_metadata(self):
return self.metadata
def get_content(self):
return self.content
def get_filename(self):
return self.filename
def get_title(self):
return self.title
def get_tags(self):
return self.tags
def get_year(self):
return self.year
def get_image(self):
return self.image_src
def get_references(self):
return self.references
def get_src(self):
return os.path.splitext(self.filename)[0]
def display_metadata(self):
output = f"Title: {self.title}\nYear: {self.year}\nDate: {self.date}\nTags: {self.tags}\nType: {self.type}"
print(output)
def main():
inst = Website('templates', 'public')
inst.build()
if __name__ == "__main__":
main()