PyCharm supports PL/Python language injections for PostgreSQL in CREATE FUNCTION and CREATE PROCEDURE operators.

Define plpythonu explicitly as the language of the function to have the Python code highlighted.

The injected Python code must be indented relative to the containing function.