diff --git a/programming_runs/generators/parse.py b/programming_runs/generators/parse.py new file mode 100644 index 0000000..926b6e5 --- /dev/null +++ b/programming_runs/generators/parse.py @@ -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```" \ No newline at end of file