IF文の条件分岐をもっと簡潔にしたい

Question

Pygameを用いた将棋プログラムを作成しています。
プログラムの中でどの駒がクリックされたか判定をするために下記のような条件分岐にしたのですが、分岐の数が多すぎて非常に見栄えの悪いコードになってしまいました。出来れば数行ほどにまとめたいのですが、どのように書くと良いか教えていただけませんでしょうか？

def ButtonAction(pieces):
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            pygame.quit()
            sys.exit()
        elif event.type == pygame.MOUSEBUTTONDOWN:
            if(pieces[0].button).collidepoint(event.pos):
                print(pieces[0].type)
            elif(pieces[1].button).collidepoint(event.pos):
                print(pieces[1].type)
            elif(pieces[2].button).collidepoint(event.pos):
                print(pieces[2].type)
            elif(pieces[3].button).collidepoint(event.pos):
                print(pieces[3].type)
            elif(pieces[4].button).collidepoint(event.pos):
                print(pieces[4].type)
            elif(pieces[5].button).collidepoint(event.pos):
                print(pieces[5].type)
            elif(pieces[6].button).collidepoint(event.pos):
                print(pieces[6].type)
            elif(pieces[7].button).collidepoint(event.pos):
                print(pieces[7].type)
            elif(pieces[8].button).collidepoint(event.pos):
                print(pieces[8].type)
            elif(pieces[9].button).collidepoint(event.pos):
                print(pieces[9].type)
            elif(pieces[10].button).collidepoint(event.pos):
                print(pieces[10].type)
            elif(pieces[11].button).collidepoint(event.pos):
                print(pieces[11].type)
            elif(pieces[12].button).collidepoint(event.pos):
                print(pieces[12].type)
            elif(pieces[13].button).collidepoint(event.pos):
                print(pieces[13].type)
            elif(pieces[14].button).collidepoint(event.pos):
                print(pieces[14].type)
            elif(pieces[15].button).collidepoint(event.pos):
                print(pieces[15].type)
            elif(pieces[16].button).collidepoint(event.pos):
                print(pieces[16].type)
            elif(pieces[17].button).collidepoint(event.pos):
                print(pieces[17].type)
            elif(pieces[18].button).collidepoint(event.pos):
                print(pieces[18].type)
            elif(pieces[19].button).collidepoint(event.pos):
                print(pieces[19].type)

Answer 1

forを使えば良いのでは？最初に使っているのでforはご存じですよね？
条件が真ならbreakでforから抜けます。

Comments

[GK.1225]

for文を使うことを忘れておりました....
解答者様のおかげでかなり簡潔なコードにすることが出来ました。ありがとうございます。

Answer 2

性能が気にならないなら、以下でもできます。
性能を気にするならotnさんのやり方の方が良いでしょう。

def ButtonAction(pieces):
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            pygame.quit()
            sys.exit()
        elif event.type == pygame.MOUSEBUTTONDOWN:
            print([piece.type for piece in pieces if piece.button.collidepoint(event.pos)][0])

Comments

[GK.1225]

こういった書き方があるのですね。とても勉強になりました、ありがとうございます。