@@ -5,7 +5,7 @@ import com.jetbrains.python.psi.PyLambdaExpression
5
5
import com.jetbrains.python.psi.types.*
6
6
7
7
enum class HintGenerator {
8
- UNION_TYPE () {
8
+ UNION_TYPE {
9
9
override fun handleType (type : PyType ? , typeEvalContext : TypeEvalContext ): String? {
10
10
if (type !is PyUnionType ) {
11
11
return null
@@ -24,7 +24,7 @@ enum class HintGenerator {
24
24
}
25
25
},
26
26
27
- COLLECTION_TYPE () {
27
+ COLLECTION_TYPE {
28
28
override fun handleType (type : PyType ? , typeEvalContext : TypeEvalContext ): String? {
29
29
if (
30
30
type is PyCollectionType
@@ -54,7 +54,7 @@ enum class HintGenerator {
54
54
}
55
55
},
56
56
57
- TUPLE_TYPE () {
57
+ TUPLE_TYPE {
58
58
override fun handleType (type : PyType ? , typeEvalContext : TypeEvalContext ): String? {
59
59
if (type !is PyTupleType ) {
60
60
return null
@@ -67,17 +67,17 @@ enum class HintGenerator {
67
67
if (type.elementCount > 2 ) {
68
68
val firstElement = generateTypeHintText(type.elementTypes[0 ], typeEvalContext)
69
69
val secondElement = generateTypeHintText(type.elementTypes[1 ], typeEvalContext)
70
-
70
+
71
71
return " ${PyNames .TUPLE } [$firstElement , $secondElement , ...]"
72
72
}
73
-
73
+
74
74
return type.elementTypes
75
75
.mapNotNull { generateTypeHintText(it, typeEvalContext) }
76
76
.joinToString(separator = " , " , prefix = " ${PyNames .TUPLE } [" , postfix = " ]" )
77
77
}
78
78
},
79
79
80
- CLASS_TYPE () {
80
+ CLASS_TYPE {
81
81
override fun handleType (type : PyType ? , typeEvalContext : TypeEvalContext ): String? {
82
82
if (type is PyClassType && type.isDefinition) {
83
83
return " ${PyNames .TYPE .replaceFirstChar { it.titlecaseChar() }} [${type.declarationElement?.name} ]"
@@ -87,17 +87,19 @@ enum class HintGenerator {
87
87
}
88
88
},
89
89
90
- FUNCTION_TYPE () {
90
+ FUNCTION_TYPE {
91
91
override fun handleType (type : PyType ? , typeEvalContext : TypeEvalContext ): String? {
92
92
if (type !is PyFunctionType ) {
93
93
return null
94
94
}
95
95
96
96
val parametersText = when (type.callable) {
97
97
is PyLambdaExpression -> type.callable.parameterList.getPresentableText(false , typeEvalContext)
98
- else -> " (${type.callable.parameterList.parameters
99
- .filter { ! it.isSelf }
100
- .joinToString(separator = " , " ) { " ${HintUtil .getParameterAnnotationValue(it)} " }} )"
98
+ else -> " (${
99
+ type.callable.parameterList.parameters
100
+ .filter { ! it.isSelf }
101
+ .joinToString(separator = " , " ) { " ${HintUtil .getParameterAnnotationValue(it)} " }
102
+ } )"
101
103
}
102
104
103
105
val callableReturnType = typeEvalContext.getReturnType(type.callable)
@@ -106,7 +108,7 @@ enum class HintGenerator {
106
108
}
107
109
},
108
110
109
- ANY_TYPE () {
111
+ ANY_TYPE {
110
112
override fun handleType (type : PyType ? , typeEvalContext : TypeEvalContext ): String {
111
113
return type?.name ? : PyNames .UNKNOWN_TYPE
112
114
}
0 commit comments