Select Word handles escape sequences in Python string literals (PY-9014)

python/src/com/jetbrains/python/editor/selectWord/PyLiteralSelectionHandler.java

@@ -1,14 +1,17 @@
 package com.jetbrains.python.editor.selectWord;
 
 import com.intellij.codeInsight.editorActions.ExtendWordSelectionHandler;
+import com.intellij.codeInsight.editorActions.SelectWordUtil;
 import com.intellij.lang.ASTNode;
 import com.intellij.openapi.editor.Editor;
 import com.intellij.openapi.util.TextRange;
 import com.intellij.psi.PsiElement;
 import com.intellij.psi.util.PsiTreeUtil;
 import com.jetbrains.python.PyTokenTypes;
+import com.jetbrains.python.lexer.PyStringLiteralLexer;
 import com.jetbrains.python.psi.PyStringLiteralExpression;
 
+import java.util.ArrayList;
 import java.util.Collections;
 import java.util.List;
 
@@ -26,10 +29,18 @@ public class PyLiteralSelectionHandler implements ExtendWordSelectionHandler {
   public List<TextRange> select(PsiElement e, CharSequence editorText, int cursorOffset, Editor editor) {
     final PyStringLiteralExpression literal = PsiTreeUtil.getParentOfType(e, PyStringLiteralExpression.class);
     if (literal != null) {
-      for (TextRange stringRange : literal.getStringValueTextRanges()) {
+      List<TextRange> ranges = literal.getStringValueTextRanges();
+      List<ASTNode> nodes = literal.getStringNodes();
+      for (int i = 0; i < ranges.size(); i++) {
+        TextRange stringRange = ranges.get(i);
         TextRange offsetRange = stringRange.shiftRight(literal.getTextRange().getStartOffset());
         if (offsetRange.contains(cursorOffset) && offsetRange.getLength() > 1) {
-          return Collections.singletonList(offsetRange);
+          List<TextRange> result = new ArrayList<TextRange>();
+          SelectWordUtil.addWordHonoringEscapeSequences(editorText, nodes.get(i).getTextRange(), cursorOffset,
+                                                        new PyStringLiteralLexer(nodes.get(i).getElementType()),
+                                                        result);
+          result.add(offsetRange);
+          return result;
         }
       }
     }

python/testData/selectWord/escapeSequence/after1.py

@@ -0,0 +1 @@
+s = 'Hello\n<selection>World</selection>!'

python/testData/selectWord/escapeSequence/before.py

@@ -0,0 +1 @@
+s = 'Hello\nWo<caret>rld!'

python/testSrc/com/jetbrains/python/PySelectWordTest.java

@@ -31,6 +31,10 @@ public class PySelectWordTest extends PyTestCase {
     doTest();
   }
 
+  public void testEscapeSequence() {  // PY-9014
+    doTest();
+  }
+
   private void doTest() {
     @NonNls final String path = "selectWord/" + getTestName(true);
     myFixture.copyDirectoryToProject(path, path);