Internal change

PiperOrigin-RevId: 697698863

Author: AleksMat

patches/pyink.patch

@@ -780,16 +780,23 @@
              or self.bracket_tracker.any_open_for_or_lambda()
          ):
              if self.is_comment:
-@@ -262,7 +278,7 @@ class Line:
+@@ -262,7 +278,14 @@ class Line:
                      return True
          return False
 
 -    def contains_uncollapsable_type_comments(self) -> bool:
++    def contains_pragma_comments(self) -> bool:
++        for comments in self.comments.values():
++            for comment in comments:
++                if is_pragma_comment(comment, self.mode):
++                    return True
++        return False
++
 +    def contains_uncollapsable_pragma_comments(self) -> bool:
          ignored_ids = set()
          try:
              last_leaf = self.leaves[-1]
-@@ -287,11 +303,9 @@ class Line:
+@@ -287,11 +310,9 @@ class Line:
          comment_seen = False
          for leaf_id, comments in self.comments.items():
              for comment in comments:
@@ -804,7 +811,7 @@
                          return True
 
                  comment_seen = True
-@@ -326,7 +340,7 @@ class Line:
+@@ -326,7 +347,7 @@ class Line:
              # line.
              for node in self.leaves[-2:]:
                  for comment in self.comments.get(id(node), []):
@@ -813,7 +820,7 @@
                          return True
 
          return False
-@@ -481,7 +495,7 @@ class Line:
+@@ -481,7 +502,7 @@ class Line:
          if not self:
              return "\n"
 
@@ -822,7 +829,7 @@
          leaves = iter(self.leaves)
          first = next(leaves)
          res = f"{first.prefix}{indent}{first.value}"
-@@ -553,7 +567,7 @@ class EmptyLineTracker:
+@@ -553,7 +574,7 @@ class EmptyLineTracker:
          lines (two on module-level).
          """
          form_feed = (
@@ -831,7 +838,7 @@
              and bool(current_line.leaves)
              and "\f\n" in current_line.leaves[0].prefix
          )
-@@ -598,7 +612,7 @@ class EmptyLineTracker:
+@@ -598,7 +619,7 @@ class EmptyLineTracker:
 
      def _maybe_empty_lines(self, current_line: Line) -> tuple[int, int]:  # noqa: C901
          max_allowed = 1
@@ -840,7 +847,7 @@
              max_allowed = 1 if self.mode.is_pyi else 2
 
          if current_line.leaves:
-@@ -615,7 +629,7 @@ class EmptyLineTracker:
+@@ -615,7 +636,7 @@ class EmptyLineTracker:
 
          # Mutate self.previous_defs, remainder of this function should be pure
          previous_def = None
@@ -849,7 +856,7 @@
              previous_def = self.previous_defs.pop()
          if current_line.is_def or current_line.is_class:
              self.previous_defs.append(current_line)
-@@ -671,10 +685,25 @@ class EmptyLineTracker:
+@@ -671,10 +692,25 @@ class EmptyLineTracker:
              )
 
          if (
@@ -877,7 +884,7 @@
          ):
              return (before or 1), 0
 
-@@ -691,8 +720,9 @@ class EmptyLineTracker:
+@@ -691,8 +727,9 @@ class EmptyLineTracker:
                  return 0, 1
              return 0, 0
 
@@ -889,7 +896,7 @@
          ):
              if self.mode.is_pyi:
                  return 0, 0
-@@ -701,7 +731,7 @@ class EmptyLineTracker:
+@@ -701,7 +738,7 @@ class EmptyLineTracker:
          comment_to_add_newlines: Optional[LinesBlock] = None
          if (
              self.previous_line.is_comment
@@ -898,7 +905,7 @@
              and before == 0
          ):
              slc = self.semantic_leading_comment
-@@ -718,9 +748,9 @@ class EmptyLineTracker:
+@@ -718,9 +755,9 @@ class EmptyLineTracker:
 
          if self.mode.is_pyi:
              if current_line.is_class or self.previous_line.is_class:
@@ -910,7 +917,7 @@
                      newlines = 1
                  elif current_line.is_stub_class and self.previous_line.is_stub_class:
                      # No blank line between classes with an empty body
-@@ -749,7 +779,11 @@ class EmptyLineTracker:
+@@ -749,7 +786,11 @@ class EmptyLineTracker:
              newlines = 1 if current_line.depth else 2
              # If a user has left no space after a dummy implementation, don't insert
              # new lines. This is useful for instance for @overload or Protocols.
@@ -923,7 +930,7 @@
                  newlines = 0
          if comment_to_add_newlines is not None:
              previous_block = comment_to_add_newlines.previous_block
-@@ -1020,7 +1054,7 @@ def can_omit_invisible_parens(
+@@ -1020,7 +1061,7 @@ def can_omit_invisible_parens(
  def _can_omit_opening_paren(line: Line, *, first: Leaf, line_length: int) -> bool:
      """See `can_omit_invisible_parens`."""
      remainder = False
@@ -932,7 +939,7 @@
      _index = -1
      for _index, leaf, leaf_length in line.enumerate_with_length():
          if leaf.type in CLOSING_BRACKETS and leaf.opening_bracket is first:
-@@ -1044,7 +1078,7 @@ def _can_omit_opening_paren(line: Line,
+@@ -1044,7 +1085,7 @@ def _can_omit_opening_paren(line: Line,
 
  def _can_omit_closing_paren(line: Line, *, last: Leaf, line_length: int) -> bool:
      """See `can_omit_invisible_parens`."""

src/pyink/ink.py

@@ -4,11 +4,13 @@
 """
 
 from collections.abc import Collection, Iterator, Sequence
