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.

112 lines
3.1 KiB

from pathlib import Path
from itertools import chain
import os
import re
3 years ago
from pelican import signals
from pelican.readers import MarkdownReader
from pelican.utils import pelican_open
3 years ago
from markdown import Markdown
ARTICLES = {}
FILES = {}
link = r'\[\[\s*(?P<filename>[\w+\s.]+)(\|\s*(?P<linkname>[\w\s]+))?\]\]'
file_re = re.compile(r'!' + link)
link_re = re.compile(link)
"""
# Test cases
[[my link]]
[[ my work ]]
[[ my work | is finished ]]
3 years ago
![[ a file.jpg ]]
![[file.jpg]]
"""
3 years ago
class ObsidianMarkdownReader(MarkdownReader):
3 years ago
"""
Change the format of various links to the accepted case of pelican.
"""
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
def replace_obsidian_links(self, text):
def replacement(match):
nonlocal self
group = match.groupdict()
filename = group['filename'].strip()
linkname = group['linkname'] if group['linkname'] else filename
linkname = linkname.strip()
path = ARTICLES.get(filename)
if path:
link_structure = '[{linkname}]({{filename}}/{path}/{filename}.md)'.format(
linkname=linkname, path=path, filename=filename
)
else:
link_structure = '{linkname}'.format(linkname=linkname)
return link_structure
text = link_re.sub(replacement, text)
return text
def read(self, source_path):
"""Parse content and metadata of markdown files
It also changes the links to the acceptable format for pelican
"""
self._source_path = source_path
self._md = Markdown(**self.settings['MARKDOWN'])
with pelican_open(source_path) as text:
text = self.replace_obsidian_links(text)
content = self._md.convert(text)
if hasattr(self._md, 'Meta'):
metadata = self._parse_metadata(self._md.Meta)
else:
metadata = {}
return content, metadata
def populate_files_and_articles(article_generator):
global ARTICLES
global FILES
base_path = Path(article_generator.path)
if not ARTICLES:
articles = chain(base_path.glob('**/*.md'), base_path.glob('**/*.md'))
for article in articles:
full_path, filename_w_ext = os.path.split(article)
filename, ext = os.path.splitext(filename_w_ext)
full_path = str(full_path).replace(str(base_path) + '/', '')
ARTICLES[filename] = full_path
if not FILES:
pass
def modify_reader(article_generator):
populate_files_and_articles(article_generator)
article_generator.readers.readers['md'] = ObsidianMarkdownReader(article_generator.settings)
3 years ago
def modify_metadata(article_generator, metadata):
3 years ago
"""
Modify the tags so we can define the tags as we are used to in obsidian.
3 years ago
"""
for tag in metadata.get('tags', []):
if '#' in tag.name:
tag.name = tag.name.replace('#', '')
3 years ago
def register():
signals.article_generator_context.connect(modify_metadata)
signals.article_generator_init.connect(modify_reader)