mirrors/pgcli
1
0
Fork 0
mirror of https://github.com/dbcli/pgcli synced 2024-05-31 01:17:54 +00:00
Code

Add logging to pgcompleter.

Browse Source
This commit is contained in:
Amjith Ramanujam 2015-01-04 22:06:42 -08:00
parent e5024c8869
commit cfa026df5d
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
pgcli/pgcompleter.py
View File

@ -1,9 +1,12 @@
from __future__ import print_function
import logging
from collections import defaultdict
from prompt_toolkit.completion import Completer, Completion
from .packages.sqlcompletion import suggest_type
from .packages.parseutils import last_word
_logger = logging.getLogger(__name__)
class PGCompleter(Completer):
keywords = ['ACCESS', 'ADD', 'ALL', 'ALTER TABLE', 'AND', 'ANY', 'AS',
'ASC', 'AUDIT', 'BETWEEN', 'BY', 'CASE', 'CHAR', 'CHECK',
@ -89,17 +92,23 @@ class PGCompleter(Completer):
document.text_before_cursor)
if category == 'columns':
_logger.debug("Completion: 'columns' Scope: %r", scope)
scoped_cols = self.populate_scoped_cols(scope)
return self.find_matches(word_before_cursor, scoped_cols)
elif category == 'columns-and-functions':
_logger.debug("Completion: 'columns-and-functions' Scope: %r",
scope)
scoped_cols = self.populate_scoped_cols(scope)
return self.find_matches(word_before_cursor, scoped_cols +
self.functions)
elif category == 'tables':
_logger.debug("Completion: 'tables' Scope: %r", scope)
return self.find_matches(word_before_cursor, self.tables)
elif category == 'databases':
_logger.debug("Completion: 'databases' Scope: %r", scope)
return self.find_matches(word_before_cursor, self.databases)
elif category == 'keywords':
_logger.debug("Completion: 'keywords' Scope: %r", scope)
return self.find_matches(word_before_cursor, self.keywords +
self.special_commands)
@ -107,5 +116,4 @@ class PGCompleter(Completer):
scoped_cols = []
for table in tables:
scoped_cols.extend(self.columns[table])
return scoped_cols