MDS/openide
1
0
Fork 0
mirror of https://gitflic.ru/project/openide/openide.git synced 2026-04-19 21:11:28 +07:00
Code Issues Packages Projects Releases Wiki Activity

PY-54801: Automate conda in PS in terminal activation.

Browse Source 
Conda installs user profile hook when user runs ``conda init``.
We call this hook manually

GitOrigin-RevId: 2f2886a1a9ce57e6daa828b0c3133b1265cbcfc6
This commit is contained in:
Ilya.Kazakevich
2022-06-30 17:10:52 +02:00
committed by intellij-monorepo-bot
parent b4d380b71d
commit 3cda641232
2 changed files with 34 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/python-terminal/resources/messages/PyTerminalBundle.properties
View File

@@ -1,2 +1,2 @@
activate.virtualenv.checkbox.text=Activate virtualenv
powershell.conda.not.activated=Failed to activate conda environment. Please open Anaconda prompt, and run `conda init powershell` there.
powershell.conda.not.activated=Failed to activate conda environment. \n Please open Anaconda prompt, and run `{0} init powershell` there.

38
python/python-terminal/src/com/jetbrains/python/sdk/PyVirtualEnvTerminalCustomizer.kt
View File

@@ -4,31 +4,42 @@ package com.jetbrains.python.sdk
import com.intellij.openapi.components.PersistentStateComponent
import com.intellij.openapi.components.State
import com.intellij.openapi.components.Storage
import com.intellij.openapi.diagnostic.logger
import com.intellij.openapi.module.ModuleManager
import com.intellij.openapi.options.UnnamedConfigurable
import com.intellij.openapi.project.Project
import com.intellij.openapi.projectRoots.Sdk
import com.intellij.openapi.util.SystemInfo
import com.intellij.openapi.vfs.VirtualFile
import com.jetbrains.python.packaging.PyCondaPackageService
import com.jetbrains.python.run.findActivateScript
import com.jetbrains.python.sdk.flavors.CondaEnvSdkFlavor
import org.jetbrains.plugins.terminal.LocalTerminalCustomizer
import org.jetbrains.plugins.terminal.TerminalOptionsProvider
import java.io.File
import java.nio.file.Path
import javax.swing.JCheckBox
import kotlin.io.path.Path
import kotlin.io.path.exists
import kotlin.io.path.isExecutable
import kotlin.io.path.name
class PyVirtualEnvTerminalCustomizer : LocalTerminalCustomizer() {
private fun generateCommandForPowerShell(sdk: Sdk, sdkHomePath: VirtualFile): Array<out String>? {
// TODO: This should be migrated to Targets API: each target provides terminal
if ((sdk.sdkAdditionalData as? PythonSdkAdditionalData)?.flavor is CondaEnvSdkFlavor) {
// Activate conda
// ' are in "Write-Host"
val errorMessage = PyTerminalBundle.message("powershell.conda.not.activated").replace('\'', '"')
// No need to escape path: conda can't have spaces
val condaActivateCommand = "try { conda activate ${sdkHomePath.parent.path} } catch { Write-Host('$errorMessage') }"
return arrayOf("powershell.exe", "-NoExit", "-Command", condaActivateCommand)
val condaPath = PyCondaPackageService.getCondaExecutable(sdk.homePath)?.let { Path(it) }
val condaActivationCommand: String
if (condaPath != null && condaPath.exists() && condaPath.isExecutable()) {
condaActivationCommand = getCondaActivationCommand(condaPath, sdkHomePath)
}
else {
logger<PyVirtualEnvTerminalCustomizer>().warn("Can't find $condaPath, will not activate conda")
condaActivationCommand = PyTerminalBundle.message("powershell.conda.not.activated", "conda")
}
return arrayOf("powershell.exe", "-NoExit", "-Command", condaActivationCommand)
}
// Activate convenient virtualenv
@@ -36,6 +47,23 @@ class PyVirtualEnvTerminalCustomizer : LocalTerminalCustomizer() {
return if (virtualEnvProfile.exists()) arrayOf("powershell.exe", "-NoExit", "-File", virtualEnvProfile.path) else null
}
/**
*``conda init`` installs conda activation hook into user profile
* We run this hook manually because we can't ask user to install hook and restart terminal
* In case of failure we ask user to run "conda init" manually
*/
private fun getCondaActivationCommand(condaPath: Path, sdkHomePath: VirtualFile): String {
// ' are in "Write-Host"
val errorMessage = PyTerminalBundle.message("powershell.conda.not.activated", condaPath).replace('\'', '"')
// No need to escape path: conda can't have spaces
return """
$condaPath shell.powershell hook | Out-String | Invoke-Expression ;
try { conda activate ${sdkHomePath.parent.path} } catch { Write-Host('$errorMessage') }
""".trim()
}
override fun customizeCommandAndEnvironment(project: Project,
command: Array<out String>,
envs: MutableMap<String, String>): Array<out String> {