fizbuz.py

from dataclasses import dataclass
from typing import Callable


def generate_fizbuz_answers(answers_count: int) -> list[str]:
    results = []
    for number in range(1, answers_count + 1):
        results.append(compile_result(number))
    return results


@dataclass
class FizzBuzzResult:
    fizz: int = 0
    buzz: int = 0

    def __add__(self, other: 'FizzBuzzResult') -> 'FizzBuzzResult':
        fizzes = self.fizz + other.fizz
        buzzes = self.buzz + other.buzz

        return FizzBuzzResult(fizz=fizzes, buzz=buzzes)


def get_fizbuz(number: int) -> FizzBuzzResult:
    result = FizzBuzzResult()

    if number % 3 == 0:
        result.fizz += 1
    if number % 5 == 0:
        result.buzz += 1

    return result


def get_extended_fizbuz(number: int) -> FizzBuzzResult:
    result = FizzBuzzResult()
    if '3' in str(number):
        result.fizz += 1
    if '5' in str(number):
        result.buzz += 1
    return result


def compile_result(number):
    scoring_function: list[Callable[[int], FizzBuzzResult]] = [get_fizbuz, get_extended_fizbuz]
    scores = list(map(lambda x: x(number), scoring_function))
    summary_fizbuz = FizzBuzzResult()
    for score in scores:
        summary_fizbuz += score

    default = str(number)
    result = 'Fizz' * summary_fizbuz.fizz + 'Buzz' * summary_fizbuz.buzz
    result = result or default  # use number themselves if no rule for given number

    return result


def main():
    print(*generate_fizbuz_answers(100), sep='\n')


if __name__ == '__main__':
    main()