Moved putStrLn and getLine inside the IO class

io1a.py

class IO:
    def __init__(self, action, arg):
        self.action = action
        self.arg = arg

    def __rshift__(self, func):  # this is "bind" (>>)
        if self.action == "getLine":
            line = input()
            return func(line)
        elif self.action == "putStrLn":
            print(self.arg)      # always returns None
            return func(None)
        elif self.action == "return":
            return func(self.arg)
        else:
            raise RuntimeError("oops")

    @staticmethod
    def unit(v):
        return IO("return", v)

    @staticmethod
    def putStrLn(text):
        return IO("putStrLn", text)

IO.getLine = IO("getLine", None)

main = (
    IO.putStrLn("What's your name?") >> (lambda _:
        IO.getLine >> (lambda name:
            IO.putStrLn("Hello " + name)
        )
    )
)

main >> (lambda _: IO.unit(None))