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

dataframe.apply method returns a copy instead of operating in-place

This commit is contained in:
Darik Gamble 2015-01-19 12:38:55 -05:00
parent 35bccedc2f
commit 4a71c1f166
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
pgcli/pgcompleter.py
View File

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