Есть задача: написать функцию, которая принимает первым аргументом строку и вторым аргументом размер чанка
Я решил с помощью генератора разбивать строку и возвращать по кускам, всё хорошо, но последний элемент None. Как это исправить?
def wrap(string, max_width):
def gen_letters(string, max_width):
count = 0
res_gen = ''
for index, i in enumerate(string):
count += 1
res_gen += i
if count == max_width:
count = 0
tmp_res = res_gen
res_gen = ''
yield tmp_res
elif count < max_width and index == len(string)-1:
yield res_gen
for i in gen_letters(string, max_width):
print(i)
Входные данные:
string: ABCDEFGHIJKLIMNOQRSTUVWXYZ
max_width: 4
Результат:
ABCD
EFGH
IJKL
IMNO
QRST
UVWX
YZ
None