Calling setup() function

Author: catilac

crates/processing_pyo3/src/lib.rs

@@ -12,7 +12,7 @@ mod glfw;
 mod graphics;
 
 use graphics::{Graphics, get_graphics, get_graphics_mut};
-use pyo3::{exceptions::PyRuntimeError, prelude::*, types::PyAny};
+use pyo3::{exceptions::PyRuntimeError, prelude::*, types::PyFunction};
 
 #[pymodule]
 fn processing(m: &Bound<'_, PyModule>) -> PyResult<()> {
@@ -38,27 +38,16 @@ fn size(module: &Bound<'_, PyModule>, width: u32, height: u32) -> PyResult<()> {
 }
 
 #[pyfunction]
-#[pyo3(pass_module, signature = (draw_fn=None))]
-fn run(module: &Bound<'_, PyModule>, draw_fn: Option<Py<PyAny>>) -> PyResult<()> {
-    loop {
-        {
-            let mut graphics = get_graphics_mut(module)?;
-            if !graphics.surface.poll_events() {
-                break;
-            }
-            graphics.begin_draw()?;
-        }
-
-        if let Some(ref draw) = draw_fn {
-            Python::attach(|py| {
-                draw.call0(py)
-                    .map_err(|e| PyRuntimeError::new_err(format!("{e}")))
-            })?;
-        }
+#[pyo3(pass_module)]
+fn run(module: &Bound<'_, PyModule>) -> PyResult<()> {
+    Python::attach(|py| {
+        let builtins = PyModule::import(py, "builtins")?;
+        let locals = builtins.getattr("locals")?.call0()?;
+        let setup_fn = locals.get_item("setup")?;
+        setup_fn.call0()?;
 
-        get_graphics(module)?.end_draw()?;
-    }
-    Ok(())
+        Ok(())
+    })
 }
 
 #[pyfunction]

rectangle.py

@@ -0,0 +1,19 @@
+from processing import *
+
+def moon_func():
+    print("Hello, Moon!")
+    
+def setup():
+    print("HELLO")
+    size(800, 600)
+
+def draw():
+    background(220)
+
+    fill(255, 0, 100)
+    stroke(0)
+    stroke_weight(2)
+    rect(100, 100, 200, 150)
+
+# TODO: this should happen implicitly on module load somehow
+run()