"""
Convert numbers from base 10 integers to base X strings and back again.
Sample usage:
>>> base20 = BaseConverter('0123456789abcdefghij')
>>> base20.from_int(1234)
'31e'
>>> base20.to_int('31e')
1234
"""


class BaseConverter(object):
    decimal_digits = "0123456789"

    def __init__(self, digits):
        self.digits = digits

    def from_int(self, i):
        return self.convert(i, self.decimal_digits, self.digits)

    def to_int(self, s):
        return int(self.convert(s, self.digits, self.decimal_digits))

    @staticmethod
    def convert(number, fromdigits, todigits):
        # Based on http://code.activestate.com/recipes/111286/
        if len(str(number)) == 0:
            return 0

        if str(number)[0] == "-":
            number = str(number)[1:]
            neg = 1
        else:
            neg = 0

        # make an integer out of the number
        x = 0
        for digit in str(number):
            x = x * len(fromdigits) + fromdigits.index(digit)

        # create the result in base 'len(todigits)'
        if x == 0:
            res = todigits[0]
        else:
            res = ""
            while x > 0:
                digit = x % len(todigits)
                res = todigits[digit] + res
                x = int(x / len(todigits))
            if neg:
                res = "-" + res
        return res


# base2 = BaseConverter('01')
# base16 = BaseConverter('0123456789ABCDEF')
# base36 = BaseConverter('0123456789abcdefghijklmnopqrstuvwxyz')
# base62 = BaseConverter(
#     '0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz'
# )

pid = BaseConverter('0123456789ABCDEF')

if __name__ == "__main__":
    base20 = BaseConverter('0123456789abcdefghij')
    print(base20)
    print(base20.from_int(1234))
    print(base20.to_int('31e'))
