lbry-sdk/lbrynet/dht/distance.py

from binascii import hexlify

from lbrynet.dht import constants
import sys
if sys.version_info > (3,):
    long = int


class Distance:
    """Calculate the XOR result between two string variables.

    Frequently we re-use one of the points so as an optimization
    we pre-calculate the long value of that point.
    """

    def __init__(self, key):
        if len(key) != constants.key_bits / 8:
            raise ValueError("invalid key length: %i" % len(key))
        self.key = key
        self.val_key_one = long(hexlify(key), 16)

    def __call__(self, key_two):
        val_key_two = long(hexlify(key_two), 16)
        return self.val_key_one ^ val_key_two

    def is_closer(self, a, b):
        """Returns true is `a` is closer to `key` than `b` is"""
        return self(a) < self(b)

    def to_contact(self, contact):
        """A convenience function for calculating the distance to a contact"""
        return self(contact.id)