0914. Rock Paper Scissors!

Rock Paper Scissors

Let's play! You have to return which player won! In case of a draw return Draw!.

Examples(Input1, Input2 --> Output):

"scissors", "paper" --> "Player 1 won!"
"scissors", "rock" --> "Player 2 won!"
"paper", "paper" --> "Draw!"

Solution:

from typing import Dict


def check_draw(func):
    def wrapper(p1, p2):
        table = {"s":"p","r":"s","p":"r"}
        return p1 == p2 and "Draw!" or func(p1, p2, table)
    return wrapper        

@check_draw
def rps(p1: str, p2: str, table: Dict[str, str]):
    return table[p1[0]] == p2[0] and "Player 1 won!" or "Player 2 won!"