mirrors
/
pgcli
mirror of https://github.com/dbcli/pgcli
1
0
Fork 0
Code

Store escaped nameds in metadata

This commit is contained in:
Darik Gamble 2015-01-19 13:09:43 -05:00
parent 98b064b829
commit 77df20f4ca
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
pgcli/pgcompleter.py
View File

@ -85,7 +85,7 @@ class PGCompleter(Completer):
def extend_schemata(self, data):
# data is a DataFrame with columns [schema]
data['schema'] = data['schema'].apply(self.unescape_name)
data['schema'] = data['schema'].apply(self.escape_name)
self.schemata = self.schemata.append(data)
self.all_completions.update(data['schema'])
@ -93,7 +93,7 @@ class PGCompleter(Completer):
# data is a DataFrame with columns [schema, table, is_visible]
data[['schema', 'table']] = \
data[['schema', 'table']].apply(self.unescaped_names)
data[['schema', 'table']].apply(self.escaped_names)
self.tables = self.tables.append(data)
self.all_completions.update(data['schema'])
@ -108,7 +108,7 @@ class PGCompleter(Completer):
# data is a DataFrame with columns [schema, table, column]
data[['schema', 'table', 'column']] = \
data[['schema', 'table', 'column']].apply(self.unescaped_names)
data[['schema', 'table', 'column']].apply(self.escaped_names)
self.columns = self.columns.append(data)
self.all_completions.update(data.column)
@ -192,7 +192,7 @@ class PGCompleter(Completer):
columns = self.columns # dataframe with columns [schema, table, column]
scoped_tbls[['schema', 'table', 'alias']] = \
scoped_tbls[['schema', 'table', 'alias']].apply(self.unescaped_names)
scoped_tbls[['schema', 'table', 'alias']].apply(self.escaped_names)
# For fully qualified tables, inner join on (schema, table)
qualed = scoped_tbls.merge(columns, how='inner', on=['schema', 'table'])