From d0b997e181adcb8cafad87ce279893c9a103ea2f Mon Sep 17 00:00:00 2001 From: Noah Shinn Date: Thu, 3 Aug 2023 22:54:32 -0700 Subject: [PATCH] add parser --- programming_runs/generators/parse.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) create mode 100644 programming_runs/generators/parse.py 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