From fb454572b2e455b22a59822b53b203b7232570e6 Mon Sep 17 00:00:00 2001
From: Wu Zhenyu <wuzy01@qq.com>
Date: Sun, 28 Aug 2022 17:29:25 +0800
Subject: [PATCH] Fix `pip completion --zsh`

---
 news/11416.bugfix.rst                    |  1 +
 src/pip/_internal/commands/completion.py | 13 ++++---------
 tests/functional/test_completion.py      | 16 ++++++----------
 3 files changed, 11 insertions(+), 19 deletions(-)
 create mode 100644 news/11416.bugfix.rst

diff --git a/news/11416.bugfix.rst b/news/11416.bugfix.rst
new file mode 100644
index 00000000000..3815b2da864
--- /dev/null
+++ b/news/11416.bugfix.rst
@@ -0,0 +1 @@
+Fix ``pip completion --zsh``.
diff --git a/src/pip/_internal/commands/completion.py b/src/pip/_internal/commands/completion.py
index deaa30899e6..30233fc7ad2 100644
--- a/src/pip/_internal/commands/completion.py
+++ b/src/pip/_internal/commands/completion.py
@@ -22,15 +22,10 @@
         complete -o default -F _pip_completion {prog}
     """,
     "zsh": """
-        function _pip_completion {{
-          local words cword
-          read -Ac words
-          read -cn cword
-          reply=( $( COMP_WORDS="$words[*]" \\
-                     COMP_CWORD=$(( cword-1 )) \\
-                     PIP_AUTO_COMPLETE=1 $words[1] 2>/dev/null ))
-        }}
-        compctl -K _pip_completion {prog}
+        #compdef -P pip[0-9.]#
+        compadd $( COMP_WORDS="$words[*]" \\
+                   COMP_CWORD=$((CURRENT-1)) \\
+                   PIP_AUTO_COMPLETE=1 $words[1] 2>/dev/null )
     """,
     "fish": """
         function __fish_complete_pip
diff --git a/tests/functional/test_completion.py b/tests/functional/test_completion.py
index b02cd4fa317..332dd120272 100644
--- a/tests/functional/test_completion.py
+++ b/tests/functional/test_completion.py
@@ -44,15 +44,10 @@
     (
         "zsh",
         """\
-function _pip_completion {
-  local words cword
-  read -Ac words
-  read -cn cword
-  reply=( $( COMP_WORDS="$words[*]" \\
-             COMP_CWORD=$(( cword-1 )) \\
-             PIP_AUTO_COMPLETE=1 $words[1] 2>/dev/null ))
-}
-compctl -K _pip_completion pip""",
+#compdef -P pip[0-9.]#
+compadd $( COMP_WORDS="$words[*]" \\
+           COMP_CWORD=$((CURRENT-1)) \\
+           PIP_AUTO_COMPLETE=1 $words[1] 2>/dev/null )""",
     ),
     (
         "powershell",
@@ -392,7 +387,8 @@ def test_completion_path_after_option(
     )
 
 
-@pytest.mark.parametrize("flag", ["--bash", "--zsh", "--fish", "--powershell"])
+# zsh completion script doesn't contain pip3
+@pytest.mark.parametrize("flag", ["--bash", "--fish", "--powershell"])
 def test_completion_uses_same_executable_name(
     autocomplete_script: PipTestEnvironment, flag: str, deprecated_python: bool
 ) -> None: