mirror of
https://github.com/ZetaKebab/quartz.git
synced 2025-01-14 22:18:43 +00:00
65 lines
1.6 KiB
Python
65 lines
1.6 KiB
Python
|
import sys
|
||
|
|
import os
|
||
|
|
from glob import glob
|
||
|
|
from bs4 import BeautifulSoup
|
||
|
|
import re
|
||
|
|
|
||
|
|
### Patterns ###
|
||
|
|
|
||
|
|
re_usage = r'(^\*\*Usage\*\*$\n\n)'
|
||
|
|
callout_title_html = '''
|
||
|
|
<div class="callout-title">
|
||
|
|
<div class="callout-icon"></div>
|
||
|
|
<div class="callout-title-inner"><p>Usage</p></div>
|
||
|
|
</div>'''
|
||
|
|
|
||
|
|
### Open file ###
|
||
|
|
|
||
|
|
def open_file(filename):
|
||
|
|
with open(filename, 'r', encoding='utf-8-sig') as file:
|
||
|
|
filedata = file.read()
|
||
|
|
return filedata
|
||
|
|
|
||
|
|
### Write to file ###
|
||
|
|
|
||
|
|
def write_file(filename, text):
|
||
|
|
with open(filename, 'w', encoding='utf-8-sig') as file:
|
||
|
|
file.write(text)
|
||
|
|
|
||
|
|
### Replace ###
|
||
|
|
|
||
|
|
def convert_pattern(text):
|
||
|
|
text = re.sub(re_usage, '', text, flags=re.MULTILINE)
|
||
|
|
|
||
|
|
soup = BeautifulSoup(text, features="html.parser")
|
||
|
|
|
||
|
|
for tag in soup.find_all('div', {"class": "usage"}):
|
||
|
|
new_blockquote = soup.new_tag("blockquote", **{"class": "callout notes", "data-callout": "notes"})
|
||
|
|
tag.wrap(new_blockquote)
|
||
|
|
|
||
|
|
calloutTitle = BeautifulSoup(callout_title_html, features="html.parser")
|
||
|
|
tag.insert_before(calloutTitle)
|
||
|
|
|
||
|
|
return str(soup)
|
||
|
|
|
||
|
|
|
||
|
|
### Execution ###
|
||
|
|
|
||
|
|
file_list = []
|
||
|
|
|
||
|
|
if len(sys.argv) > 1:
|
||
|
|
path = sys.argv[1]
|
||
|
|
for filename in glob(path + "/**/*.md", recursive=True):
|
||
|
|
if (os.path.isfile(filename)):
|
||
|
|
file_list.append(filename)
|
||
|
|
else:
|
||
|
|
print(f"Usage: python3 {sys.argv[0]} [root folder]")
|
||
|
|
print(f"Note: Please be careful, this software applies modifications recursively!")
|
||
|
|
|
||
|
|
for filename in file_list:
|
||
|
|
print(f"Found file {filename}, processing...", end='')
|
||
|
|
data = open_file(filename)
|
||
|
|
text = convert_pattern(data)
|
||
|
|
write_file(filename, text)
|
||
|
|
print(f" done!")
|