完善共享库的使用方法 about code-runner HOT 5 CLOSED

from ctypes import *

libjudge = cdll.LoadLibrary("./libjudge.so")

# 自定义基类结构体
class myStructure(Structure):
    # 结构体转字典
    def dump_dict(self):
        info = {}
        # 通过_fields_获取每一个字段
        # 检查每个字段的类型，根据不同类型分别处理
        # 支持递归迭代
        for k, v in self._fields_:
            av = getattr(self, k)
            if type(v) == type(Structure):
                av = av.dump_dict()
            elif type(v) == type(Array):
                av = cast(av, c_char_p).value.decode()
            else:
                pass
            info[k] = av
        return info

    # 字符串转换
    def __str__(self):
        info = self.dump_dict()
        return repr(info)

    # 打印信息
    def show(self):
        from pprint import pprint

        pprint(self.dump_dict())


class Result(myStructure):
    _field = [
        ("status", c_int),
        ("cpu_time_used", c_int),
        ("cpu_time_used_us", c_long),
        ("real_time_used", c_int),
        ("real_time_used_us", c_long),
        ("memory_used", c_int),
        ("signal", c_int),
        ("exit_code", c_int),
    ]


class Config(myStructure):
    _fields_ = [
        ("run_mode", c_int),
        ("check_mode", c_int),
        ("judge_mode", c_int),
        ("cmd", POINTER(c_char_p)),
        ("log_file", c_char_p),
        ("time_limit", c_int),
        ("memory_limit", c_int),
        ("in_file", c_char_p),
        ("out_file", c_char_p),
        ("user_out_file", c_char_p),
    ]


judge = libjudge.judge
judge.restype = None
judge.argtypes = [POINTER(Config)]
cmd = (c_char_p * 2)(b"./main", b"1234234")

config = Config(
    judge_mode=1,
    cmd=pointer(cmd),
    log_file=b"1.log",
    time_limit=1000,
    memory_limit=2048,
    in_file=b"./tests/1/1.in",
    out_file=b"./tests/1/1.out",
    user_out_file=b"1.tmp.out",
)
config.show()
judge(byref(config))