MDS/openide
1
0
Fork 0
mirror of https://gitflic.ru/project/openide/openide.git synced 2026-09-27 10:03:11 +07:00
Code Issues Packages Projects Releases Wiki Activity

Extracted getUnionType() and getParameterizedType()

Browse Source
This commit is contained in:
Andrey Vlasovskikh
2014-08-20 17:21:30 +04:00
parent e1e8181402
commit 88c5b7958e
1 changed files with 43 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/src/com/jetbrains/python/codeInsight/PyTypingTypeProvider.java
+43 -25
View File
@@ -72,38 +72,34 @@ public class PyTypingTypeProvider extends PyTypeProviderBase {
@Nullable
private static PyType getTypingType(@NotNull PyExpression expression, @NotNull TypeEvalContext context) {
final PyType unionType = getUnionType(expression, context);
if (unionType != null) {
return unionType;
}
final PyType parameterizedType = getParameterizedType(expression, context);
if (parameterizedType != null) {
return parameterizedType;
}
final PyType builtinCollection = getBuiltinCollection(expression, context);
if (builtinCollection != null) {
return builtinCollection;
}
final PyType genericType = getGenericType(expression, context);
if (genericType != null) {
return genericType;
}
return null;
}
@Nullable
private static PyType getUnionType(@NotNull PyExpression expression, @NotNull TypeEvalContext context) {
if (expression instanceof PySubscriptionExpression) {
final PySubscriptionExpression subscriptionExpr = (PySubscriptionExpression)expression;
final PyExpression operand = subscriptionExpr.getOperand();
final PyExpression indexExpr = subscriptionExpr.getIndexExpression();
final String operandName = resolveToQualifiedName(operand, context);
if ("typing.Union".equals(operandName)) {
return PyUnionType.union(getIndexTypes(subscriptionExpr, context));
}
else {
final PyType operandType = getType(operand, context);
if (operandType instanceof PyClassType) {
final PyClass cls = ((PyClassType)operandType).getPyClass();
if (PyNames.TUPLE.equals(cls.getQualifiedName())) {
final List<PyType> indexTypes = getIndexTypes(subscriptionExpr, context);
return PyTupleType.create(expression, indexTypes.toArray(new PyType[indexTypes.size()]));
}
else if (indexExpr != null) {
final PyType indexType = context.getType(indexExpr);
return new PyCollectionTypeImpl(cls, false, indexType);
}
}
}
}
else {
final PyType builtinCollection = getBuiltinCollection(expression, context);
if (builtinCollection != null) {
return builtinCollection;
}
final PyType genericType = getGenericType(expression, context);
if (genericType != null) {
return genericType;
}
}
return null;
}
@@ -174,6 +170,28 @@ public class PyTypingTypeProvider extends PyTypeProviderBase {
return types;
}
@Nullable
private static PyType getParameterizedType(@NotNull PyExpression expression, @NotNull TypeEvalContext context) {
if (expression instanceof PySubscriptionExpression) {
final PySubscriptionExpression subscriptionExpr = (PySubscriptionExpression)expression;
final PyExpression operand = subscriptionExpr.getOperand();
final PyExpression indexExpr = subscriptionExpr.getIndexExpression();
final PyType operandType = getType(operand, context);
if (operandType instanceof PyClassType) {
final PyClass cls = ((PyClassType)operandType).getPyClass();
if (PyNames.TUPLE.equals(cls.getQualifiedName())) {
final List<PyType> indexTypes = getIndexTypes(subscriptionExpr, context);
return PyTupleType.create(expression, indexTypes.toArray(new PyType[indexTypes.size()]));
}
else if (indexExpr != null) {
final PyType indexType = context.getType(indexExpr);
return new PyCollectionTypeImpl(cls, false, indexType);
}
}
}
return null;
}
@Nullable
private static PyType getBuiltinCollection(@NotNull PyExpression expression, @NotNull TypeEvalContext context) {
final String collectionName = resolveToQualifiedName(expression, context);