From a29ea5a99b57d8eab3a55efb6f622239b4cab174 Mon Sep 17 00:00:00 2001
From: Valentina Kiryushkina <valentina.kiryushkina@jetbrains.com>
Date: Wed, 25 Nov 2015 19:36:02 +0300
Subject: [PATCH] Handle rename

---
 .../codeInsight/PySubstitutionChunkReference.java      | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/python/src/com/jetbrains/python/codeInsight/PySubstitutionChunkReference.java b/python/src/com/jetbrains/python/codeInsight/PySubstitutionChunkReference.java
index 1db0eee06725..b39f483cbe76 100644
--- a/python/src/com/jetbrains/python/codeInsight/PySubstitutionChunkReference.java
+++ b/python/src/com/jetbrains/python/codeInsight/PySubstitutionChunkReference.java
@@ -20,6 +20,7 @@ import com.intellij.psi.PsiElement;
 import com.intellij.psi.PsiReferenceBase;
 import com.intellij.psi.util.PsiTreeUtil;
 import com.intellij.util.ArrayUtil;
+import com.intellij.util.IncorrectOperationException;
 import com.intellij.util.ObjectUtils;
 import com.jetbrains.python.inspections.PyStringFormatParser;
 import com.jetbrains.python.psi.*;
@@ -27,11 +28,11 @@ import com.jetbrains.python.psi.types.TypeEvalContext;
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.annotations.Nullable;
 
-public class PySubstitutionChunkReference extends PsiReferenceBase<StringLiteralExpression> implements PsiReferenceEx{
+public class PySubstitutionChunkReference extends PsiReferenceBase<PyStringLiteralExpression> implements PsiReferenceEx{
   private final int myPosition;
   private final PyStringFormatParser.SubstitutionChunk myChunk;
 
-  public PySubstitutionChunkReference(StringLiteralExpression element, PyStringFormatParser.SubstitutionChunk chunk, final int position) {
+  public PySubstitutionChunkReference(PyStringLiteralExpression element, PyStringFormatParser.SubstitutionChunk chunk, final int position) {
     super(element, chunk.getTextRange().shiftRight(1));
     myChunk = chunk;
     myPosition = position;
@@ -85,6 +86,11 @@ public class PySubstitutionChunkReference extends PsiReferenceBase<StringLiteral
     return null;
   }
 
+  @Override
+  public PsiElement handleElementRename(String newElementName) throws IncorrectOperationException {
+    return super.handleElementRename("{" +newElementName + "}");
+  }
+
   private PsiElement resolvePercentString() {
     PyBinaryExpression binaryExpression = PsiTreeUtil.getParentOfType(getElement(), PyBinaryExpression.class);
     final Class[] SIMPLE_RESULT_EXPRESSIONS = {