Lichen

lib/operator/binary.py

580:e703b981b9b1
2017-02-13 Paul Boddie Eliminated redundant struct usage. method-wrapper-for-context
     1 #!/usr/bin/env python     2      3 """     4 Operator support.     5      6 Copyright (C) 2010, 2013, 2015, 2016 Paul Boddie <paul@boddie.org.uk>     7      8 This program is free software; you can redistribute it and/or modify it under     9 the terms of the GNU General Public License as published by the Free Software    10 Foundation; either version 3 of the License, or (at your option) any later    11 version.    12     13 This program is distributed in the hope that it will be useful, but WITHOUT    14 ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS    15 FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more    16 details.    17     18 You should have received a copy of the GNU General Public License along with    19 this program.  If not, see <http://www.gnu.org/licenses/>.    20 """    21     22 from operator.core import binary_op, is_, is_not    23     24 # These functions defer method lookup by wrapping the attribute access in    25 # lambda functions. Thus, the appropriate methods are defined locally, but no    26 # attempt to obtain them is made until the generic function is called.    27     28 # Binary operator functions.    29     30 def add(a, b):    31     return binary_op(a, b, lambda a: a.__add__, lambda b: b.__radd__)    32     33 def and_(a, b):    34     return binary_op(a, b, lambda a: a.__and__, lambda b: b.__rand__)    35     36 def contains(a, b):    37     return in_(b, a)    38     39 def div(a, b):    40     return binary_op(a, b, lambda a: a.__div__, lambda b: b.__rdiv__)    41     42 def floordiv(a, b):    43     return binary_op(a, b, lambda a: a.__floordiv__, lambda b: b.__rfloordiv__)    44     45 def in_(a, b):    46     return b.__contains__(a)    47     48 def not_in(a, b):    49     return not b.__contains__(a)    50     51 def lshift(a, b):    52     return binary_op(a, b, lambda a: a.__lshift__, lambda b: b.__rlshift__)    53     54 def mod(a, b):    55     return binary_op(a, b, lambda a: a.__mod__, lambda b: b.__rmod__)    56     57 def mul(a, b):    58     return binary_op(a, b, lambda a: a.__mul__, lambda b: b.__rmul__)    59     60 def or_(a, b):    61     return binary_op(a, b, lambda a: a.__or__, lambda b: b.__ror__)    62     63 def pow(a, b):    64     return binary_op(a, b, lambda a: a.__pow__, lambda b: b.__rpow__)    65     66 def rshift(a, b):    67     return binary_op(a, b, lambda a: a.__rshift__, lambda b: b.__rrshift__)    68     69 def sub(a, b):    70     return binary_op(a, b, lambda a: a.__sub__, lambda b: b.__rsub__)    71     72 def xor(a, b):    73     return binary_op(a, b, lambda a: a.__xor__, lambda b: b.__rxor__)    74     75 # vim: tabstop=4 expandtab shiftwidth=4