support multiple Jupyter execute messages in one group;

also added print() function as alias for log.debug() so black
nbextension works

Author: craigbarratt

custom_components/pyscript/const.py

@@ -7,3 +7,21 @@
 SERVICE_JUPYTER_KERNEL_START = "jupyter_kernel_start"
 
 LOGGER_PATH = "custom_components.pyscript"
+
+ALLOWED_IMPORTS = {
+    "black",
+    "cmath",
+    "datetime",
+    "decimal",
+    "fractions",
+    "homeassistant.const",
+    "isort",
+    "json",
+    "math",
+    "number",
+    "random",
+    "re",
+    "statistics",
+    "string",
+    "time",
+}

custom_components/pyscript/eval.py

@@ -6,7 +6,7 @@
 import logging
 import sys
 
-from .const import LOGGER_PATH
+from .const import ALLOWED_IMPORTS, LOGGER_PATH
 
 _LOGGER = logging.getLogger(LOGGER_PATH + ".eval")
 
@@ -148,24 +148,6 @@ async def locals_func():
 }
 
 
-ALLOWED_IMPORTS = {
-    "bisect",
-    "cmath",
-    "colorsys",
-    "datetime",
-    "decimal",
-    "fractions",
-    "homeassistant.const",
-    "math",
-    "number",
-    "random",
-    "re",
-    "statistics",
-    "string",
-    "time",
-}
-
-
 #
 # Objects returned by return, break and continue statements that change execution flow
 #

custom_components/pyscript/handler.py

@@ -43,6 +43,7 @@ def __init__(self, hass):
             "log.error": self.get_logger_error,
             "log.info": self.get_logger_info,
             "log.warning": self.get_logger_warning,
+            "print": self.get_logger_debug,
         }
 
         #