From 4b356804092dd102b5cb2308b46774c30529940e Mon Sep 17 00:00:00 2001 From: Elijah Date: Fri, 4 Oct 2024 17:21:42 +0000 Subject: [PATCH] minor refactor --- reflex/components/datadisplay/shiki_code_block.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/reflex/components/datadisplay/shiki_code_block.py b/reflex/components/datadisplay/shiki_code_block.py index f34bb51c6..0f95d1b54 100644 --- a/reflex/components/datadisplay/shiki_code_block.py +++ b/reflex/components/datadisplay/shiki_code_block.py @@ -779,10 +779,11 @@ class ShikiHighLevelCodeBlock(ShikiCodeBlock): raise VarTypeError( f"code should be string literal or a StringVar type. Got {type(code)} instead." ) + regex_pattern = r"[\/#]+ *\[!code.*?\]" if isinstance(code, Var): return string_replace_operation( - code, StringVar(_js_expr=r"/[\/#]+ *\[!code.*?\]/g", _var_type=str), "" + code, StringVar(_js_expr=f"/{regex_pattern}/g", _var_type=str), "" ) if isinstance(code, str): return re.sub(r"[\/#]+ *\[!code.*?\]", "", code)