continue work on annotation of keyword-only args

This commit is contained in:
Dmitry Jemerov
2010-01-21 16:24:42 +03:00
parent 3c8137a80f
commit ba973874a1
4 changed files with 31 additions and 26 deletions

View File

@@ -157,6 +157,7 @@ ANN.starred.param.after.kwparam=* parameter after ** parameter
ANN.regular.param.after.vararg=regular parameter after * parameter ANN.regular.param.after.vararg=regular parameter after * parameter
ANN.regular.param.after.keyword=regular parameter after ** parameter ANN.regular.param.after.keyword=regular parameter after ** parameter
ANN.non.default.param.after.default=non-default parameter follows default parameter ANN.non.default.param.after.default=non-default parameter follows default parameter
ANN.named.arguments.after.star=named arguments must follow bare *
ANN.star.import.at.top.only='import *' only allowed at module level ANN.star.import.at.top.only='import *' only allowed at module level

View File

@@ -22,34 +22,41 @@ public class ParameterListAnnotator extends PyAnnotator {
boolean hadKeywordContainer = false; boolean hadKeywordContainer = false;
boolean hadDefaultValue = false; boolean hadDefaultValue = false;
boolean hadSingleStar = false; boolean hadSingleStar = false;
boolean hadParamsAfterSingleStar = false;
@Override @Override
public void visitNamedParameter(PyNamedParameter parameter, boolean first, boolean last) { public void visitNamedParameter(PyNamedParameter parameter, boolean first, boolean last) {
if (parameterNames.contains(parameter.getName())) { if (parameterNames.contains(parameter.getName())) {
getHolder().createErrorAnnotation(parameter, PyBundle.message("ANN.duplicate.param.name")); markError(parameter, PyBundle.message("ANN.duplicate.param.name"));
} }
parameterNames.add(parameter.getName()); parameterNames.add(parameter.getName());
if (parameter.isPositionalContainer()) { if (parameter.isPositionalContainer()) {
if (hadKeywordContainer) { if (hadKeywordContainer) {
getHolder().createErrorAnnotation(parameter, PyBundle.message("ANN.starred.param.after.kwparam")); markError(parameter, PyBundle.message("ANN.starred.param.after.kwparam"));
} }
hadPositionalContainer = true; hadPositionalContainer = true;
} }
else if (parameter.isKeywordContainer()) { else if (parameter.isKeywordContainer()) {
hadKeywordContainer = true; hadKeywordContainer = true;
if (hadSingleStar && !hadParamsAfterSingleStar) {
markError(parameter, PyBundle.message("ANN.named.arguments.after.star"));
}
} }
else { else {
if (hadSingleStar) {
hadParamsAfterSingleStar = true;
}
if (hadPositionalContainer && !languageLevel.isPy3K()) { if (hadPositionalContainer && !languageLevel.isPy3K()) {
getHolder().createErrorAnnotation(parameter, PyBundle.message("ANN.regular.param.after.vararg")); markError(parameter, PyBundle.message("ANN.regular.param.after.vararg"));
} }
else if (hadKeywordContainer) { else if (hadKeywordContainer) {
getHolder().createErrorAnnotation(parameter, PyBundle.message("ANN.regular.param.after.keyword")); markError(parameter, PyBundle.message("ANN.regular.param.after.keyword"));
} }
if (parameter.getDefaultValue() != null) { if (parameter.getDefaultValue() != null) {
hadDefaultValue = true; hadDefaultValue = true;
} }
else { else {
if (hadDefaultValue && !hadSingleStar && (!languageLevel.isPy3K() || !hadPositionalContainer)) { if (hadDefaultValue && !hadSingleStar && (!languageLevel.isPy3K() || !hadPositionalContainer)) {
getHolder().createErrorAnnotation(parameter, PyBundle.message("ANN.non.default.param.after.default")); markError(parameter, PyBundle.message("ANN.non.default.param.after.default"));
} }
} }
} }
@@ -58,6 +65,9 @@ public class ParameterListAnnotator extends PyAnnotator {
@Override @Override
public void visitSingleStarParameter(PySingleStarParameter param, boolean first, boolean last) { public void visitSingleStarParameter(PySingleStarParameter param, boolean first, boolean last) {
hadSingleStar = true; hadSingleStar = true;
if (last) {
markError(param, PyBundle.message("ANN.named.arguments.after.star"));
}
} }
} }
); );

View File

@@ -1,19 +1,3 @@
/*
* Copyright 2005 Pythonid Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS"; BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.jetbrains.python.validation; package com.jetbrains.python.validation;
import com.jetbrains.python.psi.PyElementVisitor; import com.jetbrains.python.psi.PyElementVisitor;
@@ -21,11 +5,7 @@ import com.intellij.lang.annotation.AnnotationHolder;
import com.intellij.psi.PsiElement; import com.intellij.psi.PsiElement;
/** /**
* Created by IntelliJ IDEA. * @author yole
* User: yole
* Date: 12.06.2005
* Time: 13:43:02
* To change this template use File | Settings | File Templates.
*/ */
public abstract class PyAnnotator extends PyElementVisitor { public abstract class PyAnnotator extends PyElementVisitor {
private AnnotationHolder _holder; private AnnotationHolder _holder;
@@ -47,4 +27,8 @@ public abstract class PyAnnotator extends PyElementVisitor {
setHolder(null); setHolder(null);
} }
} }
protected void markError(PsiElement element, String message) {
getHolder().createErrorAnnotation(element, message);
}
} }

View File

@@ -1,2 +1,12 @@
def keywordonly_sum(*, k1=0, k2): def keywordonly_sum(*, k1=0, k2):
return k1 + k2 return k1 + k2
def keywordonly_sum_bad1(p, <error descr="named arguments must follow bare *">*</error>):
pass
def keywordonly_sum_bad2(p1, *, <error descr="named arguments must follow bare *">**k1</error>):
pass
def keywordonly_and_kwarg_sum(*, k1, k2, **kwarg):
return k1 + k2 + sum(kwarg.values())