[pypy-svn] r14055 - pypy/branch/dist-2.4.1/pypy/lib

[gintas at codespeak.net]

Author: gintas
Date: Fri Jul  1 20:30:32 2005
New Revision: 14055

Modified:
   pypy/branch/dist-2.4.1/pypy/lib/operator.py
Log:
Implementeg attrgetter and itemgetter;
added type checking to delslice and getslice.


Modified: pypy/branch/dist-2.4.1/pypy/lib/operator.py
==============================================================================
--- pypy/branch/dist-2.4.1/pypy/lib/operator.py	(original)
+++ pypy/branch/dist-2.4.1/pypy/lib/operator.py	Fri Jul  1 20:30:32 2005
@@ -17,6 +17,10 @@
     'and_(a, b) -- Same as a & b.'
     return obj1 & obj2
 __and__ = and_
+def attrgetter(attr):
+    def f(obj):
+        return getattr(obj, attr)
+    return f
 def concat(obj1, obj2):
     'concat(a, b) -- Same as a + b, for a and b sequences.'
     return obj1 + obj2  # XXX cPython only works on types with sequence api
@@ -40,6 +44,8 @@
 __delitem__ = delitem
 def delslice(obj, start, end):
     'delslice(a, b, c) -- Same as del a[b:c].'
+    if not isinstance(start, int) or not isinstance(end, int):
+        raise TypeError("an integer is expected")
     del obj[start:end]
 __delslice__ = delslice
 def div(a,b):
@@ -64,6 +70,8 @@
 __getitem__ = getitem
 def getslice(a, start, end):
     'getslice(a, b, c) -- Same as a[b:c].'
+    if not isinstance(start, int) or not isinstance(end, int):
+        raise TypeError("an integer is expected")
     return a[start:end] 
 __getslice__ = getslice
 def gt(a,b):
@@ -107,6 +115,10 @@
 def is_not(a, b):
     'is_not(a, b) -- Same as a is not b.'
     return a is not b 
+def itemgetter(idx):
+    def f(obj):
+        return obj[idx]
+    return f
 def le(a, b):
     'le(a, b) -- Same as a<=b.'
     return a <= b