show function definnition

Author: dkuku

dbcommands.sql

@@ -628,13 +628,13 @@ ORDER BY c.oid::pg_catalog.regclass::pg_catalog.text
 -- name:  show_function_definition
 -- docs: ("\\sf", "\\sf[+] FUNCNAME", "Show a function's definition.")
 SELECT
-    pg_catalog.pg_get_functiondef(
-        SELECT
-            coalesce(
-                get_coordinates::pg_catalog.regprocedure::pg_catalog.oid,
-                get_coordinates::pg_catalog.regproc::pg_catalog.oid)
-        ) AS source
-
+pg_catalog.pg_get_functiondef(
+    CASE
+    WHEN :pattern LIKE :bracket_wildcard
+    THEN :pattern::pg_catalog.regprocedure::pg_catalog.oid
+    ELSE :pattern::pg_catalog.regproc::pg_catalog.oid
+    END
+) AS source
 
 -- name:  list_foreign_tables
 -- docs: ("\\dE", "\\dE[+] [pattern]", "List foreign tables.", aliases=())

pgspecial/dbcommands.py

@@ -1607,19 +1607,10 @@ def rowcount(self):
 
 @special_command("\\sf", "\\sf[+] FUNCNAME", "Show a function's definition.")
 def show_function_definition(cur, pattern, verbose):
-    params = {"pattern": pattern}
-    if "(" in pattern:
-        sql = "SELECT %(pattern)s::pg_catalog.regprocedure::pg_catalog.oid"
-    else:
-        sql = "SELECT %(pattern)s::pg_catalog.regproc::pg_catalog.oid"
-    log.debug("%s, %s", sql, params)
-    cur.execute(sql, params)
-    (foid,) = cur.fetchone()
-
-    params = {"foid": foid}
-    sql = "SELECT pg_catalog.pg_get_functiondef(%(foid)s) as source"
-    log.debug("%s, %s", sql, params)
-    cur.execute(sql, params)
+    cur.execute(
+        queries.show_function_definition.sql,
+        {"pattern": pattern, "bracket_wildcard": "%(%"},
+    )
     if cur.description:
         headers = [x.name for x in cur.description]
         if verbose: