56 lines
2.0 KiB
Python
56 lines
2.0 KiB
Python
import re
|
|
from typing import TYPE_CHECKING
|
|
from mdcollections.base_types import MDPage
|
|
|
|
def js_to_python(js_str):
|
|
"""Convert JavaScript object notation to Python dictionary syntax."""
|
|
# Remove any 'option =' prefix and trailing semicolon
|
|
js_str = re.sub(r'^option\s*=\s*', '', js_str)
|
|
js_str = re.sub(r';(\s*)$', '', js_str)
|
|
|
|
# Convert JavaScript property names to Python dictionary keys
|
|
js_str = re.sub(r'(\b\w+):', r'"\1":', js_str)
|
|
|
|
# Convert single quotes to double quotes for string values
|
|
# First, replace escaped single quotes with a placeholder
|
|
js_str = js_str.replace("\\'", "___ESCAPED_QUOTE___")
|
|
# Then replace regular single quotes with double quotes
|
|
js_str = js_str.replace("'", '"')
|
|
# Finally, restore escaped single quotes
|
|
js_str = js_str.replace("___ESCAPED_QUOTE___", "\\'")
|
|
|
|
# Handle trailing commas
|
|
js_str = re.sub(r',(\s*[}\]])', r'\1', js_str)
|
|
|
|
# Handle special JavaScript values
|
|
js_str = js_str.replace('true', 'True').replace('false', 'False').replace('null', 'None')
|
|
|
|
# Remove any comments
|
|
js_str = re.sub(r'//.*?\n|/\*.*?\*/', '', js_str, flags=re.DOTALL)
|
|
|
|
return js_str.strip()
|
|
|
|
def process_markdown_echarts(page: MDPage) -> MDPage:
|
|
"""Convert ```echarts blocks to ```py sl blocks that use st_echarts."""
|
|
if not isinstance(page, MDPage):
|
|
raise TypeError("page must be a MDPage")
|
|
|
|
def replace_echarts_block(match):
|
|
echarts_code = match.group(1).strip()
|
|
python_code = js_to_python(echarts_code)
|
|
|
|
# Create the streamlit code block
|
|
streamlit_code = f"""```py sl
|
|
from streamlit_echarts import st_echarts
|
|
option = {python_code}
|
|
st_echarts(options=option, height="400px")
|
|
```"""
|
|
return streamlit_code
|
|
|
|
# Process all echarts code blocks
|
|
processed_content = re.sub(r"```echarts\n(.*?)\n```", replace_echarts_block, page.content, flags=re.DOTALL)
|
|
|
|
page.content_ = processed_content
|
|
|
|
return page
|