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 '没有获奖'

下面是调用函数的结果：