Matrix-Mirrors/pantalaimon
1
0
Fork 0
mirror of https://github.com/matrix-org/pantalaimon.git synced 2025-02-12 04:41:26 -05:00
Code Issues Packages Projects Releases Wiki Activity

Prepare for prompt-toolkit 3.0

Browse Source
This commit is contained in:
Denis Kasak 2019-12-05 20:18:13 +01:00
parent 3a77e107fc
commit 57e08221de
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
pantalaimon/panctl.py
View File

@ -24,14 +24,18 @@ from typing import List
import attr import attr
import click import click
from gi.repository import GLib from gi.repository import GLib
from prompt_toolkit import __version__ as ptk_version
from prompt_toolkit import HTML, PromptSession, print_formatted_text from prompt_toolkit import HTML, PromptSession, print_formatted_text
from prompt_toolkit.completion import Completer, Completion, PathCompleter from prompt_toolkit.completion import Completer, Completion, PathCompleter
from prompt_toolkit.document import Document from prompt_toolkit.document import Document
from prompt_toolkit.eventloop.defaults import use_asyncio_event_loop
from prompt_toolkit.patch_stdout import patch_stdout from prompt_toolkit.patch_stdout import patch_stdout
from pydbus import SessionBus from pydbus import SessionBus
use_asyncio_event_loop() PTK2 = ptk_version.startswith('2.')
if PTK2:
from prompt_toolkit.eventloop.defaults import use_asyncio_event_loop
use_asyncio_event_loop()
class ParseError(Exception): class ParseError(Exception):
@ -568,7 +572,10 @@ class PanCtl:
while True: while True:
with patch_stdout(): with patch_stdout():
try: try:
result = await promptsession.prompt(async_=True) if PTK2:
result = await promptsession.prompt(async_=True)
else:
result = await promptsession.prompt_async()
except EOFError: except EOFError:
break break