llvm/llvm/test/Transforms/Inline/ML/Inputs/interactive_main.py

import interactive_host
import sys


def main(args):
    class Advisor:
        to_return = False

        def advice(self, _):
            # The adice will be a sequence of yes/no/yes/no/...
            # see ../interactive-mode.ll
            self.to_return = not self.to_return
            return int(self.to_return)

    a = Advisor()
    interactive_host.run_interactive(args[0], a.advice, args[1:])


if __name__ == "__main__":
    main(sys.argv[1:])