Support for accessing Python scope

src/hooks/useGlobals.ts

@@ -0,0 +1,24 @@
+import { Remote } from 'comlink'
+import { Runner } from '../types/Runner'
+
+interface UseGlobalsProps {
+  runner: Remote<Runner> | undefined
+}
+
+export default function useGlobals(props: UseGlobalsProps) {
+  const { runner } = props
+
+  const getGlobal = (variable: string) => {
+    return runner?.getGlobal(variable)
+  }
+
+  // eslint-disable-next-line @typescript-eslint/no-explicit-any
+  const setGlobal = (variable: string, value: any) => {
+    return runner?.setGlobal(variable, value)
+  }
+
+  return {
+    getGlobal,
+    setGlobal
+  }
+}

src/hooks/usePython.ts

@@ -1,6 +1,7 @@
 import { useContext, useEffect, useMemo, useRef, useState } from 'react'
 import { PythonContext, suppressedMessages } from '../providers/PythonProvider'
 import { proxy, Remote, wrap } from 'comlink'
+import useGlobals from './useGlobals'
 import useFilesystem from './useFilesystem'
 
 import { Packages } from '../types/Packages'
@@ -32,6 +33,8 @@ export default function usePython(props?: UsePythonProps) {
   const workerRef = useRef<Worker>()
   const runnerRef = useRef<Remote<PythonRunner>>()
 
+  const { getGlobal, setGlobal } = useGlobals({ runner: runnerRef?.current })
+
   const {
     readFile,
     writeFile,
@@ -144,14 +147,18 @@ import sys
 
 sys.tracebacklimit = 0
 
+# Patch time.sleep
 import time
 def sleep(seconds):
     start = now = time.time()
     while now - start < seconds:
         now = time.time()
 time.sleep = sleep
 
+globals_ = {}
+
 def run(code, preamble=''):
+    global globals_
     globals_ = {}
     try:
         exec(preamble, globals_)
@@ -251,6 +258,8 @@ del sys
     isReady,
     isRunning,
     interruptExecution,
+    getGlobal,
+    setGlobal,
     readFile,
     writeFile,
     mkdir,

src/types/Runner.ts

@@ -4,6 +4,10 @@ export interface Runner {
     onLoad: ({ version, banner }: { version: string; banner?: string }) => void,
     packages?: string[][]
   ) => Promise<void>
+  // eslint-disable-next-line @typescript-eslint/no-explicit-any
+  getGlobal: (variable: string) => any
+  // eslint-disable-next-line @typescript-eslint/no-explicit-any
+  setGlobal: (variable: string, value: any) => void
   interruptExecution: () => void
   readFile: (name: string) => void
   writeFile: (name: string, data: string) => void

src/workers/python-console-worker.ts

@@ -12,8 +12,12 @@ interface Pyodide {
     mkdir: (name: string) => void
     rmdir: (name: string) => void
   }
-  // eslint-disable-next-line @typescript-eslint/no-explicit-any
-  globals: any
+  globals: {
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    get: (variable: string) => any
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    set: (variable: string, value: any) => void
+  }
   isPyProxy: (value: unknown) => boolean
 }
 

src/workers/python-worker.ts

@@ -12,8 +12,12 @@ interface Pyodide {
     mkdir: (name: string) => void
     rmdir: (name: string) => void
   }
-  // eslint-disable-next-line @typescript-eslint/no-explicit-any
-  globals: any
+  globals: {
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    get: (variable: string) => any
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    set: (variable: string, value: any) => void
+  }
   isPyProxy: (value: unknown) => boolean
 }
 
@@ -60,6 +64,17 @@ const python = {
   },
   async run(code: string) {
     await self.pyodide.runPythonAsync(code)
+    const g = self.pyodide.globals.get('globals_') // dict
+    console.debug('y', g.get('y')) // 5
+  },
+  getGlobal(variable: string) {
+    const g = self.pyodide.globals.get('globals_') // undefined
+    console.debug('y', g.get('y')) // (crashed in prev step)
+
+    return g.get(variable)
+  },
+  setGlobal(variable: string, value: unknown) {
+    return self.pyodide.globals.set(variable, value)
   },
   readFile(name: string) {
     return self.pyodide.FS.readFile(name, { encoding: 'utf8' })

website/docs/examples/playground.mdx

@@ -0,0 +1,26 @@
+---
+sidebar_position: 0
+draft: true
+---
+
+import ScopeExample from '../../src/components/ScopeExample'
+
+# Playground
+
+Used for testing bugs etc.
+
+```python
+# x = 2 + 3
+y = 2 + 3
+print(y)
+```
+
+<ScopeExample />
+
+<!-- pyodide.globals.set("x", 'x will be now string');
+
+pyodide.globals.set("alert", alert); -->
+
+```python
+square(3)
+```

website/src/components/ScopeExample.tsx

@@ -0,0 +1,30 @@
+import React, { useState } from 'react'
+import { usePython } from '@site/../dist'
+
+export default function ScopeExample() {
+  const { getGlobal, setGlobal } = usePython()
+
+  const [x, setX] = useState()
+
+  return (
+    <div className="mb-10">
+      <h2>Scope</h2>
+      <button
+        onClick={async () => {
+          try {
+            const g = await getGlobal('x')
+            console.log('g', g)
+
+            setX(g)
+          } catch (error) {
+            console.error('err getting g', error)
+          }
+        }}
+      >
+        Get x
+      </button>
+      <br />
+      <span>x: {JSON.stringify(x)}</span>
+    </div>
+  )
+}