Function _worker

backend_api_python/app/utils/safe_exec.py:367–411

(code, input_data, max_memory_mb, result_pipe)

        return {'success': False, 'error': f"Unsafe code rejected: {err}", 'result': None}

    def _worker(code, input_data, max_memory_mb, result_pipe):
        try:
            if sys.platform != 'win32':
                try:
                    import resource
                    mem = max_memory_mb * 1024 * 1024
                    resource.setrlimit(resource.RLIMIT_AS, (mem, mem))
                except Exception:
                    pass

            import numpy as np
            import pandas as pd

            exec_env = {
                '__builtins__': build_safe_builtins(),
                'np': np,
                'pd': pd,
            }
            if input_data:
                exec_env.update(input_data)

            pre_import = "import numpy as np\nimport pandas as pd\n"
            exec(pre_import, exec_env)
            exec(code, exec_env)

            # Extract only picklable, non-module results
            output = {}
            for k, v in exec_env.items():
                if k.startswith('_') or k in ('np', 'pd', '__builtins__'):
                    continue
                try:
                    pickle.dumps(v)
                    output[k] = v
                except Exception:
                    pass

            result_pipe.send({'success': True, 'error': None, 'result': output})
        except Exception as e:
            result_pipe.send({
                'success': False,
                'error': f"{type(e).__name__}: {e}",
                'result': None,
            })
        finally:
            result_pipe.close()

    parent_conn, child_conn = multiprocessing.Pipe(duplex=False)