MDS/openide
1
0
Fork 0
mirror of https://gitflic.ru/project/openide/openide.git synced 2025-12-14 18:05:27 +07:00
Code Issues Packages Projects Releases Wiki Activity

Don't convert invalid identifiers to parameters.

Browse Source
This commit is contained in:
Semyon Proshev
2017-10-03 18:36:11 +03:00
parent fbdcd38462
commit a7a110247e
4 changed files with 16 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
python/src/com/jetbrains/python/codeInsight/intentions/ConvertVariadicParamIntention.java
View File

@@ -77,7 +77,7 @@ public class ConvertVariadicParamIntention extends PyBaseIntentionAction {
final PyExpression firstArgument = ArrayUtil.getFirstElement(call.getArguments());
final String firstArgumentValue = PyStringLiteralUtil.getStringValue(firstArgument);
if (firstArgumentValue != null &&
PyNames.isIdentifierString(firstArgumentValue) &&
PyNames.isIdentifier(firstArgumentValue) &&
(caretInParameterList || PsiTreeUtil.isAncestor(call, element, true))) {
return true;
}
@@ -87,7 +87,7 @@ public class ConvertVariadicParamIntention extends PyBaseIntentionAction {
final PyExpression indexExpression = subscription.getIndexExpression();
final String indexValue = PyStringLiteralUtil.getStringValue(indexExpression);
if (indexValue != null &&
PyNames.isIdentifierString(indexValue) &&
PyNames.isIdentifier(indexValue) &&
(caretInParameterList || PsiTreeUtil.isAncestor(subscription, element, true))) {
return true;
}
@@ -137,6 +137,7 @@ public class ConvertVariadicParamIntention extends PyBaseIntentionAction {
.ofNullable(subscription.getIndexExpression())
.map(indexExpression -> PyUtil.as(indexExpression, PyStringLiteralExpression.class))
.map(PyStringLiteralExpression::getStringValue)
.filter(PyNames::isIdentifier)
.map(indexValue -> elementGenerator.createExpressionFromText(LanguageLevel.forElement(function), indexValue))
.ifPresent(
parameter -> {
@@ -161,9 +162,10 @@ public class ConvertVariadicParamIntention extends PyBaseIntentionAction {
.map(ArrayUtil::getFirstElement)
.map(firstArgument -> PyUtil.as(firstArgument, PyStringLiteralExpression.class))
.map(PyStringLiteralExpression::getStringValue)
.filter(PyNames::isIdentifier)
.ifPresent(
indexValue -> {
final PyNamedParameter parameterWithDefaultValue = getParameterWithDefaultValue(elementGenerator, call, indexValue);
final PyNamedParameter parameterWithDefaultValue = createParameterWithDefaultValue(elementGenerator, call, indexValue);
final PyExpression parameter = elementGenerator.createExpressionFromText(LanguageLevel.forElement(function), indexValue);
final PyParameter keywordContainer = getKeywordContainer(function);
@@ -239,9 +241,9 @@ public class ConvertVariadicParamIntention extends PyBaseIntentionAction {
}
@Nullable
private static PyNamedParameter getParameterWithDefaultValue(@NotNull PyElementGenerator elementGenerator,
@NotNull PyCallExpression call,
@NotNull String parameterName) {
private static PyNamedParameter createParameterWithDefaultValue(@NotNull PyElementGenerator elementGenerator,
@NotNull PyCallExpression call,
@NotNull String parameterName) {
final PyExpression[] arguments = call.getArguments();
if (arguments.length > 1) {
return elementGenerator.createParameter(parameterName + "=" + arguments[1].getText());

2
python/testData/intentions/convertVariadicParamInvalidIdentifiers.py Normal file
View File

@@ -0,0 +1,2 @@
def foo(**kwargs<caret>):
return kwargs["bar"] + kwargs["&"] + kwargs.get("|") + kwargs["not"]

2
python/testData/intentions/convertVariadicParamInvalidIdentifiers_after.py Normal file
View File

@@ -0,0 +1,2 @@
def foo(bar, **kwargs):
return bar + kwargs["&"] + kwargs.get("|") + kwargs["not"]

4
python/testSrc/com/jetbrains/python/intentions/PyIntentionTest.java
View File

@@ -227,6 +227,10 @@ public class PyIntentionTest extends PyTestCase {
doNegativeTest(PyBundle.message("INTN.convert.variadic.param"));
}
public void testConvertVariadicParamInvalidIdentifiers() {
doTest(PyBundle.message("INTN.convert.variadic.param"));
}
public void testConvertTripleQuotedString() { //PY-2697
doTest(PyBundle.message("INTN.triple.quoted.string"));
}