我要訂便當 (1): 用 Python + Selenium 控制瀏覽器取得訂單

from selenium import webdriver

url = 'https://dinbendon.net/do/login'
driver = webdriver.Chrome()
driver.get(url)  # 連線到訂便當頁面

time.sleep(2) # 演一下

username = "Hello"
password = "password"

# 輸入帳密
driver.find_element_by_name("username").send_keys(username)
driver.find_element_by_name("password").send_keys(password)

# 輸入驗證碼
ques = driver.find_elements_by_class_name("alignRight")[2].text # 有點強硬地拿到整串問題
temp = re.findall(r"\d+\.?\d*", ques) # 用正規表達式把數字取出
a = int(temp[0])
b = int(temp[1])
c = a + b
driver.find_element_by_name("result").send_keys(c)

# 提交表單
driver.find_element_by_name("submit").click() 

# 取出訂單表格列
rows = driver.find_elements_by_css_selector(
    "div#inProgressBox>table>tbody>tr")

if len(rows) == 0:
    driver.close()
    return list()
    
# 取出每一列資料的文字
bandons = [list(map(getText, row.find_elements_by_css_selector(
    "td>div>a>span"))) for row in rows]  
driver.close()

# 做成一張表
tableHeader = ['人數', '發起人', '目標']
bandons_df = pandas.DataFrame(bandons, columns=tableHeader)

from selenium import webdriver
import re
import time
import pandas

# 自動檢查團購便當網
def main():
    url = 'https://dinbendon.net/do/login'
    order = fetch_bandon(url)
    print_order(order)

def fetch_bandon(url, username="guest", password="guest"):
    ''' 開啟瀏覽器並連線到便當網取得資料 '''

    driver = webdriver.Chrome()
    driver.get(url)  # 連線到訂便當頁面
    time.sleep(1) # 演一下

    # 輸入帳密
    driver.find_element_by_name("username").send_keys(username)
    driver.find_element_by_name("password").send_keys(password)

    # 輸入驗證碼
    ques = driver.find_elements_by_class_name("alignRight")[2].text
    temp = re.findall(r"\d+\.?\d*", ques)
    answer = int(temp[0]) + int(temp[1])
    driver.find_element_by_name("result").send_keys(answer)

    # 提交表單
    driver.find_element_by_name("submit").click()
    time.sleep(1)

    # 取出訂單表格列
    rows = driver.find_elements_by_css_selector(
        "div#inProgressBox>table>tbody>tr")

    if len(rows) == 0:
        driver.close()
        return list()
        
    # 取出每一列資料的文字
    bandons = [list(map(getText, row.find_elements_by_css_selector(
        "td>div>a>span"))) for row in rows]  
    driver.close()

    # 做成一張表
    tableHeader = ['人數', '發起人', '目標']
    bandons_df = pandas.DataFrame(bandons, columns=tableHeader)

    return bandons_df


def print_order(data):
    '''列印訂單資料，看起來整齊一點'''
    for index, row in data.iterrows():
        if row is not None:
            print('({hcount:>4s}) {orderer}: {order:<40s}'.format(
                orderer = str(row['發起人']),
                order = str(row['目標']),
                hcount = str(row['人數'])))


def getText(x):
    return x.text


if __name__ == '__main__':
    main()

options = webdriver.ChromeOptions()
options.add_argument('headless')
driver = webdriver.Chrome(options=options)