diff --git a/src/main/java/io/github/hostar/pythonInlay/PythonParametersInlayProvider.java b/src/main/java/io/github/hostar/pythonInlay/PythonParametersInlayProvider.java index 6dfc63d..460d9e3 100644 --- a/src/main/java/io/github/hostar/pythonInlay/PythonParametersInlayProvider.java +++ b/src/main/java/io/github/hostar/pythonInlay/PythonParametersInlayProvider.java @@ -29,6 +29,8 @@ public class PythonParametersInlayProvider implements InlayHintsProvider KEY = new SettingsKey<>(CODE_LENS_ID); + private static Boolean addColonSuffix = False; + @Nullable @Override public InlayHintsCollector getCollectorFor(@NotNull PsiFile file, @@ -189,6 +191,9 @@ private void processElement(@NotNull PsiElement element, @NotNull InlayHintsSink if (!match) { addSink(sink, paramName, args[finalPosition]); + if addColonSuffix{ + paramName += ":"; + } existingInlaysDictionary.add(new ExistingInlay(paramName, args[finalPosition].getTextOffset())); } } @@ -258,6 +263,9 @@ private void processElement(@NotNull PsiElement element, @NotNull InlayHintsSink if (!match) { addSink(sink, paramName, args[position]); position++; + if addColonSuffix{ + paramName += ":"; + } existingInlaysDictionary.add(new ExistingInlay(paramName, finalArgs1[finalPosition1].getTextOffset())); } }