From ecbd9e8cf71f13068c7e6803a534e00363114c91 Mon Sep 17 00:00:00 2001
From: Shantanu <12621235+hauntsaninja@users.noreply.github.com>
Date: Mon, 6 Nov 2023 16:58:43 -0800
Subject: [PATCH] Fix crash with f-string docstrings (#4019)

Python does not consider f-strings to be docstrings, so we probably
shouldn't be formatting them as such

Fixes #4018

Co-authored-by: Alex Waygood <Alex.Waygood@Gmail.com>
---
 CHANGES.md                                    |  3 +++
 src/black/nodes.py                            |  2 +-
 tests/data/cases/docstring_preview.py         |  3 ++-
 tests/data/cases/f_docstring.py               | 20 +++++++++++++++++++
 ...view_docstring_no_string_normalization.py} |  0
 5 files changed, 26 insertions(+), 2 deletions(-)
 create mode 100644 tests/data/cases/f_docstring.py
 rename tests/data/cases/{docstring_preview_no_string_normalization.py => preview_docstring_no_string_normalization.py} (100%)

diff --git a/CHANGES.md b/CHANGES.md
index a68f87bfc12..b1fe25ef625 100644
--- a/CHANGES.md
+++ b/CHANGES.md
@@ -13,6 +13,9 @@
 
 - Fix crash on formatting code like `await (a ** b)` (#3994)
 
+- No longer treat leading f-strings as docstrings. This matches Python's behaviour and
+  fixes a crash (#4019)
+
 ### Preview style
 
 - Multiline dictionaries and lists that are the sole argument to a function are now
diff --git a/src/black/nodes.py b/src/black/nodes.py
index 5f6b280c035..fff8e05a118 100644
--- a/src/black/nodes.py
+++ b/src/black/nodes.py
@@ -529,7 +529,7 @@ def is_docstring(leaf: Leaf) -> bool:
         return False
 
     prefix = get_string_prefix(leaf.value)
-    if "b" in prefix or "B" in prefix:
+    if set(prefix).intersection("bBfF"):
         return False
 
     if prev_siblings_are(
diff --git a/tests/data/cases/docstring_preview.py b/tests/data/cases/docstring_preview.py
index ff4819acb67..a3c656be2f8 100644
--- a/tests/data/cases/docstring_preview.py
+++ b/tests/data/cases/docstring_preview.py
@@ -58,7 +58,8 @@ def docstring_almost_at_line_limit():
 
 
 def docstring_almost_at_line_limit_with_prefix():
-    f"""long docstring................................................................"""
+    f"""long docstring................................................................
+    """
 
 
 def mulitline_docstring_almost_at_line_limit():
diff --git a/tests/data/cases/f_docstring.py b/tests/data/cases/f_docstring.py
new file mode 100644
index 00000000000..667f550b353
--- /dev/null
+++ b/tests/data/cases/f_docstring.py
@@ -0,0 +1,20 @@
+def foo(e):
+    f""" {'.'.join(e)}"""
+
+def bar(e):
+    f"{'.'.join(e)}"
+
+def baz(e):
+    F""" {'.'.join(e)}"""
+
+# output
+def foo(e):
+    f""" {'.'.join(e)}"""
+
+
+def bar(e):
+    f"{'.'.join(e)}"
+
+
+def baz(e):
+    f""" {'.'.join(e)}"""
diff --git a/tests/data/cases/docstring_preview_no_string_normalization.py b/tests/data/cases/preview_docstring_no_string_normalization.py
similarity index 100%
rename from tests/data/cases/docstring_preview_no_string_normalization.py
rename to tests/data/cases/preview_docstring_no_string_normalization.py