python实现盲盒抽奖功能(减库存)

#猜盲盒
import random
#用字典嵌套字典的形式存储所有奖品信息
w = {
    '一等奖': {
        '电饭煲': 2,
        '空调': 1,
        '洗衣机': 1
    },
    '二等奖': {
        '球鞋': 2,
        '手表': 5,
        '滑板': 3
    },
    '三等奖': {
        '对联': 4,
        '手袋': 2,
        '钙片': 3
    }
}


def lottery_plus(level=None):
    
    if level == '一等奖': 
        w_num = w.get('一等奖')  # 库存字典 {'电饭煲':2, '空调':1, '洗衣机':1}
        jackpot = []  # 奖池：谁的库存不为0进奖池 ['电饭煲','空调','洗衣机']
        # 谁的库存不为0进奖池
        for k in w_num.keys():  # 遍历['电饭煲','空调','洗衣机']
            if w_num.get(k) != 0:  # 如果库存不为0
                jackpot.append(k)  # 进奖池 [].append('电饭煲')
        try:
            result = random.choice(jackpot) #随机抽取一个奖品
            # 库存-1
            w.get('一等奖')[result] -= 1
            print('一等奖池：', end=' ')
            print(jackpot)
            return '恭喜获得一等奖' + result
        except:
            return '很遗憾，一等奖没有了'  #random.choice([]) 传入空的列表时会报错，列表为空时说明奖品已经都抽走了，所有奖品库存为零
        
    elif level == '二等奖':  
        w_num = w.get('二等奖')  
        jackpot = []  
        # 谁的库存不为0进奖池
        for k in w_num.keys():  
            if w_num.get(k) != 0:
                jackpot.append(k)  
        try:
            result = random.choice(jackpot)
            # 库存-1
            w.get('二等奖')[result] -= 1
            print('二等奖池：', end=' ')
            print(jackpot)
            return '恭喜获得二等奖' + result
        except:
            return '很遗憾，二等奖没有了'
        
    elif level == '三等奖': 
        w_num = w.get('三等奖')  
        jackpot = []  
        # 谁的库存不为0进奖池
        for k in w_num.keys():  
            if w_num.get(k) != 0: 
                jackpot.append(k)  
        try:
            result = random.choice(jackpot)
            # 库存-1
            w.get('三等奖')[result] -= 1
            print('三等奖池：', end=' ')
            print(jackpot)
            return '恭喜获得三等奖' + result
        except:
            return '很遗憾，三等奖没有了'
    else:
        return '没有获奖'