add parser
parent
851b46779c
commit
d0b997e181
@ -0,0 +1,14 @@
|
|||||||
|
import re
|
||||||
|
from typing import Optional
|
||||||
|
|
||||||
|
|
||||||
|
def parse_code_block(string: str, lang: str) -> Optional[str]:
|
||||||
|
code_pattern = fr"```{lang}\n(.*?)\n```"
|
||||||
|
match = re.search(code_pattern, string, re.DOTALL)
|
||||||
|
if match:
|
||||||
|
return match.group(1)
|
||||||
|
else:
|
||||||
|
return None
|
||||||
|
|
||||||
|
def add_code_block(string: str, lang: str) -> str:
|
||||||
|
return f"```{lang}\n{string}\n```"
|
Loading…
Reference in New Issue