diff options
| author | Georgy Frolov <gosha@fro.lv> | 2021-03-11 03:59:48 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2021-03-10 16:59:48 -0800 |
| commit | 8aa3f92666241692c0fe75832c62aebca5c6b865 (patch) | |
| tree | ccdacc1552b7c7532e133648180c2929dc1c475d | |
| parent | 469d22f2b6da815aa95b27c9342c3db9e37f9c3b (diff) | |
fix import (#1252)
| -rw-r--r-- | pgcli/pgtoolbar.py | 23 |
1 files changed, 13 insertions, 10 deletions
diff --git a/pgcli/pgtoolbar.py b/pgcli/pgtoolbar.py index 441a733d..41f903da 100644 --- a/pgcli/pgtoolbar.py +++ b/pgcli/pgtoolbar.py @@ -1,20 +1,23 @@ -from packaging.version import parse as parse_version +from pkg_resources import packaging import prompt_toolkit from prompt_toolkit.key_binding.vi_state import InputMode from prompt_toolkit.application import get_app +parse_version = packaging.version.parse + +vi_modes = { + InputMode.INSERT: "I", + InputMode.NAVIGATION: "N", + InputMode.REPLACE: "R", + InputMode.INSERT_MULTIPLE: "M", +} +if parse_version(prompt_toolkit.__version__) >= parse_version("3.0.6"): + vi_modes[InputMode.REPLACE_SINGLE] = "R" + def _get_vi_mode(): - modes = { - InputMode.INSERT: "I", - InputMode.NAVIGATION: "N", - InputMode.REPLACE: "R", - InputMode.INSERT_MULTIPLE: "M", - } - if parse_version(prompt_toolkit.__version__) >= parse_version("3.0.6"): - modes[InputMode.REPLACE_SINGLE] = "R" - return modes[get_app().vi_state.input_mode] + return vi_modes[get_app().vi_state.input_mode] def create_toolbar_tokens_func(pgcli): |