feat: Get complex postgres argument components directly.

Author: DanCardin

src/sqlalchemy_declarative_extensions/dialects/mysql/function.py

@@ -109,7 +109,9 @@ def normalize(self) -> Function:
                     norm_type = type_map.get(type_str.lower(), type_str.lower())
                     normalized_parameters.append(f"{name} {norm_type}")
                 else:
-                    normalized_parameters.append(param) # Keep as is if format unexpected
+                    normalized_parameters.append(
+                        param
+                    )  # Keep as is if format unexpected
 
         return replace(
             self,

src/sqlalchemy_declarative_extensions/dialects/mysql/query.py

@@ -92,7 +92,7 @@ def get_functions_mysql(connection: Connection) -> Sequence[BaseFunction]:
     functions = []
     for f in connection.execute(functions_query, {"schema": database}).fetchall():
         parameters = None
-        if f.parameters: # Parameter string might be None if no parameters
+        if f.parameters:  # Parameter string might be None if no parameters
             parameters = [p.strip() for p in f.parameters.split(",")]
 
         functions.append(

src/sqlalchemy_declarative_extensions/dialects/mysql/schema.py

@@ -87,7 +87,9 @@
     select(
         column("SPECIFIC_NAME").label("routine_name"),
         func.group_concat(
-            text("concat(PARAMETER_NAME, ' ', DTD_IDENTIFIER) ORDER BY ORDINAL_POSITION SEPARATOR ', '"),
+            text(
+                "concat(PARAMETER_NAME, ' ', DTD_IDENTIFIER) ORDER BY ORDINAL_POSITION SEPARATOR ', '"
+            ),
         ).label("parameters"),
     )
     .select_from(table("PARAMETERS", schema="INFORMATION_SCHEMA"))
@@ -107,7 +109,7 @@
         routine_table.c.sql_data_access.label("data_access"),
         parameters_subquery.c.parameters.label("parameters"),
     )
-    .select_from( # Join routines with the parameter subquery
+    .select_from(  # Join routines with the parameter subquery
         routine_table.outerjoin(
             parameters_subquery,
             routine_table.c.routine_name == parameters_subquery.c.routine_name,

src/sqlalchemy_declarative_extensions/dialects/postgresql/__init__.py

@@ -1,5 +1,7 @@
 from sqlalchemy_declarative_extensions.dialects.postgresql.function import (
     Function,
+    FunctionParam,
+    FunctionReturn,
     FunctionSecurity,
     FunctionVolatility,
 )
@@ -43,6 +45,8 @@
     "DefaultGrantTypes",
     "Function",
     "FunctionGrants",
+    "FunctionParam",
+    "FunctionReturn",
     "FunctionSecurity",
     "FunctionVolatility",
     "Grant",