cores/runcode.py at main · sodeom/cores · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
import re
import textwrap
from utils import dbg
import utils
from coderunner import run_block

# -----------------------------
# Regex patterns for extraction
# -----------------------------
DO_BLOCK_RE = re.compile(r'do:\s*\{([\s\S]*?)\}', re.MULTILINE)
END_BLOCK_RE = re.compile(r'end:\s*\{([\s\S]*?)\}', re.MULTILINE)
TIMES_BLOCK_RE = re.compile(r'times:\s*\((\d+)\)\s*\{([\s\S]*?)\}', re.MULTILINE)

# -----------------------------
# Extraction functions
# -----------------------------
def extract_do_block(text):
    match = DO_BLOCK_RE.search(text)
    return textwrap.dedent(match.group(1)).strip() if match else None

def extract_end_block(text):
    match = END_BLOCK_RE.search(text)
    return textwrap.dedent(match.group(1)).strip() if match else None

def extract_times_block(text):
    match = TIMES_BLOCK_RE.search(text)
    if not match:
        return None, None
    return int(match.group(1)), textwrap.dedent(match.group(2)).strip()

# -----------------------------
# Main
# -----------------------------
def main(code):
    dbg(f"[Running] Do code")

    do_code = extract_do_block(code)
    run_block(do_code) if do_code else None

    dbg(f"[Ends] Do code\n")

    # Extract END block
    utils.END_CODE = extract_end_block(code)

    # Extract TIMES block
    times_number, times_code = extract_times_block(code)
    return times_number, times_code