Lichen

Annotated lib/operator/comparison.py

932:c07b0dd14f85
12 months ago Paul Boddie Moved integer instantiation support to library functions.
paul@6 1
#!/usr/bin/env python
paul@6 2
paul@6 3
"""
paul@6 4
Operator support.
paul@6 5
paul@762 6
Copyright (C) 2010, 2013, 2015, 2016, 2017 Paul Boddie <paul@boddie.org.uk>
paul@6 7
paul@6 8
This program is free software; you can redistribute it and/or modify it under
paul@6 9
the terms of the GNU General Public License as published by the Free Software
paul@6 10
Foundation; either version 3 of the License, or (at your option) any later
paul@6 11
version.
paul@6 12
paul@6 13
This program is distributed in the hope that it will be useful, but WITHOUT
paul@6 14
ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
paul@6 15
FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
paul@6 16
details.
paul@6 17
paul@6 18
You should have received a copy of the GNU General Public License along with
paul@6 19
this program.  If not, see <http://www.gnu.org/licenses/>.
paul@6 20
"""
paul@6 21
paul@6 22
from operator.core import binary_op
paul@762 23
from native import int_eq, int_ge, int_gt, int_le, int_lt, int_ne, is_int
paul@6 24
paul@6 25
# These functions defer method lookup by wrapping the attribute access in
paul@6 26
# lambda functions. Thus, the appropriate methods are defined locally, but no
paul@6 27
# attempt to obtain them is made until the generic function is called.
paul@6 28
paul@6 29
# Comparison functions.
paul@6 30
paul@6 31
def eq(a, b):
paul@762 32
    if is_int(a) and is_int(b):
paul@762 33
        return int_eq(a, b)
paul@278 34
    return binary_op(a, b, lambda a: a.__eq__, lambda b: b.__eq__, False)
paul@6 35
paul@6 36
def ge(a, b):
paul@762 37
    if is_int(a) and is_int(b):
paul@762 38
        return int_ge(a, b)
paul@6 39
    return binary_op(a, b, lambda a: a.__ge__, lambda b: b.__le__)
paul@6 40
paul@6 41
def gt(a, b):
paul@762 42
    if is_int(a) and is_int(b):
paul@762 43
        return int_gt(a, b)
paul@6 44
    return binary_op(a, b, lambda a: a.__gt__, lambda b: b.__lt__)
paul@6 45
paul@6 46
def le(a, b):
paul@762 47
    if is_int(a) and is_int(b):
paul@762 48
        return int_le(a, b)
paul@6 49
    return binary_op(a, b, lambda a: a.__le__, lambda b: b.__ge__)
paul@6 50
paul@6 51
def lt(a, b):
paul@762 52
    if is_int(a) and is_int(b):
paul@762 53
        return int_lt(a, b)
paul@6 54
    return binary_op(a, b, lambda a: a.__lt__, lambda b: b.__gt__)
paul@6 55
paul@6 56
def ne(a, b):
paul@762 57
    if is_int(a) and is_int(b):
paul@762 58
        return int_ne(a, b)
paul@278 59
    return binary_op(a, b, lambda a: a.__ne__, lambda b: b.__ne__, True)
paul@6 60
paul@6 61
# vim: tabstop=4 expandtab shiftwidth=4