edsl 0.1.36.dev6__py3-none-any.whl → 0.1.37__py3-none-any.whl

edsl/utilities/repair_functions.py

@@ -1,28 +1,28 @@
-import json
-from edsl.utilities.utilities import valid_json
-
-
-def extract_json_from_string(s):
-    """Extract a JSON string from a string."""
-    # Find the first occurrence of '{'
-    start_idx = s.find("{")
-    # Find the last occurrence of '}'
-    end_idx = s.rfind("}")
-    # If both '{' and '}' are found in the string
-    if start_idx != -1 and end_idx != -1 and start_idx < end_idx:
-        # Extract the substring from start_idx to end_idx (inclusive)
-        json_str = s[start_idx : end_idx + 1]
-        try:
-            return json.loads(json_str)
-        except json.JSONDecodeError:
-            raise ValueError("Invalid JSON string")
-    else:
-        raise ValueError("No JSON object found in string")
-
-
-if __name__ == "__main__":
-    text = (
-        'Sure - here is some JSON { "key": "value", "number": 123, "array": [1, 2, 3] }'
-    )
-    extracted_json = extract_json_from_string(text)
-    d = extracted_json
+import json
+from edsl.utilities.utilities import valid_json
+
+
+def extract_json_from_string(s):
+    """Extract a JSON string from a string."""
+    # Find the first occurrence of '{'
+    start_idx = s.find("{")
+    # Find the last occurrence of '}'
+    end_idx = s.rfind("}")
+    # If both '{' and '}' are found in the string
+    if start_idx != -1 and end_idx != -1 and start_idx < end_idx:
+        # Extract the substring from start_idx to end_idx (inclusive)
+        json_str = s[start_idx : end_idx + 1]
+        try:
+            return json.loads(json_str)
+        except json.JSONDecodeError:
+            raise ValueError("Invalid JSON string")
+    else:
+        raise ValueError("No JSON object found in string")
+
+
+if __name__ == "__main__":
+    text = (
+        'Sure - here is some JSON { "key": "value", "number": 123, "array": [1, 2, 3] }'
+    )
+    extracted_json = extract_json_from_string(text)
+    d = extracted_json

edsl/utilities/restricted_python.py

@@ -1,70 +1,70 @@
-from RestrictedPython import compile_restricted, safe_globals
-from RestrictedPython.Guards import (
-    safe_builtins,
-    full_write_guard,
-    guarded_iter_unpack_sequence,
-)
-
-
-def guarded_iter(obj, allowed_types=(list, tuple, set, dict, range)):
-    """Ensures iteration is only performed on safe, allowable types."""
-    if not isinstance(obj, allowed_types):
-        raise TypeError(f"Iteration over {type(obj).__name__} not allowed.")
-    return iter(obj)
-
-
-class FunctionAForbiddenAttributeException(Exception):
-    """Exception for errors during function execution when forbidden key is used."""
-
-
-def default_guarded_getitem(ob, key):
-    sensitive_python_keys = [
-        "__dict__",
-        "__class__",
-        "__module__",
-        "__bases__",
-        "__mro__",
-        "__subclasses__",
-        "__func__",
-        "__self__",
-        "__closure__",
-        "__code__",
-        "__globals__",
-        "__call__",
-        "__getattr__",
-        "__getattribute__",
-        "__delattr__",
-        "__setattr__",
-    ]
-    if key in sensitive_python_keys:
-        raise FunctionAForbiddenAttributeException(
-            f"Access denied for attribute: {key}"
-        )
-
-    return ob.get(key)
-
-
-def create_restricted_function(function_name, source_code, loop_activated=True):
-    """Activate the function using RestrictedPython with basic restrictions."""
-    safe_env = safe_globals.copy()
-    safe_env["__builtins__"] = {**safe_builtins}
-    safe_env["_getitem_"] = default_guarded_getitem
-
-    if loop_activated:
-        safe_env["_getiter_"] = guarded_iter
-        safe_env["_iter_unpack_sequence_"] = guarded_iter_unpack_sequence
-
-    tmp_source_code = source_code.split("def ")
-    if len(tmp_source_code) >= 2:
-        source_code = "def " + tmp_source_code[1]
-
-    byte_code = compile_restricted(source_code, "<string>", "exec")
-    loc = {}
-    try:
-        exec(byte_code, safe_env, loc)
-        func = loc[function_name]
-    except Exception as e:
-        print("Creating restricted function error:", e)
-        raise e
-
-    return func
+from RestrictedPython import compile_restricted, safe_globals
+from RestrictedPython.Guards import (
+    safe_builtins,
+    full_write_guard,
+    guarded_iter_unpack_sequence,
+)
+
+
+def guarded_iter(obj, allowed_types=(list, tuple, set, dict, range)):
+    """Ensures iteration is only performed on safe, allowable types."""
+    if not isinstance(obj, allowed_types):
+        raise TypeError(f"Iteration over {type(obj).__name__} not allowed.")
+    return iter(obj)
+
+
+class FunctionAForbiddenAttributeException(Exception):
+    """Exception for errors during function execution when forbidden key is used."""
+
+
+def default_guarded_getitem(ob, key):
+    sensitive_python_keys = [
+        "__dict__",
+        "__class__",
+        "__module__",
+        "__bases__",
+        "__mro__",
+        "__subclasses__",
+        "__func__",
+        "__self__",
+        "__closure__",
+        "__code__",
+        "__globals__",
+        "__call__",
+        "__getattr__",
+        "__getattribute__",
+        "__delattr__",
+        "__setattr__",
+    ]
+    if key in sensitive_python_keys:
+        raise FunctionAForbiddenAttributeException(
+            f"Access denied for attribute: {key}"
+        )
+
+    return ob.get(key)
+
+
+def create_restricted_function(function_name, source_code, loop_activated=True):
+    """Activate the function using RestrictedPython with basic restrictions."""
+    safe_env = safe_globals.copy()
+    safe_env["__builtins__"] = {**safe_builtins}
+    safe_env["_getitem_"] = default_guarded_getitem
+
+    if loop_activated:
+        safe_env["_getiter_"] = guarded_iter
+        safe_env["_iter_unpack_sequence_"] = guarded_iter_unpack_sequence
+
+    tmp_source_code = source_code.split("def ")
+    if len(tmp_source_code) >= 2:
+        source_code = "def " + tmp_source_code[1]
+
+    byte_code = compile_restricted(source_code, "<string>", "exec")
+    loc = {}
+    try:
+        exec(byte_code, safe_env, loc)
+        func = loc[function_name]
+    except Exception as e:
+        print("Creating restricted function error:", e)
+        raise e
+
+    return func