Fixed telnet examples.

jonathanslenders · jonathanslenders · commit a1fa6f34d8e1 · 2019-06-10T11:51:33.000+02:00
diff --git a/examples/telnet/chat-app.py b/examples/telnet/chat-app.py
@@ -10,7 +10,7 @@
 
 from prompt_toolkit.contrib.telnet.server import TelnetServer
 from prompt_toolkit.formatted_text import HTML
-from prompt_toolkit.shortcuts import clear, prompt
+from prompt_toolkit.shortcuts import clear, prompt, PromptSession
 
 # Set up logging
 logging.basicConfig()
@@ -29,14 +29,15 @@
 
 async def interact(connection):
     write = connection.send
+    prompt_session = PromptSession()
 
     # When a client is connected, erase the screen from the client and say
     # Hello.
     clear()
     write('Welcome to our chat application!\n')
     write('All connected clients will receive what you say.\n')
 
-    name = await prompt(message='Type your name: ', async_=True)
+    name = await prompt_session.prompt_async(message='Type your name: ')
 
     # Random color.
     color = random.choice(COLORS)
@@ -53,7 +54,7 @@ async def interact(connection):
         # Set Application.
         while True:
             try:
-                result = await prompt(message=prompt_msg, async_=True)
+                result = await prompt_session.prompt_async(message=prompt_msg)
                 _send_to_everyone(connection, name, result, color)
             except KeyboardInterrupt:
                 pass
diff --git a/examples/telnet/toolbar.py b/examples/telnet/toolbar.py
@@ -8,7 +8,7 @@
 
 from prompt_toolkit.completion import WordCompleter
 from prompt_toolkit.contrib.telnet.server import TelnetServer
-from prompt_toolkit.shortcuts import prompt
+from prompt_toolkit.shortcuts import PromptSession
 
 # Set up logging
 logging.basicConfig()
@@ -26,11 +26,11 @@ async def interact(connection):
     def get_toolbar():
         return 'Bottom toolbar...'
 
-    result = await prompt(
+    session = PromptSession()
+    result = await session.prompt_async(
         'Say something: ',
         bottom_toolbar=get_toolbar,
-        completer=animal_completer,
-        async_=True)
+        completer=animal_completer)
 
     connection.send('You said: {}\n'.format(result))
     connection.send('Bye.\n')
