Add back Python Console support for CodeWhisperer (#3595)

* Add back Python Console support for CodeWhisperer

* bug fix

Author: andrewyuq

jetbrains-core/src/software/aws/toolkits/jetbrains/services/codewhisperer/language/CodeWhispererLanguageManager.kt

@@ -33,7 +33,12 @@ class CodeWhispererLanguageManager {
     // plugin features when developing CodeWhisperer features.
     fun getLanguage(vFile: VirtualFile): CodeWhispererProgrammingLanguage {
         val fileTypeName = vFile.fileType.name.lowercase()
-        val fileExtension = vFile.extension?.lowercase() ?: return CodeWhispererUnknownLanguage.INSTANCE
+        val fileExtension = vFile.extension?.lowercase()
+
+        // We want to support Python Console which does not have a file extension
+        if (fileExtension == null && !fileTypeName.contains("python")) {
+            return CodeWhispererUnknownLanguage.INSTANCE
+        }
         return when {
             fileTypeName.contains("python") -> CodeWhispererPython.INSTANCE
             fileTypeName.contains("javascript") -> CodeWhispererJavaScript.INSTANCE

jetbrains-core/tst/software/aws/toolkits/jetbrains/services/codewhisperer/CodeWhispererLanguageManagerTest.kt

@@ -100,6 +100,7 @@ class CodeWhispererLanguageManagerTest {
         testGetProgrammingLanguageUtil(listOf("c++"), languageExtensionsMap[CodeWhispererC.INSTANCE], CodeWhispererC::class.java)
         testGetProgrammingLanguageUtil(listOf("Shell Script"), languageExtensionsMap[CodeWhispererShell.INSTANCE], CodeWhispererShell::class.java)
         testGetProgrammingLanguageUtil(listOf("Rust"), languageExtensionsMap[CodeWhispererRust.INSTANCE], CodeWhispererRust::class.java)
+        testGetProgrammingLanguageUtil(listOf("Python Console"), listOf(null), CodeWhispererPython::class.java)
     }
 
     @Test
@@ -113,7 +114,7 @@ class CodeWhispererLanguageManagerTest {
 
     private fun <T : CodeWhispererProgrammingLanguage> testGetProgrammingLanguageUtil(
         fileTypeNames: List<String>,
-        fileExtensions: List<String>?,
+        fileExtensions: List<String?>?,
         expectedLanguage: Class<T>
     ) {
         fileExtensions?.forEach { fileExtension ->