Function MasterConsole

Master.py:243–342

()

MasterPrompt = "\033[1;4;37mMaster\033[0m > "
ConsolePrompt = "\033[1;4;37mMaster\033[0m > "
async def MasterConsole():
    while True:
        #Python 3.7 3.8 
        #print(ConsolePrompt, end="")
        #console_cmd = await asyncio.get_running_loop().run_in_executor(None, input)

        #Python 3.9 3.10 3.11 ......
        if Puppet_Master.current_session:
            ConsolePrompt = f"(\033[1;4;34m{Puppet_Master.current_session['hash'][:8]}\033[0m) > "
        else:
            ConsolePrompt = MasterPrompt
        console_cmd = await asyncio.to_thread(input, ConsolePrompt)
        console_cmd = console_cmd.strip()

        if console_cmd == "exit":
            for handler in Puppet_Master.handlers:
                handler.close()
            break

        elif console_cmd.split() and console_cmd.split()[0] == "sessions" :
            print(
                    "Sessions {}".format(len(Puppet_Master.sessions)).ljust(32," ") + " " +
                    "Hostname".ljust(20," ") + " " +
                    "Username".ljust(8," ") + " " +
                    "CPU "
                    )
            print(
                    "="*32 + " " +
                    "="*20 + " " +
                    "="*8 + " " +
                    "="*3  + " " 
                    )
            for session in Puppet_Master.sessions:
                print(f"\033[1;37m{session['hash']}\033[0m {session['hostname'].ljust(20,' ')} {session['username'].ljust(8,' ')} {session['core'].ljust(3,' ')} {session['sockname']} -> {session['peername'].ljust(21,' ')} {session['inittime']} {session['org']} ")
            
        elif console_cmd.split() and console_cmd.split()[0] == "listerner":
            print()

        elif console_cmd.split() and console_cmd.split()[0] == "execute"  :
            if Puppet_Master.current_session or "-i" in console_cmd.split():
                #备注：需要优先处理-i后的会话,暂未实现
                if "-c" in console_cmd.split():
                    execute_cmd = console_cmd.split()[console_cmd.split().index("-c") +1]

                elif len(console_cmd.split()) > 1 :
                    execute_cmd = " ".join( console_cmd.split()[console_cmd.split().index("execute")+1:] )

                execute_result = await Puppet_Master.execute_cmd( execute_cmd )

        elif console_cmd.split() and console_cmd.split()[0] == "BATCH-EXECUTE"  :
                if len(console_cmd.split()) > 1 :
                    execute_cmd = " ".join( console_cmd.split()[console_cmd.split().index("BATCH-EXECUTE")+1:] )

                for session in Puppet_Master.sessions:
                    Puppet_Master.current_session = Puppet_Master.sessions[Puppet_Master.sessions.index(session)]
                    execute_result = await Puppet_Master.execute_cmd( execute_cmd )
                    PrintInfo( f"{Puppet_Master.current_session['hash']} {Puppet_Master.current_session['peername']} {Puppet_Master.current_session['org']} Executed." )