Added a range with a higher limit

restrictedpython_code_eval.py

@@ -62,6 +62,34 @@ def limited_tuple(seq=None):
 limited_builtins['tuple'] = limited_tuple
 
 
+def limited_range(iFirst, *args):
+    # limited range function from Martijn Pieters
+    RANGELIMIT = 10000
+    if not len(args):
+        iStart, iEnd, iStep = 0, iFirst, 1
+    elif len(args) == 1:
+        iStart, iEnd, iStep = iFirst, args[0], 1
+    elif len(args) == 2:
+        iStart, iEnd, iStep = iFirst, args[0], args[1]
+    else:
+        raise AttributeError('range() requires 1-3 int arguments')
+    if iStep == 0:
+        raise ValueError('zero step for range()')
+    iLen = int((iEnd - iStart) / iStep)
+    if iLen < 0:
+        iLen = 0
+    if iLen >= RANGELIMIT:
+        raise ValueError(
+            'To be created range() object would be too large, '
+            'in RestrictedPython we only allow {limit} '
+            'elements in a range.'.format(limit=str(RANGELIMIT)),
+        )
+    return range(iStart, iEnd, iStep)
+
+
+limited_builtins['range'] = limited_range
+
+
 def safer_getattr_allowing_string_format(object, name, default=None, getattr=getattr):
     """Getattr implementation allowing str.format(), but preventing access to
     private attributes.