forked from CrossDark/CrossDown
332 lines
9.9 KiB
Python
332 lines
9.9 KiB
Python
from typing import *
|
||
import re
|
||
|
||
|
||
class Header:
|
||
def __init__(self, text: str):
|
||
self.text = text
|
||
|
||
def __call__(self, *args, **kwargs):
|
||
"""
|
||
渲染标题
|
||
:return: 处理后的文本
|
||
"""
|
||
h6 = re.sub(r'###### (.*?)\n', r'<h6>\1</h6>\n', self.text) # H6
|
||
h5 = re.sub(r'##### (.*?)\n', r'<h5>\1</h5>\n', h6) # H5
|
||
h4 = re.sub(r'#### (.*?)\n', r'<h4>\1</h4>\n', h5) # H4
|
||
h3 = re.sub(r'### (.*?)\n', r'<h3>\1</h3>\n', h4) # H3
|
||
h2 = re.sub(r'## (.*?)\n', r'<h2>\1</h2>\n', h3) # H2
|
||
h1 = re.sub(r'# (.*?)\n', r'<h1>\1</h1>\n', h2) # H1
|
||
return h1
|
||
|
||
|
||
class Style:
|
||
"""
|
||
渲染字体样式
|
||
"""
|
||
|
||
def __init__(self, text: str):
|
||
"""
|
||
初始化
|
||
:param text: cd文本
|
||
"""
|
||
self.text = text
|
||
|
||
def italic(self):
|
||
"""
|
||
斜体
|
||
:return:
|
||
"""
|
||
self.text = re.sub(r'\*([^*\n]+)\*', r'<i>\1</i>', self.text)
|
||
|
||
def bold(self):
|
||
"""
|
||
粗体
|
||
:return:
|
||
"""
|
||
self.text = re.sub(r'\*\*([^*\n]+)\*\*', r'<b>\1</b>', self.text)
|
||
|
||
def underline(self):
|
||
"""
|
||
下划线
|
||
:return:
|
||
"""
|
||
self.text = re.sub(r'~([^~\n]+)~', r'<u>\1</u>', self.text)
|
||
|
||
def strikethrough(self):
|
||
"""
|
||
删除线
|
||
:return:
|
||
"""
|
||
self.text = re.sub(r'~~([^~\n]+)~~', r'<s>\1</s>', self.text)
|
||
|
||
def highlight(self):
|
||
"""
|
||
高亮
|
||
:return:
|
||
"""
|
||
self.text = re.sub(r'==([^=\n]+)==', r'<mark>\1</mark>', self.text)
|
||
|
||
def __call__(self, *args, **kwargs):
|
||
"""
|
||
一键运行
|
||
:param args:
|
||
:param kwargs:
|
||
:return:
|
||
"""
|
||
self.bold()
|
||
self.italic()
|
||
self.strikethrough()
|
||
self.underline()
|
||
self.highlight()
|
||
return self.text
|
||
|
||
|
||
class Function:
|
||
"""
|
||
添加特殊功能
|
||
"""
|
||
|
||
def __init__(self, text: str):
|
||
"""
|
||
初始化
|
||
:param text: cd文本
|
||
"""
|
||
self.text = text
|
||
|
||
def link(self):
|
||
self.text = re.sub(r'\[([^\[\]\n]+)]\(([^()\n]+)\)', r'<a href="\2">\1</a>', self.text)
|
||
|
||
def __call__(self, *args, **kwargs):
|
||
"""
|
||
一键运行
|
||
:param args:
|
||
:param kwargs:
|
||
:return:
|
||
"""
|
||
self.link()
|
||
return self.text
|
||
|
||
|
||
class Value:
|
||
def __init__(self, text: str):
|
||
self.text = text
|
||
self.values = {
|
||
key: value for key, value in re.findall(r'\{([^{}]+)} ?= ?(.+?)(?=\n|$)', text)
|
||
} # 从text中提取所有变量并转换成字典
|
||
|
||
def __call__(self, *args, **kwargs) -> Tuple[str, Dict[str, str]]:
|
||
"""
|
||
将所有变量赋值并移除变量定义
|
||
:param args:
|
||
:param kwargs:
|
||
:return: 赋值后的正文
|
||
"""
|
||
text = self.text
|
||
for k, v in self.values.items():
|
||
text = re.sub(r'\{' + k + '} ?= ?(.+?)(?=\n|$)', '', text) # 移除变量的定义
|
||
text = re.sub(r'\{' + k + '}', fr'{v}', text) # 给变量赋值
|
||
return text, self.values
|
||
|
||
|
||
class CodeBlock:
|
||
def __init__(self, text: str):
|
||
"""
|
||
找出代码块并移除代码标识
|
||
:param text: 输入的文本
|
||
"""
|
||
self.codes = [i for i in re.findall(r'`([^`]*)`', text) if i != ''] # 找出代码快
|
||
self.text = re.sub(r'``', '', text) # 移除代码标识`
|
||
|
||
def __call__(self, *args, **kwargs):
|
||
"""
|
||
临时移除代码块
|
||
:param args:
|
||
:param kwargs:
|
||
:return: 不含代码的文本
|
||
"""
|
||
for index, item in enumerate(self.codes): # 替换代码块为-@@-(ID)-@@-
|
||
self.text = re.sub(fr'`{re.escape(item)}`', f'-@@-{index}-@@-', self.text) # 同时转译特殊字符
|
||
return self.text
|
||
|
||
def rendering(self, values: Dict[str, str]):
|
||
"""
|
||
渲染代码
|
||
:param values: 变量字典
|
||
:return:
|
||
"""
|
||
for index, code in enumerate(self.codes):
|
||
if re.match(r'\{[^$]*}', code): # 是变量
|
||
# 给变量赋值
|
||
key = re.findall(r'\{([^}]+)}', code)[0] # 查找变量名
|
||
code = re.sub(r'\{' + key + '}', fr'{values[key]}', code)
|
||
self.codes[index] = code # 给变量赋值
|
||
if re.search(r'\n', code): # 是多行代码
|
||
head = re.findall(r'(.*?)\n', code)[0]
|
||
if head == 'python':
|
||
pass
|
||
elif head == 'shell':
|
||
pass
|
||
elif re.match(r'\$[^$]*\$', code): # 是LaTex代码
|
||
self.codes[index] = re.sub(fr'\$([^$]*)\$', r'\(\1\)', code)
|
||
|
||
def restore(self, new_text: str):
|
||
"""
|
||
将渲染好的代码重新放回处理好的正文
|
||
:param new_text: 处理好的正文
|
||
:return: 加上代码的文章
|
||
"""
|
||
for index, item in enumerate(self.codes):
|
||
new_text = re.sub(fr'-@@-{index}-@@-', f'{item}', new_text, flags=re.DOTALL)
|
||
return new_text
|
||
|
||
|
||
class Escape:
|
||
"""
|
||
转义\后字符
|
||
"""
|
||
|
||
def __init__(self, text: str):
|
||
"""
|
||
找出转义符并转义
|
||
:param text: 输入的文本
|
||
"""
|
||
self.text = text
|
||
self.escapes = {
|
||
i: f'--@|{i}|@--' for i in re.findall(r'\\(.)', text)
|
||
} # 找出要转义的字符
|
||
print(self.escapes)
|
||
|
||
def __call__(self, *args, **kwargs):
|
||
"""
|
||
临时移除代码块
|
||
:param args:
|
||
:param kwargs:
|
||
:return: 不含代码的文本
|
||
"""
|
||
# TODO
|
||
for index, item in enumerate(self.escapes): # 替换代码块为-@@-(ID)-@@-
|
||
self.text = re.sub(fr'{index}', f'-@@-{index}-@@-', self.text) # 同时转译特殊字符
|
||
return self.text
|
||
|
||
def rendering(self, values: Dict[str, str]):
|
||
"""
|
||
渲染代码
|
||
:param values: 变量字典
|
||
:return:
|
||
"""
|
||
for index, code in enumerate(self.codes):
|
||
if re.match(r'\{[^$]*}', code): # 是变量
|
||
# 给变量赋值
|
||
key = re.findall(r'\{([^}]+)}', code)[0] # 查找变量名
|
||
code = re.sub(r'\{' + key + '}', fr'{values[key]}', code)
|
||
self.codes[index] = code # 给变量赋值
|
||
if re.search(r'\n', code): # 是多行代码
|
||
head = re.findall(r'(.*?)\n', code)[0]
|
||
if head == 'python':
|
||
pass
|
||
elif head == 'shell':
|
||
pass
|
||
elif re.match(r'\$[^$]*\$', code): # 是LaTex代码
|
||
self.codes[index] = re.sub(fr'\$([^$]*)\$', r'\(\1\)', code)
|
||
|
||
def restore(self, new_text: str):
|
||
"""
|
||
将渲染好的代码重新放回处理好的正文
|
||
:param new_text: 处理好的正文
|
||
:return: 加上代码的文章
|
||
"""
|
||
for index, item in enumerate(self.codes):
|
||
new_text = re.sub(fr'-@@-{index}-@@-', f'{item}', new_text, flags=re.DOTALL)
|
||
return new_text
|
||
|
||
|
||
class Basic:
|
||
def __init__(self, text: str):
|
||
self.text: str = text
|
||
|
||
def paragraph(self):
|
||
"""
|
||
为普通的行套上段落标签
|
||
"""
|
||
self.text = re.sub(r'<p>(<.+?>.*?<.+?>)</p>\n',
|
||
r'\1\n', # 移除已被标签包裹的行的额外的<p>标签
|
||
'\n'.join(
|
||
[
|
||
f'<p>{line}</p>' for line in self.text.splitlines() if
|
||
not re.search( # 把所有非空的行都套上<p>标签
|
||
r'^\s*\n?$', line # 识别空行或空白行
|
||
)
|
||
]
|
||
)
|
||
)
|
||
|
||
def __call__(self, *args, **kwargs):
|
||
self.paragraph()
|
||
return self.text
|
||
|
||
|
||
def add_indent_to_string(input_string: str, indent_spaces: int = 4):
|
||
"""
|
||
给字符串中的每一行前面加上缩进。
|
||
:param input_string: 原始字符串，可以包含多行。
|
||
:param indent_spaces: 每行前面要添加的空格数，默认为4。
|
||
|
||
:return: 带缩进的新字符串。
|
||
"""
|
||
# 使用字符串的splitlines()方法分割原始字符串为行列表
|
||
lines = input_string.splitlines()
|
||
|
||
# 遍历行列表，给每行前面加上相应的缩进，并重新组合成字符串
|
||
indented_string = "\n".join(f"{' ' * indent_spaces}{line}" for line in lines)
|
||
|
||
return indented_string
|
||
|
||
|
||
def body(text: str) -> Tuple[str, Dict[str, str]]:
|
||
"""
|
||
渲染正文部分
|
||
:param text: 输入正文
|
||
:return: 输出渲染后的正文
|
||
"""
|
||
Escape(text)
|
||
text, values = Value(text)() # 提取变量并赋值到文本中
|
||
text = Header(text)() # 渲染标题
|
||
text = Style(text)() # 渲染字体样式
|
||
text = Function(text)() # 渲染特殊功能
|
||
text = Basic(text)() # 渲染基础格式
|
||
|
||
# text = Basic.paragraph(text) # 渲染段落
|
||
return text, values
|
||
|
||
|
||
def main(origen: str):
|
||
# 预处理
|
||
code_block = CodeBlock(origen) # 获取代码内容
|
||
text = code_block() # 暂时移除代码
|
||
text, values = body(text) # 处理正文
|
||
code_block.rendering(values) # 渲染代码
|
||
return code_block.restore(text) # 放回代码
|
||
|
||
|
||
if __name__ == '__main__':
|
||
with open('test.md', encoding='utf-8') as test:
|
||
cd = main(test.read())
|
||
with open('test.html', 'w', encoding='utf-8') as html:
|
||
html.write(f"""<!DOCTYPE html>
|
||
<html lang="zh-CN">
|
||
<head>
|
||
<meta charset="UTF-8">
|
||
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
||
<title>UTF-8编码示例</title>
|
||
<script type="text/javascript" async
|
||
src="https://cdnjs.cloudflare.com/ajax/libs/mathjax/2.7.7/MathJax.js?config=TeX-MML-AM_CHTML">
|
||
</script>
|
||
<!-- 可以在这里添加其他元数据和CSS链接 -->
|
||
</head>
|
||
<body>
|
||
{add_indent_to_string(cd, 4)}
|
||
</body>
|
||
</html>
|
||
""")
|