That's very easy by using the minted package, and even in Debian, it is installed by default:

    \begin{minted}{python}
    def square(x):
        return x * x
    \end{minted}

Does not look difficult to me.