Python制作一個(gè)隨機(jī)抽獎(jiǎng)小工具的實(shí)現(xiàn)

nameList_column = [
    [sg.Text('人員名單：')],
    [sg.Listbox(values=[], size=(20, 10), key='nameList')],
]
result_column = [
    [sg.Text('中獎(jiǎng)記錄：')],
    [sg.Multiline('', size=(48, 10), key='result', text_color='DeepPink')],
]

# 主題設(shè)置
sg.theme('SystemDefaultForReal')

# 布局設(shè)置
layout = [[sg.Text('選擇參與抽獎(jiǎng)人員名單文件:', font=('微軟雅黑', 12)), sg.InputText('', key='_file', size=(50, 1), font=('微軟雅黑', 10), enable_events=True), sg.FileBrowse('打開', file_types=(('Text Files', '*.xlsx'),), size=(10, 1), font=('微軟雅黑', 11))],
          [sg.Frame(layout=[
              [sg.Text('本輪獎(jiǎng)項(xiàng):', font=('微軟雅黑', 12)), sg.Combo(['特等獎(jiǎng)', '一等獎(jiǎng)', '二等獎(jiǎng)', '三等獎(jiǎng)', '四等獎(jiǎng)', '五等獎(jiǎng)', '六等獎(jiǎng)'], font=('微軟雅黑', 10), default_value='特等獎(jiǎng)', size=(15, 5), key='_type'),
               sg.Text('本輪人數(shù):', font=('微軟雅黑', 12)), sg.InputText('5', key='_num', size=(38, 1), font=('微軟雅黑', 10))],
          ],
              title='抽獎(jiǎng)設(shè)置', title_color='red', relief=sg.RELIEF_SUNKEN, tooltip='請(qǐng)進(jìn)行抽獎(jiǎng)設(shè)置后再開始抽獎(jiǎng)')],
          [sg.Multiline(size=(48, 5), font=(
              '微軟雅黑', 18), text_color='Blue', key='luckyName', justification='center')],
          [sg.Column(nameList_column), sg.Column(result_column)],
          [sg.Text('操作說明:', font=('微軟雅黑', 12))],
          [sg.Text('①先選擇參與抽獎(jiǎng)的人員名單xlsx文件,人員名單文件包含ID和name兩個(gè)字段\n②獲獎(jiǎng)名單將存在小工具所在文件夾，重置會(huì)刪除歷史記錄文件', font=('微軟雅黑', 10)),
           sg.Text('', font=('微軟雅黑', 12), size=(5, 1)),
           sg.Button('開始抽獎(jiǎng)', font=('微軟雅黑', 12), button_color='Orange'),
           sg.Button('結(jié)束', font=('微軟雅黑', 12), button_color='red'),
           sg.Button('重置', font=('微軟雅黑', 12), button_color='red'), ],
          ]

# 創(chuàng)建窗口
window = sg.Window('抽獎(jiǎng)小工具，作者@微信公眾號(hào)：可以叫我才哥', layout,
                   font=('微軟雅黑', 12), default_element_size=(50, 1))

def nameList(window):
    fileName = values['_file']
    try:
        wb = openpyxl.load_workbook(fileName)
        active_sheet = wb.active
        names = [cell_object.value for cell_object in list(active_sheet.columns)[1]][1:]
        ids = [cell_object.value for cell_object in list(active_sheet.columns)[0]][1:]
        names = [name+'_'+str(id_) for name, id_ in zip(names, ids)]
        window['nameList'].update(names)
        return names
    except:
        sg.popup('請(qǐng)選擇正確格式的的人員名單文件', title='提示',)

def Result(window, names):
    global is_run, luckyNames
    _type = values['_type']                # 本輪獎(jiǎng)項(xiàng)類型
    _num = int(values['_num'])             # 本輪人數(shù)

    while True:
        randomName = random.sample(names, k=_num)
        luckyName = '   '.join(randomName)
        window['luckyName'].update(luckyName)

        if not is_run:
            headers = ['獎(jiǎng)項(xiàng)', '名單']
            toCsv(headers, [_type]*len(randomName), randomName, lucky)
            luckyNames = luckyNames + _type+' : '+luckyName+'\n\n'
            window['result'].update(luckyNames)
            return
        time.sleep(0.088)

def toCsv(headers, col1, col2, file):
    # 存在則追加，不存在則新建
    if os.path.exists(lucky):
        with open(lucky, 'a', encoding='utf_8_sig', newline='') as csvfile:
            writer = csv.writer(csvfile)
            writer.writerows(zip(col1, col2))
    else:
        with open(lucky, 'w', encoding='utf_8_sig', newline='') as csvfile:
            writer = csv.writer(csvfile)
            writer.writerow(headers)
            writer.writerows(zip(col1, col2))

# 初始狀態(tài)
is_run = False
luckyNames = ''

# 事件循環(huán)
while True:
    event, values = window.read()
    if event in (None, '關(guān)閉程序'):
        break
    if event == '_file':
        nameList(window)

    if event == '開始抽獎(jiǎng)':
        if is_run:
            sg.popup('抽獎(jiǎng)進(jìn)行中，無需重復(fù)操作......', title='提示')
            continue
        try:
            names = nameList(window)               # 人員名單
            _num = int(values['_num'])             # 本輪人數(shù)
            lucky = '中獎(jiǎng)名單.csv'                 # 中獎(jiǎng)名單
            if os.path.exists(lucky):
                with open('中獎(jiǎng)名單.csv', 'r', encoding='utf_8_sig') as f:
                    reader = csv.reader(f)
                    selectedNames = set([i[1] for i in reader][1:])
                names_set = set(names)-selectedNames
            else:
                names_set = set(names)
            if len(names_set) >= _num:
                is_run = True
                _thread.start_new_thread(Result, (window, names_set))
            else:
                sg.popup(
                    f'請(qǐng)選擇正確本輪抽獎(jiǎng)人數(shù)(當(dāng)前 {len(names_set)} 個(gè)未中獎(jiǎng)人數(shù))', title='提示')
        except:
            sg.popup('請(qǐng)選擇正確本輪抽獎(jiǎng)人數(shù)(別超過總?cè)藬?shù)哦)', title='提示')
    elif event == '結(jié)束':
        is_run = False
    elif event == '重置':
        if is_run:
            sg.popup('抽獎(jiǎng)進(jìn)行中，請(qǐng)等待抽獎(jiǎng)結(jié)束后重置...', title='提示')
            continue
        yes_no = sg.popup_yes_no(
            '重置會(huì)清楚歷史數(shù)據(jù)，是否執(zhí)行此操作？？', text_color='red', title='提示')
        if yes_no == 'Yes':
            try:
                os.remove(lucky)
                luckyNames = ''
                window['result'].update(luckyNames)
                window['luckyName'].update(luckyNames)
                sg.popup('抽獎(jiǎng)歷史記錄已被重置......', title='提示')
            except:
                sg.popup('無抽獎(jiǎng)歷史記錄......', title='提示')
window.close()