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")

 

生成文档的部分截图：