Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -463,6 +463,8 @@ _BLOCKED_MODULES = frozenset({
|
|
| 463 |
|
| 464 |
def _safe_import(name: str, globals_dict=None, locals_dict=None, fromlist=(), level=0):
|
| 465 |
"""Restricted import that only allows whitelisted modules."""
|
|
|
|
|
|
|
| 466 |
base_module = name.split('.')[0]
|
| 467 |
|
| 468 |
if base_module in _BLOCKED_MODULES or name in _BLOCKED_MODULES:
|
|
@@ -471,7 +473,7 @@ def _safe_import(name: str, globals_dict=None, locals_dict=None, fromlist=(), le
|
|
| 471 |
if base_module not in _ALLOWED_MODULES and name not in _ALLOWED_MODULES:
|
| 472 |
raise SandboxViolationError(f"Import of '{name}' is not allowed. Allowed modules: {', '.join(sorted(_ALLOWED_MODULES))}")
|
| 473 |
|
| 474 |
-
return
|
| 475 |
|
| 476 |
|
| 477 |
def _create_sandbox_builtins() -> Dict[str, Any]:
|
|
|
|
| 463 |
|
| 464 |
def _safe_import(name: str, globals_dict=None, locals_dict=None, fromlist=(), level=0):
|
| 465 |
"""Restricted import that only allows whitelisted modules."""
|
| 466 |
+
import builtins as _builtins
|
| 467 |
+
|
| 468 |
base_module = name.split('.')[0]
|
| 469 |
|
| 470 |
if base_module in _BLOCKED_MODULES or name in _BLOCKED_MODULES:
|
|
|
|
| 473 |
if base_module not in _ALLOWED_MODULES and name not in _ALLOWED_MODULES:
|
| 474 |
raise SandboxViolationError(f"Import of '{name}' is not allowed. Allowed modules: {', '.join(sorted(_ALLOWED_MODULES))}")
|
| 475 |
|
| 476 |
+
return _builtins.__import__(name, globals_dict, locals_dict, fromlist, level)
|
| 477 |
|
| 478 |
|
| 479 |
def _create_sandbox_builtins() -> Dict[str, Any]:
|