# Copyright (c) 2017 The Regents of the University of Michigan.
# All Rights Reserved. Licensed according to the terms of the Revised
# BSD License. See LICENSE.txt for details.

class CheckDigitNumber:

    def __init__ (self, number = None):
        self.__set_number(number)

    def get_check_digit (self):
        if self:
            return self.generate_from_int(self.number)

        else:
            return None

    def has_valid_check_digit (self):
        if self:
            digit = self.number % 10
            static = self.number // 10
            return digit == self.generate_from_int(static)

        else:
            return False

    def __bool__ (self):
        return self.number is not None

    def __repr__ (self):
        return "<{} {}>".format(self.__class__.__name__,
                                repr(self.number))

    def __set_number (self, number):
        if isinstance(number, int):
            self.number = number

        elif isinstance(number, str):
            self.__try_to_extract_number_from_str(number)

        else:
            self.number = None

    def __try_to_extract_number_from_str (self, number):
        try:
            self.number = int(number)

        except ValueError:
            self.number = None

# Copyright (c) 2017 The Regents of the University of Michigan.
# All Rights Reserved. Licensed according to the terms of the Revised
# BSD License. See LICENSE.txt for details.

class CheckDigitNumber:

    def __init__ (self, number = None):
        self.__set_number(number)

    def generate_from_int (self, n):
        raise NotImplementedError

    def get_check_digit (self):
        if self:
            return self.generate_from_int(self.number)

        else:
            return None

    def has_valid_check_digit (self):
        if self:
            digit = self.number % 10
            static = self.number // 10
            return digit == self.generate_from_int(static)

        else:
            return False

    def __bool__ (self):
        return self.number is not None

    def __repr__ (self):
        return "<{} {}>".format(self.__class__.__name__,
                                repr(self.number))

    def __set_number (self, number):
        if isinstance(number, int):
            self.number = number

        elif isinstance(number, str):
            self.__try_to_extract_number_from_str(number)

        else:
            self.number = None

    def __try_to_extract_number_from_str (self, number):
        try:
            self.number = int(number)

        except ValueError:
            self.number = None

Make it clear that the user must implement generate_from_int
Make it clear that the user must implement generate_from_int
