Function extract_code

scrapegraphai/utils/cleanup_code.py:8–16

Module for extracting code

(code: str)

def extract_code(code: str) -> str:
    """
    Module for extracting code
    """
    pattern = r"```(?:python)?\n(.*?)```"

    match = re.search(pattern, code, re.DOTALL)

    return match.group(1) if match else code