Using a generator for tetris pieces

day17.py

pieces = [{3,4,5,6},
          {4, 3+1j, 4+1j, 5+1j, 4+2j},
          {3, 4, 5, 5+1j, 5+2j},
          {3, 3+1j, 3+2j, 3+3j},
          {3, 4, 3+1j, 4+1j}]


def make_piece():
    level = 0
    for piece in cycle(pieces):
        level = yield {part + (level * 1j) for part in piece}