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