Function gen_message

tn-cli/client.py:93–194

Client message generator: reads user input as string, converts to pb.ClientMsg, and yields

(scheme, secret, args)

# Generator of protobuf messages.
def gen_message(scheme, secret, args):
    """Client message generator: reads user input as string,
    converts to pb.ClientMsg, and yields"""
    import random
    import threading
    from input_handler import stdin
    from commands import hiMsg, loginMsg, serialize_cmd

    random.seed()
    id = random.randint(10000,60000)

    # Asynchronous input-output
    tn_globals.InputThread = threading.Thread(target=stdin, args=(tn_globals.InputQueue,))
    tn_globals.InputThread.daemon = True
    tn_globals.InputThread.start()

    try:
        from importlib.metadata import version
    except ImportError:
        from importlib_metadata import version
    import platform

    APP_NAME = "tn-cli"
    APP_VERSION = "3.0.1"
    LIB_VERSION = version("tinode_grpc")
    GRPC_VERSION = version("grpcio")

    user_agent = APP_NAME + "/" + APP_VERSION + " (" + \
        platform.system() + "/" + platform.release() + "); gRPC-python/" + LIB_VERSION + "+" + GRPC_VERSION

    msg = hiMsg(id, args.background, user_agent, LIB_VERSION)
    if tn_globals.Verbose:
        stdoutln("\r=> " + to_json(msg))
    yield msg

    if scheme != None:
        id += 1
        login = lambda:None
        setattr(login, 'scheme', scheme)
        setattr(login, 'secret', secret)
        setattr(login, 'cred', None)
        msg = loginMsg(id, login, args)
        if tn_globals.Verbose:
            stdoutln("\r=> " + to_json(msg))
        yield msg

    print_prompt = True

    while True:
        try:
            if not tn_globals.WaitingFor and tn_globals.InputQueue:
                id += 1
                inp = tn_globals.InputQueue.popleft()

                if inp == 'exit' or inp == 'quit' or inp == '.exit' or inp == '.quit':
                    # Drain the output queue.
                    while pop_from_output_queue():
                        pass