path: root/mushi.py

#!/usr/bin/env python3

import argparse
import socket
import code
import readline
import signal
import sys

def SigHandler_SIGINT(signum, frame):
    print()
    sys.exit(0)

def SigHandler_Timeout(signum, frame):
    print("mushi timed out.")
    raise Exception("time out")

class Argparser(object):
    def __init__(self):
        self.parser = argparse.ArgumentParser()
        self.parser.add_argument("--bool", action="store_true", help="bool", default=False)
        self.parser.add_argument("--dbg", action="store_true", help="debug", default=False)

    def parse(self, args):
        self.args, self.rest = self.parser.parse_known_args(args)

# write code here
def premain(argparser):
    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    server_address = ("", 10111)
    sock.connect(server_address)
    msg = " ".join(argparser.rest)
    sock.sendall(bytes(msg, "utf-8"))
    output = sock.recv(4096)
    out2 = [c for c in output.decode("utf-8") if c != "\0"]
    print("".join(out2))
    sock.close()

def main(argv):
    argparser = Argparser()
    argparser.parse(argv)
    if argparser.args.dbg:
        try:
            premain(argparser)
        except:
            variables = globals().copy()
            variables.update(locals())
            shell = code.InteractiveConsole(variables)
            shell.interact(banner="DEBUG REPL")
    else:
        premain(argparser)

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