diff options
| author | Brett <brett@nextglass.co> | 2015-06-20 07:46:03 -0400 |
|---|---|---|
| committer | Brett <brett@nextglass.co> | 2015-06-20 07:46:03 -0400 |
| commit | c1221af5442aa34ea3b437b2e53ba6f893f457e0 (patch) | |
| tree | 54798debc3d8912b845b818f8de6a805f2fd898a | |
| parent | 44528026d292737641ce8011b5096ade8f87a9a6 (diff) | |
add autocompletion for named queries
| -rw-r--r-- | pgcli/packages/sqlcompletion.py | 3 | ||||
| -rw-r--r-- | pgcli/pgcompleter.py | 6 |
2 files changed, 9 insertions, 0 deletions
diff --git a/pgcli/packages/sqlcompletion.py b/pgcli/packages/sqlcompletion.py index 21c08189..c622faf1 100644 --- a/pgcli/packages/sqlcompletion.py +++ b/pgcli/packages/sqlcompletion.py @@ -132,6 +132,9 @@ def suggest_special(text): return [{'type': 'schema'}, {'type': rel_type, 'schema': []}] + if cmd == '\\n': + return [{'type': 'namedquery'}] + return [{'type': 'keyword'}, {'type': 'special'}] diff --git a/pgcli/pgcompleter.py b/pgcli/pgcompleter.py index 55aa958c..e61ad96d 100644 --- a/pgcli/pgcompleter.py +++ b/pgcli/pgcompleter.py @@ -4,6 +4,7 @@ import re from prompt_toolkit.completion import Completer, Completion from .packages.sqlcompletion import suggest_type from .packages.parseutils import last_word +from .packages.pgspecial.namedqueries import namedqueries try: from collections import Counter @@ -334,6 +335,11 @@ class PGCompleter(Completer): fuzzy=False) completions.extend(types) + elif suggestion['type'] == 'namedquery': + queries = self.find_matches(word_before_cursor, namedqueries.list(), + start_only=True, fuzzy=False) + completions.extend(queries) + return completions def populate_scoped_cols(self, scoped_tbls): |