0903. L1: Bartender, drinks!

Complete the function that receives as input a string, and produces outputs according to the following table:

Input	Output
"Jabroni"	"Patron Tequila"
"School Counselor"	"Anything with Alcohol"
"Programmer"	"Hipster Craft Beer"
"Bike Gang Member"	"Moonshine"
"Politician"	"Your tax dollars"
"Rapper"	"Cristal"
anything else	"Beer"

Note: anything else is the default case: if the input to the function is not any of the values in the table, then the return value should be "Beer".

Make sure you cover the cases where certain words do not show up with correct capitalization. For example, the input "pOLitiCIaN" should still return "Your tax dollars".

Solution:

from dataclasses import dataclass


@dataclass
class UserData:
    id: int
    profession: str

@dataclass
class RequestData:
    user_id: int
    drink: str


USER_DATA = [
    UserData(id=1, profession="Jabroni"),
    UserData(id=2, profession="School Counselor"),
    UserData(id=3, profession="Programmer"),
    UserData(id=4, profession="Bike Gang Member"),
    UserData(id=5, profession="Politician"),
    UserData(id=6, profession="Rapper"),
]

REQUEST_DATA = [
    RequestData(user_id=1, drink="Patron Tequila"),
    RequestData(user_id=2, drink="Anything with Alcohol"),
    RequestData(user_id=3, drink="Hipster Craft Beer"),
    RequestData(user_id=4, drink="Moonshine"),
    RequestData(user_id=5, drink="Your tax dollars"),
    RequestData(user_id=6, drink="Cristal"),
]


def get_user_id(func):
    def wrapper(profession: str):
        user_id = 0
        for user in USER_DATA:
            if user.profession.casefold() == profession.casefold():
                user_id = user.id
        return func(user_id)
    return wrapper

@get_user_id
def get_drink_by_profession(user_id: int) -> str:
    drink = 'Beer'
    for request in REQUEST_DATA:
        if request.user_id == user_id:
            drink = request.drink
    return drink