+import copy
 import re
-from typing import Optional, Union
+from typing import Any, Optional, Union
 
 from blib2to3.pgen2.token import ASYNC, FSTRING_START, NEWLINE, STRING
-from blib2to3.pytree import type_repr
+from blib2to3.pytree import NL, type_repr
+from pyink.lines import Line
 from pyink.mode import Quote
 from pyink.nodes import LN, Leaf, Node, STANDALONE_COMMENT, Visitor, syms
 from pyink.strings import STRING_PREFIX_CHARS
@@ -95,6 +97,37 @@ def _match_to_unicode(match: re.Match[str]) -> str:
     return re.sub(r"[<>&]", _match_to_unicode, src)
 
 
+def deepcopy_line(line: Line) -> Line:
+    """Calculates a deep copy of a Line object.
+
+    Deep-copying a Line object is not trivial because it contains various
+    dictionaries mapping id(NL) -> NL, where NL stands for Node or Leaf. Because
+    all objects are copied, also the ids in dictionaries need to be updated.
+
+    The function first finds all NL objects and calculates the id mapping. Then
+    it updates all dictionaries.
+
+    Args:
+      line: The Line object to copy.
+
+    Returns:
+      A deep copy of the Line object with updated references.
+    """
+    memo: dict[int, Any] = {}
+    line_copy = copy.deepcopy(line, memo=memo)
+
+    line_copy.comments = {
+        id(memo[leaf_id]): comment_leaves
+        for leaf_id, comment_leaves in line_copy.comments.items()
+    }
+    line_copy.bracket_tracker.delimiters = {
+        id(memo[leaf_id]): priority
+        for leaf_id, priority in line_copy.bracket_tracker.delimiters.items()
+    }
+
+    return line_copy
+
+
 def convert_unchanged_lines(src_node: Node, lines: Collection[tuple[int, int]]):
     """Converts unchanged lines to STANDALONE_COMMENT.
 

src/pyink/lines.py

@@ -278,6 +278,13 @@ def contains_implicit_multiline_string_with_comments(self) -> bool:
                     return True
         return False
 
+    def contains_pragma_comments(self) -> bool:
+        for comments in self.comments.values():
+            for comment in comments:
+                if is_pragma_comment(comment, self.mode):
+                    return True
+        return False
+
     def contains_uncollapsable_pragma_comments(self) -> bool:
         ignored_ids = set()
         try: