servicelib-examples

from servicelib.process import Process

​

​

def mapd(context, inputFile, outputFormat, options):

    inp = context.get_data(inputFile)

    out = context.create_result(outputFormat)

​

    plan_file = context.create_temp_file()

​

    class p(Process):

        max_output_size = 10 * 1024 * 1024  # extend default of 10 KB of stdout

​

        def __init__(self):

            cmd = [

                "/usr/local/bin/mir",

                str(inp),

                str(out.path),

                "--dump-plan-file={}".format(plan_file),

            ]

            for k, v in options.items():

                if not isinstance(v, list):

                    v = [v]

                cmd.append("--{}={}".format(k, "/".join([str(x) for x in v])))

​

            env = {"MIR_DEBUG": "1", "MIR_CACHE_PATH": "/tmp/cache/mir"}

            super(p, self).__init__("mapd", cmd, env)

​

        def results(self):

            with open(plan_file, "rt") as f:

                out["plan"] = f.read()

            return out

​

    return context.spawn_process(p())

​

​

def main():

    from servicelib import service

​

    service.start_services(

        {"name": "mapd", "execute": mapd,}

    )