diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2023-08-14 09:53:08 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2023-08-14 09:53:08 +0000 |
commit | 6d1cb3a9e9138ce198ec3e344ca2d0f865e0c762 (patch) | |
tree | 8cef3c436dc2a3c6301c5b61bc5d8f1362ee918e /mycli/completion_refresher.py | |
parent | Adding upstream version 1.26.1. (diff) | |
download | mycli-upstream/1.27.0.tar.xz mycli-upstream/1.27.0.zip |
Adding upstream version 1.27.0.upstream/1.27.0
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'mycli/completion_refresher.py')
-rw-r--r-- | mycli/completion_refresher.py | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/mycli/completion_refresher.py b/mycli/completion_refresher.py index 8eb3de9..5d5f40f 100644 --- a/mycli/completion_refresher.py +++ b/mycli/completion_refresher.py @@ -3,7 +3,7 @@ from .packages.special.main import COMMANDS from collections import OrderedDict from .sqlcompleter import SQLCompleter -from .sqlexecute import SQLExecute +from .sqlexecute import SQLExecute, ServerSpecies class CompletionRefresher(object): @@ -113,6 +113,8 @@ def refresh_users(completer, executor): @refresher('functions') def refresh_functions(completer, executor): completer.extend_functions(executor.functions()) + if executor.server_info.species == ServerSpecies.TiDB: + completer.extend_functions(completer.tidb_functions, builtin=True) @refresher('special_commands') def refresh_special(completer, executor): @@ -121,3 +123,8 @@ def refresh_special(completer, executor): @refresher('show_commands') def refresh_show_commands(completer, executor): completer.extend_show_items(executor.show_candidates()) + +@refresher('keywords') +def refresh_keywords(completer, executor): + if executor.server_info.species == ServerSpecies.TiDB: + completer.extend_keywords(completer.tidb_keywords, replace=True) |