X-Git-Url: https://git.madduck.net/etc/vim.git/blobdiff_plain/6e97c5f47cbec72c72c27aefb206589dd84707a7..f87df0e3c8735de416b6392ce7f21c6ba194424d:/src/black/mode.py

diff --git a/src/black/mode.py b/src/black/mode.py
index b6d1a1f..35a072c 100644
--- a/src/black/mode.py
+++ b/src/black/mode.py
@@ -4,6 +4,7 @@ Mostly around Python language feature support per version and Black configuratio
 chosen by the user.
 """
 
+from hashlib import sha256
 import sys
 
 from dataclasses import dataclass, field
@@ -126,7 +127,6 @@ class Preview(Enum):
     """Individual preview style features."""
 
     string_processing = auto()
-    hug_simple_powers = auto()
 
 
 class Deprecated(UserWarning):
@@ -142,6 +142,7 @@ class Mode:
     is_ipynb: bool = False
     magic_trailing_comma: bool = True
     experimental_string_processing: bool = False
+    python_cell_magics: Set[str] = field(default_factory=set)
     preview: bool = False
 
     def __post_init__(self) -> None:
@@ -161,7 +162,9 @@ class Mode:
         """
         if feature is Preview.string_processing:
             return self.preview or self.experimental_string_processing
-        return self.preview
+        # TODO: Remove type ignore comment once preview contains more features
+        #  than just ESP
+        return self.preview  # type: ignore
 
     def get_cache_key(self) -> str:
         if self.target_versions:
@@ -180,5 +183,6 @@ class Mode:
             str(int(self.magic_trailing_comma)),
             str(int(self.experimental_string_processing)),
             str(int(self.preview)),
+            sha256((",".join(sorted(self.python_cell_magics))).encode()).hexdigest(),
         ]
         return ".".join(parts)