fix: improve f-strings even more (#10)

jvllmr · pre-commit-ci[bot] · web-flow · commit bec6cbed8773 · 2025-01-18T01:55:51.000+01:00
* fix: improve f-strings even more * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
diff --git a/src/unparser.rs b/src/unparser.rs
@@ -1035,10 +1035,18 @@ impl Unparser {
 
             let has_single = escaped_source.contains("'");
             let has_double = escaped_source.contains("\"");
-            let has_single_doc = escaped_source.contains("'''");
-            if has_single && has_double && has_single_doc {
+
+            if has_single
+                && has_double
+                && escaped_source.starts_with("\"")
+                && escaped_source.ends_with("\"")
+            {
                 escaped_source = replace_first_and_last(&escaped_source, "\"\"\"")
-            } else if has_single && has_double {
+            } else if has_single
+                && has_double
+                && escaped_source.starts_with("'")
+                && escaped_source.ends_with("'")
+            {
                 escaped_source = replace_first_and_last(&escaped_source, "'''")
             } else if has_single {
                 escaped_source = replace_first_and_last(&escaped_source, "\"")
diff --git a/test_files/f_string.py b/test_files/f_string.py
@@ -19,4 +19,7 @@
         for (k, v) in attribs.items() # noqa
     )
 )
+
+completion_init_lines = [f"source '{completion_path}'"] # noqa
+
 # fmt: on

Original file line number	Diff line number	Diff line change
`@@ -19,4 +19,7 @@`
`19`	`19`	`for (k, v) in attribs.items() # noqa`
`20`	`20`	`)`
`21`	`21`	`)`
	`22`	`+`
	`23`	`+completion_init_lines = [f"source '{completion_path}'"] # noqa`
	`24`	`+`
`22`	`25`	`# fmt: on`