fix_annotate_command.py

from __future__ import absolute_import, print_function

import json
import shlex
import subprocess

from .fix_annotate_json import BaseFixAnnotateFromSignature, FixAnnotateJson as _FixAnnotateJson

class FixAnnotateCommand(BaseFixAnnotateFromSignature):
    # run after FixAnnotateJson
    run_order = _FixAnnotateJson.run_order + 1

    command = None

    @classmethod
    def set_command(cls, command):
        cls.command = command

    def get_command(self, filename, lineno):
        return shlex.split(self.command.format(filename=filename, lineno=lineno))

    def get_types(self, node, results, funcname):
        cmd = self.get_command(self.filename, node.get_lineno())
        try:
            out = subprocess.check_output(cmd, stderr=subprocess.STDOUT)
        except subprocess.CalledProcessError as err:
            self.log_message("Line %d: Failed calling `%s`: %s" %
                             (node.get_lineno(), self.command,
                              err.output.rstrip()))
            return None

        data = json.loads(out)
        signature = data[0]['signature']
        return signature['arg_types'], signature['return_type']