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

Add \connect to the list of commands that need completion refresh.

Browse Source
This commit is contained in:
Amjith Ramanujam 2015-03-14 13:35:15 -07:00
parent 7e61c6ddfc
commit 5be22f57a6
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
pgcli/main.py
View File

@ -350,7 +350,8 @@ def format_output(cur, headers, status, table_format):
if is_expanded_output():
output.append(expanded_table(cur, headers))
else:
output.append(tabulate(cur, headers, tablefmt=table_format, missingval='<null>'))
output.append(tabulate(cur, headers, tablefmt=table_format,
missingval='<null>'))
if status: # Only print the status if it's not None.
output.append(status)
return output
@ -361,8 +362,8 @@ def need_completion_refresh(queries):
for query in sqlparse.split(queries):
try:
first_token = query.split()[0]
return first_token.lower() in ('alter', 'create', 'use', '\c',
'drop')
return first_token.lower() in ('alter', 'create', 'use', '\\c',
'\\connect', 'drop')
except Exception:
return False