• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1"""
2Operator Interface
3
4This module exports a set of functions corresponding to the intrinsic
5operators of Python.  For example, operator.add(x, y) is equivalent
6to the expression x+y.  The function names are those used for special
7methods; variants without leading and trailing '__' are also provided
8for convenience.
9
10This is the pure Python implementation of the module.
11"""
12
13__all__ = ['abs', 'add', 'and_', 'attrgetter', 'concat', 'contains', 'countOf',
14           'delitem', 'eq', 'floordiv', 'ge', 'getitem', 'gt', 'iadd', 'iand',
15           'iconcat', 'ifloordiv', 'ilshift', 'imatmul', 'imod', 'imul',
16           'index', 'indexOf', 'inv', 'invert', 'ior', 'ipow', 'irshift',
17           'is_', 'is_not', 'isub', 'itemgetter', 'itruediv', 'ixor', 'le',
18           'length_hint', 'lshift', 'lt', 'matmul', 'methodcaller', 'mod',
19           'mul', 'ne', 'neg', 'not_', 'or_', 'pos', 'pow', 'rshift',
20           'setitem', 'sub', 'truediv', 'truth', 'xor']
21
22from builtins import abs as _abs
23
24
25# Comparison Operations *******************************************************#
26
27def lt(a, b):
28    "Same as a < b."
29    return a < b
30
31def le(a, b):
32    "Same as a <= b."
33    return a <= b
34
35def eq(a, b):
36    "Same as a == b."
37    return a == b
38
39def ne(a, b):
40    "Same as a != b."
41    return a != b
42
43def ge(a, b):
44    "Same as a >= b."
45    return a >= b
46
47def gt(a, b):
48    "Same as a > b."
49    return a > b
50
51# Logical Operations **********************************************************#
52
53def not_(a):
54    "Same as not a."
55    return not a
56
57def truth(a):
58    "Return True if a is true, False otherwise."
59    return True if a else False
60
61def is_(a, b):
62    "Same as a is b."
63    return a is b
64
65def is_not(a, b):
66    "Same as a is not b."
67    return a is not b
68
69# Mathematical/Bitwise Operations *********************************************#
70
71def abs(a):
72    "Same as abs(a)."
73    return _abs(a)
74
75def add(a, b):
76    "Same as a + b."
77    return a + b
78
79def and_(a, b):
80    "Same as a & b."
81    return a & b
82
83def floordiv(a, b):
84    "Same as a // b."
85    return a // b
86
87def index(a):
88    "Same as a.__index__()."
89    return a.__index__()
90
91def inv(a):
92    "Same as ~a."
93    return ~a
94invert = inv
95
96def lshift(a, b):
97    "Same as a << b."
98    return a << b
99
100def mod(a, b):
101    "Same as a % b."
102    return a % b
103
104def mul(a, b):
105    "Same as a * b."
106    return a * b
107
108def matmul(a, b):
109    "Same as a @ b."
110    return a @ b
111
112def neg(a):
113    "Same as -a."
114    return -a
115
116def or_(a, b):
117    "Same as a | b."
118    return a | b
119
120def pos(a):
121    "Same as +a."
122    return +a
123
124def pow(a, b):
125    "Same as a ** b."
126    return a ** b
127
128def rshift(a, b):
129    "Same as a >> b."
130    return a >> b
131
132def sub(a, b):
133    "Same as a - b."
134    return a - b
135
136def truediv(a, b):
137    "Same as a / b."
138    return a / b
139
140def xor(a, b):
141    "Same as a ^ b."
142    return a ^ b
143
144# Sequence Operations *********************************************************#
145
146def concat(a, b):
147    "Same as a + b, for a and b sequences."
148    if not hasattr(a, '__getitem__'):
149        msg = "'%s' object can't be concatenated" % type(a).__name__
150        raise TypeError(msg)
151    return a + b
152
153def contains(a, b):
154    "Same as b in a (note reversed operands)."
155    return b in a
156
157def countOf(a, b):
158    "Return the number of items in a which are, or which equal, b."
159    count = 0
160    for i in a:
161        if i is b or i == b:
162            count += 1
163    return count
164
165def delitem(a, b):
166    "Same as del a[b]."
167    del a[b]
168
169def getitem(a, b):
170    "Same as a[b]."
171    return a[b]
172
173def indexOf(a, b):
174    "Return the first index of b in a."
175    for i, j in enumerate(a):
176        if j is b or j == b:
177            return i
178    else:
179        raise ValueError('sequence.index(x): x not in sequence')
180
181def setitem(a, b, c):
182    "Same as a[b] = c."
183    a[b] = c
184
185def length_hint(obj, default=0):
186    """
187    Return an estimate of the number of items in obj.
188    This is useful for presizing containers when building from an iterable.
189
190    If the object supports len(), the result will be exact. Otherwise, it may
191    over- or under-estimate by an arbitrary amount. The result will be an
192    integer >= 0.
193    """
194    if not isinstance(default, int):
195        msg = ("'%s' object cannot be interpreted as an integer" %
196               type(default).__name__)
197        raise TypeError(msg)
198
199    try:
200        return len(obj)
201    except TypeError:
202        pass
203
204    try:
205        hint = type(obj).__length_hint__
206    except AttributeError:
207        return default
208
209    try:
210        val = hint(obj)
211    except TypeError:
212        return default
213    if val is NotImplemented:
214        return default
215    if not isinstance(val, int):
216        msg = ('__length_hint__ must be integer, not %s' %
217               type(val).__name__)
218        raise TypeError(msg)
219    if val < 0:
220        msg = '__length_hint__() should return >= 0'
221        raise ValueError(msg)
222    return val
223
224# Generalized Lookup Objects **************************************************#
225
226class attrgetter:
227    """
228    Return a callable object that fetches the given attribute(s) from its operand.
229    After f = attrgetter('name'), the call f(r) returns r.name.
230    After g = attrgetter('name', 'date'), the call g(r) returns (r.name, r.date).
231    After h = attrgetter('name.first', 'name.last'), the call h(r) returns
232    (r.name.first, r.name.last).
233    """
234    __slots__ = ('_attrs', '_call')
235
236    def __init__(self, attr, *attrs):
237        if not attrs:
238            if not isinstance(attr, str):
239                raise TypeError('attribute name must be a string')
240            self._attrs = (attr,)
241            names = attr.split('.')
242            def func(obj):
243                for name in names:
244                    obj = getattr(obj, name)
245                return obj
246            self._call = func
247        else:
248            self._attrs = (attr,) + attrs
249            getters = tuple(map(attrgetter, self._attrs))
250            def func(obj):
251                return tuple(getter(obj) for getter in getters)
252            self._call = func
253
254    def __call__(self, obj):
255        return self._call(obj)
256
257    def __repr__(self):
258        return '%s.%s(%s)' % (self.__class__.__module__,
259                              self.__class__.__qualname__,
260                              ', '.join(map(repr, self._attrs)))
261
262    def __reduce__(self):
263        return self.__class__, self._attrs
264
265class itemgetter:
266    """
267    Return a callable object that fetches the given item(s) from its operand.
268    After f = itemgetter(2), the call f(r) returns r[2].
269    After g = itemgetter(2, 5, 3), the call g(r) returns (r[2], r[5], r[3])
270    """
271    __slots__ = ('_items', '_call')
272
273    def __init__(self, item, *items):
274        if not items:
275            self._items = (item,)
276            def func(obj):
277                return obj[item]
278            self._call = func
279        else:
280            self._items = items = (item,) + items
281            def func(obj):
282                return tuple(obj[i] for i in items)
283            self._call = func
284
285    def __call__(self, obj):
286        return self._call(obj)
287
288    def __repr__(self):
289        return '%s.%s(%s)' % (self.__class__.__module__,
290                              self.__class__.__name__,
291                              ', '.join(map(repr, self._items)))
292
293    def __reduce__(self):
294        return self.__class__, self._items
295
296class methodcaller:
297    """
298    Return a callable object that calls the given method on its operand.
299    After f = methodcaller('name'), the call f(r) returns r.name().
300    After g = methodcaller('name', 'date', foo=1), the call g(r) returns
301    r.name('date', foo=1).
302    """
303    __slots__ = ('_name', '_args', '_kwargs')
304
305    def __init__(self, name, /, *args, **kwargs):
306        self._name = name
307        if not isinstance(self._name, str):
308            raise TypeError('method name must be a string')
309        self._args = args
310        self._kwargs = kwargs
311
312    def __call__(self, obj):
313        return getattr(obj, self._name)(*self._args, **self._kwargs)
314
315    def __repr__(self):
316        args = [repr(self._name)]
317        args.extend(map(repr, self._args))
318        args.extend('%s=%r' % (k, v) for k, v in self._kwargs.items())
319        return '%s.%s(%s)' % (self.__class__.__module__,
320                              self.__class__.__name__,
321                              ', '.join(args))
322
323    def __reduce__(self):
324        if not self._kwargs:
325            return self.__class__, (self._name,) + self._args
326        else:
327            from functools import partial
328            return partial(self.__class__, self._name, **self._kwargs), self._args
329
330
331# In-place Operations *********************************************************#
332
333def iadd(a, b):
334    "Same as a += b."
335    a += b
336    return a
337
338def iand(a, b):
339    "Same as a &= b."
340    a &= b
341    return a
342
343def iconcat(a, b):
344    "Same as a += b, for a and b sequences."
345    if not hasattr(a, '__getitem__'):
346        msg = "'%s' object can't be concatenated" % type(a).__name__
347        raise TypeError(msg)
348    a += b
349    return a
350
351def ifloordiv(a, b):
352    "Same as a //= b."
353    a //= b
354    return a
355
356def ilshift(a, b):
357    "Same as a <<= b."
358    a <<= b
359    return a
360
361def imod(a, b):
362    "Same as a %= b."
363    a %= b
364    return a
365
366def imul(a, b):
367    "Same as a *= b."
368    a *= b
369    return a
370
371def imatmul(a, b):
372    "Same as a @= b."
373    a @= b
374    return a
375
376def ior(a, b):
377    "Same as a |= b."
378    a |= b
379    return a
380
381def ipow(a, b):
382    "Same as a **= b."
383    a **=b
384    return a
385
386def irshift(a, b):
387    "Same as a >>= b."
388    a >>= b
389    return a
390
391def isub(a, b):
392    "Same as a -= b."
393    a -= b
394    return a
395
396def itruediv(a, b):
397    "Same as a /= b."
398    a /= b
399    return a
400
401def ixor(a, b):
402    "Same as a ^= b."
403    a ^= b
404    return a
405
406
407try:
408    from _operator import *
409except ImportError:
410    pass
411else:
412    from _operator import __doc__
413
414# All of these "__func__ = func" assignments have to happen after importing
415# from _operator to make sure they're set to the right function
416__lt__ = lt
417__le__ = le
418__eq__ = eq
419__ne__ = ne
420__ge__ = ge
421__gt__ = gt
422__not__ = not_
423__abs__ = abs
424__add__ = add
425__and__ = and_
426__floordiv__ = floordiv
427__index__ = index
428__inv__ = inv
429__invert__ = invert
430__lshift__ = lshift
431__mod__ = mod
432__mul__ = mul
433__matmul__ = matmul
434__neg__ = neg
435__or__ = or_
436__pos__ = pos
437__pow__ = pow
438__rshift__ = rshift
439__sub__ = sub
440__truediv__ = truediv
441__xor__ = xor
442__concat__ = concat
443__contains__ = contains
444__delitem__ = delitem
445__getitem__ = getitem
446__setitem__ = setitem
447__iadd__ = iadd
448__iand__ = iand
449__iconcat__ = iconcat
450__ifloordiv__ = ifloordiv
451__ilshift__ = ilshift
452__imod__ = imod
453__imul__ = imul
454__imatmul__ = imatmul
455__ior__ = ior
456__ipow__ = ipow
457__irshift__ = irshift
458__isub__ = isub
459__itruediv__ = itruediv
460__ixor__ = ixor
461