为不同会员消费者买单

想问一下这道Python题该怎么做，不太明白该如何去做，应该怎么样的方式来解答

sum = eval(input())
flag = input()
if flag=='2':
    print(sum*0.9)
elif flag=='1':
    print(sum*0.95)
elif flag=='0':
    print(sum*0.99)