preproc.py

from typing import List, Tuple, Dict, Any

from exp import CONSTR_DICT
from exp.utility import read_dataset, first_available


def fmt(text, wchar, *attrs) -> str:
    param, fmt_str = (wchar, lambda v: f'{wchar}[{attrs.index(v)}]')
    for t in sorted(attrs, key=len, reverse=True):
        text = text.replace(t, fmt_str(t))
    return f'lambda {param}: {text}'.strip()


def pred_convert(
        imm: List[str], pred: List[str], attr: List[str]
) -> Tuple[CONSTR_DICT, Dict[str, Any]]:
    """Convert text constraints to executable expressions"""
    imm = [(k, ((k,), False)) for k in map(attr.index, imm or [])]
    fd, mut, wchar = {}, {}, first_available(attr)
    for p in (pred or []):
        s = [t for t in attr if t in p]
        a = first_available(list(fd.keys()))
        v = fmt(p, wchar, *s)
        sources = list(map(attr.index, s))
        mut[a] = (tuple(sources), eval(v),)
        fd[a] = {'attrs': dict(zip(sources, s)), 'exec': v, 'text': p}
    return {**dict(imm), **dict(mut)}, fd


def pred_parse(config):
    attrs = read_dataset(config['dataset'])[0]
    config['constraints'], config['p_config'] = pred_convert(
        config['constraints']['immutable'],
        config['constraints']['predicates'], attrs)
    return config