25. Реализовать бесконечный счётчик с ограничением

Условие задачи Реализовать функцию counter(), которая работает как генератор чисел. Если в функцию передано число n, генератор должен выдавать числа от 0 до n - 1. Если аргумент не передан, генератор должен работать бесконечно, начиная с 0, пока выполнение не будет остановлено вручную.

# Сделать бесконечный счётчик
def counter:
    ...


for a in counter(5):
    print(a, end=' ')  # -> [0, 1, 2, 3, 4]
print()


for a in counter():
    print(a, end=' ')  # -> [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
    if a == 9:
        break