Source code for ldaptor.numberalloc

"""Find an available uidNumber/gidNumber/other similar number."""

from ldaptor.protocols import pureldap


class freeNumberGuesser:
    def __init__(self, makeAGuess, min=None, max=None):
        self.makeAGuess = makeAGuess
        self.min = min
        if self.min is None:
            self.min = 0
        self.max = max

    def startGuessing(self):
        d = self.makeAGuess(self.min)
        d.addCallback(self._nextGuess, self.min)
        return d

    def _nextGuess(self, found, lastGuess):
        if found:
            self.min = lastGuess
        else:
            self.max = lastGuess

        if self.max == self.min or self.max == self.min + 1:
            return self.max

        max = self.max
        if max is None:
            max = self.min + 1000

        guess = (max + self.min) // 2
        d = self.makeAGuess(guess)
        d.addCallback(self._nextGuess, guess)
        return d


class ldapGuesser:
    def __init__(self, ldapObject, numberType):
        self.numberType = numberType
        self.ldapObject = ldapObject

    def guess(self, num):
        d = self.ldapObject.search(
            filterObject=pureldap.LDAPFilter_equalityMatch(
                attributeDesc=pureldap.LDAPAttributeDescription(value=self.numberType),
                assertionValue=pureldap.LDAPAssertionValue(value=str(num)),
            ),
            sizeLimit=1,
        )
        d.addCallback(lambda results: len(results))
        return d


def getFreeNumber(ldapObject, numberType, min=None, max=None):
    g = freeNumberGuesser(ldapGuesser(ldapObject, numberType).guess, min=min, max=max)
    return g.startGuessing()