From 2d70ac0b33b448d5ef51c0856571068dd0754af6 Mon Sep 17 00:00:00 2001
From: Shantanu <12621235+hauntsaninja@users.noreply.github.com>
Date: Thu, 27 Oct 2022 15:04:28 -0700
Subject: [PATCH] Add hidden options to disable bytes promotion (#13952)

It might be useful to run mypy_primer without promotions in typeshed.
This would give us more confidence in changes stemming from
https://github.com/python/typeshed/issues/9001
---
 mypy/main.py                    |  6 ++++++
 mypy/options.py                 | 16 +++++++++++++---
 mypy/semanal_classprop.py       |  4 ++++
 test-data/unit/check-flags.test | 15 +++++++++++++++
 4 files changed, 38 insertions(+), 3 deletions(-)

diff --git a/mypy/main.py b/mypy/main.py
index 360a8ed1df17..405596c20991 100644
--- a/mypy/main.py
+++ b/mypy/main.py
@@ -1121,6 +1121,12 @@ def add_invertible_flag(
     parser.add_argument(
         "--enable-incomplete-features", action="store_true", help=argparse.SUPPRESS
     )
+    parser.add_argument(
+        "--disable-bytearray-promotion", action="store_true", help=argparse.SUPPRESS
+    )
+    parser.add_argument(
+        "--disable-memoryview-promotion", action="store_true", help=argparse.SUPPRESS
+    )
 
     # options specifying code to check
     code_group = parser.add_argument_group(
diff --git a/mypy/options.py b/mypy/options.py
index b89ad97708c1..3a08ff9455ee 100644
--- a/mypy/options.py
+++ b/mypy/options.py
@@ -56,9 +56,16 @@ class BuildType:
     "warn_unused_ignores",
 }
 
-OPTIONS_AFFECTING_CACHE: Final = (PER_MODULE_OPTIONS | {"platform", "bazel", "plugins"}) - {
-    "debug_cache"
-}
+OPTIONS_AFFECTING_CACHE: Final = (
+    PER_MODULE_OPTIONS
+    | {
+        "platform",
+        "bazel",
+        "plugins",
+        "disable_bytearray_promotion",
+        "disable_memoryview_promotion",
+    }
+) - {"debug_cache"}
 
 # Features that are currently incomplete/experimental
 TYPE_VAR_TUPLE: Final = "TypeVarTuple"
@@ -329,6 +336,9 @@ def __init__(self) -> None:
         # Deprecated reverse version of the above, do not use.
         self.enable_recursive_aliases = False
 
+        self.disable_bytearray_promotion = False
+        self.disable_memoryview_promotion = False
+
     # To avoid breaking plugin compatibility, keep providing new_semantic_analyzer
     @property
     def new_semantic_analyzer(self) -> bool:
diff --git a/mypy/semanal_classprop.py b/mypy/semanal_classprop.py
index b5a702592144..5d21babcc597 100644
--- a/mypy/semanal_classprop.py
+++ b/mypy/semanal_classprop.py
@@ -165,6 +165,10 @@ def add_type_promotion(
     if not promote_targets:
         if defn.fullname in TYPE_PROMOTIONS:
             target_sym = module_names.get(TYPE_PROMOTIONS[defn.fullname])
+            if defn.fullname == "builtins.bytearray" and options.disable_bytearray_promotion:
+                target_sym = None
+            elif defn.fullname == "builtins.memoryview" and options.disable_memoryview_promotion:
+                target_sym = None
             # With test stubs, the target may not exist.
             if target_sym:
                 target_info = target_sym.node
diff --git a/test-data/unit/check-flags.test b/test-data/unit/check-flags.test
index 03c2d1f38b82..5a075dd6efef 100644
--- a/test-data/unit/check-flags.test
+++ b/test-data/unit/check-flags.test
@@ -2128,3 +2128,18 @@ Ts = TypeVarTuple("Ts")  # E: "TypeVarTuple" support is experimental, use --enab
 from typing_extensions import TypeVarTuple
 Ts = TypeVarTuple("Ts")  # OK
 [builtins fixtures/tuple.pyi]
+
+
+[case testDisableBytearrayPromotion]
+# flags: --disable-bytearray-promotion
+def f(x: bytes) -> None: ...
+f(bytearray(b"asdf"))  # E: Argument 1 to "f" has incompatible type "bytearray"; expected "bytes"
+f(memoryview(b"asdf"))
+[builtins fixtures/primitives.pyi]
+
+[case testDisableMemoryviewPromotion]
+# flags: --disable-memoryview-promotion
+def f(x: bytes) -> None: ...
+f(bytearray(b"asdf"))
+f(memoryview(b"asdf"))  # E: Argument 1 to "f" has incompatible type "memoryview"; expected "bytes"
+[builtins fixtures/primitives.pyi]