PY-43407 SciView(feat): First Plotly support in Scientific View

GitOrigin-RevId: 9f7a56449aafc5096d6fba01dbccd049b89b68bb

python/helpers/pycharm_matplotlib_backend/sitecustomize.py

@@ -10,7 +10,7 @@ def debug(message):
         sys.stderr.write("\n")
 
 
-debug("Executing PyCharm's `sitecustomize`")
+debug("Executing PyCharm's Matplotlib `sitecustomize`")
 modules_list = []
 
 try:

python/helpers/pycharm_plotly_backend/sitecustomize.py

@@ -0,0 +1,116 @@
+import os
+import sys
+import traceback
+
+SHOW_DEBUG_INFO = os.getenv('PYCHARM_DEBUG', 'False').lower() in ['true', '1']
+
+
+def debug(message):
+    if SHOW_DEBUG_INFO:
+        sys.stderr.write(message)
+        sys.stderr.write("\n")
+
+
+def init_plotly_render():
+    from datalore.display import display, debug
+
+    is_python_3_or_higher = sys.version_info[0] >= 3
+    if not is_python_3_or_higher:
+        debug("Pycharm Plotly backend is not supported for Python 2")
+        return
+
+    # noinspection PyProtectedMember
+    from plotly.io._base_renderers import ExternalRenderer
+    # noinspection PyProtectedMember
+    from plotly.io._renderers import renderers
+
+    class DisplayDataObject:
+        def __init__(self, html_string: str):
+            self.html_string = html_string
+
+        def _repr_display_(self):
+            body = {
+                'html_string': self.html_string
+            }
+            return 'pycharm-plotly-image', body
+
+    class PycharmRenderer(ExternalRenderer):
+        def __init__(
+                self,
+                config=None,
+                auto_play=False,
+                using=None,
+                new=0,
+                autoraise=True,
+                post_script=None,
+                animation_opts=None,
+        ):
+            self.config = config
+            self.auto_play = auto_play
+            self.using = using
+            self.new = new
+            self.autoraise = autoraise
+            self.post_script = post_script
+            self.animation_opts = animation_opts
+
+        def render(self, fig_dict):
+            from plotly.io import to_html
+
+            html = to_html(
+                fig_dict,
+                config=self.config,
+                auto_play=self.auto_play,
+                include_plotlyjs=True,
+                include_mathjax="cdn",
+                post_script=self.post_script,
+                full_html=True,
+                animation_opts=self.animation_opts,
+                default_width="100%",
+                default_height="100%",
+                validate=False,
+            )
+            display(DisplayDataObject(html))
+
+    debug("Set plotly default render")
+    renderers._default_renderers = [PycharmRenderer()]
+
+
+debug("Executing PyCharm's Ploly `sitecustomize`")
+modules_list = []
+
+try:
+    # We want to import users sitecustomize.py file if any
+    sitecustomize = "sitecustomize"
+    parent_dir = os.path.abspath(os.path.join(__file__, os.pardir))
+    if parent_dir in sys.path:
+        sys.path.remove(parent_dir)
+
+        if sitecustomize in sys.modules:
+            pycharm_sitecustomize_module = sys.modules.pop(sitecustomize)
+
+            try:
+                import sitecustomize
+            except ImportError:
+                debug("User doesn't have a custom `sitecustomize`")
+                # return our module if we failed to find any other sitecustomize
+                # to prevent KeyError importing 'site.py'
+                sys.modules[sitecustomize] = pycharm_sitecustomize_module
+
+        sys.path.append(parent_dir)
+
+    # Use matplotlib backend from pycharm
+    modules_list = list(sys.modules.keys())
+
+    init_plotly_render()
+    debug("Custom plotly backend was set for Plots tool window")
+except:
+    # fallback in case matplotlib is not loaded correctly
+    debug("Cannot init plotly backend")
+    if SHOW_DEBUG_INFO:
+        traceback.print_exc()
+
+    keys = list(sys.modules.keys())
+    if modules_list:
+        for key in keys:
+            if key not in modules_list:
+                sys.modules.pop(key)

python/src/com/jetbrains/python/run/PythonScripts.kt

@@ -155,10 +155,14 @@ fun addHelperEntriesToPythonPath(envs: MutableMap<String, TargetEnvironmentFunct
       .map(PathMapping::localPath)
   val targetPathSeparator = targetPlatform.platform.pathSeparator
 
-  fun <T> T?.onResolutionFailure(message: String): T? =
-    this ?: if (failOnError) error(message)
-    // log the error and skip this entry
-    else also { LOG.error(message) }
+  fun <T> T?.onResolutionFailure(message: String): T? {
+    if (this != null)
+      return this
+    if (failOnError)
+      error(message)
+    LOG.error(message)
+    return null
+  }
 
   val uploads = pythonPathEntries.mapNotNull { pythonPathEntry ->
     pythonPathEntry

python/src/com/jetbrains/python/run/target/HelpersAwareTargetEnvironmentRequest.kt

@@ -51,11 +51,14 @@ infix fun Path.to(targetPathFun: TargetEnvironmentFunction<FullPathOnTarget>): P
 fun String.tryResolveAsPythonHelperDir(mappings: PythonHelpersMappings): PathMapping? {
   val (communityHelpers, proHelpers) = mappings
   val thisLocalPath = Path.of(this)
-  return listOfNotNull(communityHelpers, proHelpers)
+  val rootPaths = listOfNotNull(communityHelpers, proHelpers)
     .filter { (localPath) -> FileUtil.isAncestor(localPath.absolutePathString(), this, false) }
+  return rootPaths
     .firstNotNullOfOrNull { (localPath, targetPathFun) ->
-      FileUtil.getRelativePath(localPath.absolutePathString(), this, Platform.current().fileSeparator)?.let { relativePath ->
-        thisLocalPath to targetPathFun.getRelativeTargetPath(relativePath)
-      }
+      val relativePath = FileUtil.getRelativePath(localPath.absolutePathString(), this, Platform.current().fileSeparator)
+      relativePath ?: return@firstNotNullOfOrNull null
+      val relTargetPath = targetPathFun.getRelativeTargetPath(relativePath)
+      thisLocalPath to relTargetPath
+
     }
 }