ナベアツをつくろう

nabeatsu.py

#
# イテレータ上手く使えばナベアツを錬成できるのでは?
#

import sys
from typing import Optional, Union


class Nabeatsu:

    def __init__(self, start: int, end: int) -> None:
        self.current = start
        self.end = end

    def __iter__(self):
        return self

    def __next__(self) -> Union[int, str]:
        current = self.current

        if current + 1 > self.end:
            raise StopIteration

        self.current += 1

        # 3の倍数と3のつく数値の時だけアホになります
        if current > 0 and (current % 3 == 0 or str(current).find('3') >= 0):

            current_least_digit = int(str(current)[-1])

            current_shouts = [
                "ｩｩ-!", "ｨｨ--ｯ!!", "ｨ-!", "ｧｧｧ--ﾝ!!", "ｫｫ-ﾝ!!",
                "ｫ--!", "ｩｩ-!", "ｧｧ--!", "ｨ-!", "ｩ-!"
            ]
            return f"{current}{current_shouts[current_least_digit]}"

        return current


args = sys.argv


def value_or_none(list, index):
    try:
        return list[index]
    except IndexError:
        return None


start = int(value_or_none(args, 1) or "0")
end = int(value_or_none(args, 2) or "100") + 1

nabeatsu = Nabeatsu(start, end)
for n in nabeatsu:
    print(n)