zsh: Add Python completion generation