\df suggests only schemas and functions

2 files changed, 12 insertions, 0 deletions

diff --git a/pgcli/packages/sqlcompletion.py b/pgcli/packages/sqlcompletion.py
index 5185c8be..91bd616d 100644
--- a/pgcli/packages/sqlcompletion.py
+++ b/pgcli/packages/sqlcompletion.py
@@ -103,6 +103,11 @@ def suggest_special(text):
             return [{'type': 'table', 'schema': schema}]
         else:
             return [{'type': 'schema'}, {'type': 'table', 'schema': []}]
+    elif cmd[1:] == 'df':
+        if schema:
+            return [{'type': 'function', 'schema': schema}]
+        else:
+            return [{'type': 'schema'}, {'type': 'function', 'schema': []}]
 
     return [{'type': 'keyword'}, {'type': 'special'}]
 
diff --git a/tests/test_pgspecial.py b/tests/test_pgspecial.py
index 9d2834bc..e16631a2 100644
--- a/tests/test_pgspecial.py
+++ b/tests/test_pgspecial.py
@@ -27,3 +27,10 @@ def test_d_dot_suggests_schema_qualified_tables():
     suggestions = suggest_type('\d myschema.xxx', '\d myschema.xxx')
     assert suggestions == [{'type': 'table', 'schema': 'myschema'}]
 
+def test_df_suggests_schema_or_function():
+    suggestions = suggest_type('\\df xxx', '\\df xxx')
+    assert sorted_dicts(suggestions) == sorted_dicts([
+        {'type': 'function', 'schema': []}, {'type': 'schema'}])
+
+    suggestions = suggest_type('\\df myschema.xxx', '\\df myschema.xxx')
+    assert suggestions == [{'type': 'function', 'schema': 'myschema'}]