Selenium的筆記+Python代碼自動生成docx文檔（含表格、插入圖片）

1.安裝

pip install selenium

2.下載web驅動，配置到環境變量或者Python目錄下。

https://github.com/mozilla/geckodriver/releases  【firefox】

http://chromedriver.storage.googleapis.com/index.html  【Chrome】

https://blog.csdn.net/huilan_same/article/details/51896672 【Chrome與Chrome驅動對應版本】

我的Python目錄

3.元素定位：

通過id定位：find_element_by_id()

通過name定位：find_element_by_name()

通過class屬性定位：find_element_by_class_name()

通過tag定位：find_element_by_tag_name()

通過link定位：find_element_by_link_text()

通過xpath定位：find_element_by_xpath()

4.操作：

Alert:

切到alert:driver.switch_to_alert()

取消：x.dismiss()

接受：x.accept()

Frame:

切換到iframe:driver.switch_to_frame()

釋放：driver.switch_to_default_content()

等待：

1、強制等待3秒：time.sleep(3)

2、顯示等待：每隔0.5秒檢測，是否執行下一步，最長等待300秒超時。

WebDriverWait(driver=self.driver,timeout=300,poll_frequency=0.5,ignored_exceptions=None).until( EC.text_to_be_present_in_element((By.XPATH, ‘//*[@id=“VolumeTable”]/tbody/tr[1]/td[4]/label’), u‘可用’))

5.實現了報銷測試費文檔自動生成。

# -*- coding: utf-8 -*-
import pymysql
from docx import Document
from docx.shared import Inches
import conf
import time
from selenium import webdriver
import datetime
def get_file_name():
    today = datetime.date.today()# 獲取今天日期
    first_day = datetime.date(day=1, month=today.month, year=today.year)#獲取本月第1天的日期
    lastMonth = first_day - datetime.timedelta(days=1)#獲取上個月的最後1天日期
    year = str(lastMonth.year)#年份
    month = str(lastMonth.month)#上個月的月份
    doc_path = u'E:\測試訂單明細.%s.%s.docx'%(year,month)
    return doc_path

def get_data(*sql_list):#根據sql語句，獲取數據庫的數據
    conn = pymysql.connect(conf.test_dbhost,conf.test_user,conf.test_passd,conf.test_dbname,port=3306,charset="utf8")
    cur = conn.cursor()
    for sql in sql_list:
        cur.execute(sql)
    conn.commit()
    results = cur.fetchall()
    cur.close()
    conn.close()
    return results

def get_order_id(outer_order_id):#根據outer_order_id查詢 order_id
    sql = "SELECT order_id FROM oh_order_info WHERE outer_order_id='%s'"%outer_order_id
    order_id = get_data(sql)
    return order_id[0][0]

def get_original_price(outer_order_id):#根據outer_order_id 去查詢order_product表，獲取訂單商品信息，返回此訂單商品總金額
    order_id = get_order_id(outer_order_id)
    sql_product = "SELECT original_price,qty FROM oh_order_product WHERE order_id= '%s'"%order_id
    original_price_data = get_data(sql_product)
    original_price = float(0)
    for i in original_price_data:
        original_price = original_price + float(i[0]*i[1])/100
    return original_price

def get_order_fee(outer_order_id):#根據outer_order_id 獲取訂單各項費用
    sql = "SELECT outer_shipping_amount,real_shipping_amount,promotion_amount,discount_amount," \
          "outer_act_amount,outer_packing_amount,cash_amount,sku_amount FROM oh_order_info " \
          "WHERE outer_order_id='%s' "%outer_order_id
    order_fee = get_data(sql)
    shipping_amount   = float(order_fee[0][0])/100 + float(order_fee[0][1])/100 # 配送費
    promotion_quanshi = float(order_fee[0][2])/100 + float(order_fee[0][3])/100 # 全時優惠
    promotion_outer   = float(order_fee[0][4])/100                              # 三方優惠
    outer_packing     = float(order_fee[0][5])/100                              # 包裝費
    cash_amount       = float(order_fee[0][6])/100                              # 差異金額
    original_price    = float(order_fee[0][7])/100                              # 使用order_info表的sku_amount字段最爲商品總金額
    #original_price    = get_original_price(outer_order_id)-cash_amount         # 商品原總額
    paid_amount = original_price+shipping_amount+outer_packing-promotion_quanshi-promotion_outer#用戶實付金額
    return original_price,shipping_amount,paid_amount

def new_doc(fee_data,doc_path,fee):#新建一個word文檔,寫入彙總表的數據
    document = Document()
    p_total = document.add_paragraph()
    r_total = p_total.add_run(u'測試訂單費用匯總表：')
    r_total.font.bold = True
    table = document.add_table(1,5,style="Light List Accent 5")
    heading_cells = table.rows[0].cells
    heading_cells[0].text = u'序號'
    heading_cells[1].text = u'訂單號'
    heading_cells[2].text = u'商品總額'
    heading_cells[3].text = u'運費'
    heading_cells[4].text = u'實付金額'
    total = 0 #記錄總實付金額
    number = len(fee_data)-1
    for i in range(0,len(fee_data)):
        outer_order_id = str(fee_data[i][0]) # 獲取訂單號
        order_fee = get_order_fee(outer_order_id) #獲取訂單的商品總額、運費、實付金額
        cells = table.add_row().cells
        cells[0].text = str(i+1)          #寫入序號列數據
        cells[1].text = outer_order_id    #寫入訂單號列數據
        cells[2].text = str(order_fee[0]) #寫入商品總額列數據
        cells[3].text = str(order_fee[1]) #寫入運費列數據
        cells[4].text = str(order_fee[2]) #寫入實付金額列數據
        total = total + order_fee[2]
        if total > fee:#如果實付總額大於傳入的金額，終止寫入數據,並記錄序號
            number = i
            break
    print (number+1,"個訂單")
    total = str(float(total))
    document.add_paragraph(u'實付金額總計：' + total + u' 元。')#寫入實付總額
    document.add_page_break()#分頁
    p_detail = document.add_paragraph()
    r_detail = p_detail.add_run(u'測試訂單明細：')
    r_detail.font.bold = True
    #開始截圖
    browser = webdriver.Firefox()#使用Firefox瀏覽器
    #browser = webdriver.Chrome()#使用Chrome瀏覽器
    browser.set_window_size(1600, 1000)
    for i in range(0,number+1):#開始寫入訂單明細部分
        order_no = str(fee_data[i][0])
        order_fee = get_order_fee(order_no)
        row_str = str(i+1) + '.' + u'訂單號：' + order_no + u'實付金額：' + str(order_fee[2]) + u'元。'
        document.add_paragraph(row_str)# 寫入訂單號、實付數據
        url = "http://*.*.*.*/order_info?outer_order_id="+order_no
        browser.get(url)
        time.sleep(3)
        png_file_name = r'E:\png\%s.png'%order_no
        browser.save_screenshot(png_file_name)
        document.add_picture(png_file_name,width=Inches(6.0))#插入圖片,圖片縮放爲6英尺
    document.save(doc_path)
    browser.quit()

if __name__ == "__main__":
    #sql語句篩選實付金額在5元和39元之間的訂單
    sql = "SELECT outer_order_id FROM oh_order_info WHERE " \
      "order_create_time between '2018-08-11 9:00:00' and '2018-08-20 23:00:00' " \
      "AND paid_amount between '500' and '2900' AND order_status!=40 AND order_status!=60 ORDER BY id DESC"
    fee_data = get_data(sql)
    doc_path = get_file_name()
    fee = 475 #拉取多少元的訂單,大於此值則終止
    new_doc(fee_data,doc_path,fee)
    print ("done")

 

生成文檔的部分截圖：