Merge pull request #4 from takemikami/support_liststyle_argnames

Support liststyle argnames

Author: takemikami

build.gradle.kts

@@ -5,7 +5,7 @@ plugins {
 }
 
 group = "com.github.takemikami.intellij.plugin.pytestparametrize"
-version = "0.0.1"
+version = "0.0.2"
 
 repositories {
     mavenCentral()
@@ -14,7 +14,7 @@ repositories {
 // Configure Gradle IntelliJ Plugin
 // Read more: https://plugins.jetbrains.com/docs/intellij/tools-gradle-intellij-plugin.html
 intellij {
-    version.set("2021.3")
+    version.set("2023.1")
     type.set("PC") // Target IDE Platform
 
 //    plugins.set(listOf(/* Plugin Dependencies */))
@@ -33,7 +33,7 @@ tasks {
 
     patchPluginXml {
         sinceBuild.set("213")
-        untilBuild.set("223.*")
+        untilBuild.set("231.*")
     }
 
     signPlugin {

src/main/kotlin/com/github/takemikami/intellij/plugin/pytestparametrize/PytestParametrizeInlayHintsCollector.kt

@@ -24,9 +24,13 @@ class PytestParametrizeInlayHintsCollector(
                 && element.arguments.size >= 2) {
             val names = element.arguments[0]
             val valList = element.arguments[1]
-            if (names !is PyStringLiteralExpression || valList !is PyListLiteralExpression) { return true }
+            if (valList !is PyListLiteralExpression
+                    || (names !is PyStringLiteralExpression && names !is PyListLiteralExpression)) { return true }
 
-            val nameKeys = names.stringValue.split(",").map{it.trim()}
+            val nameKeys = if (names is PyStringLiteralExpression)
+                names.stringValue.split(",").map{it.trim()}
+            else
+                names.children.map{it.text.removeSurrounding("\"").removeSurrounding("\'")}
             val hintName: InlayPresentation = factory.seq()
             for ((idx, paramset) in valList.elements.withIndex()) {
                 if (paramset !is PyParenthesizedExpression) { continue }