通过读取excel数据和mysql数据库数据做对比（二）-代码编写测试

通过上一步，环境已搭建好了。

下面开始实战，

首先，编写链接mysql的函数conn_sql.py

import pymysql

def sql_conn(u,pwd,h,db):
    conn=pymysql.connect(user=u,passwd=pwd,host=h,db=db)
    #print("连接数据库"+db+"成功了！！")
    return conn

在编写，查询数据库的语句:sql.py

import pymysql
import conn_sql

def dd_cz(dd_h):
    conn=conn_sql.sql_conn("数据库连接名称","数据库密码","数据库ip","数据库名称")
    cursor=conn.cursor()
    cursor.execute("select count(id) from user where ordnum=%s",dd_h)
    data=cursor.fetchone()
    #print('data=',data)
    return data

注意点：（在测试时，有一个问题就是，在cursor.execute("select count(id) from user where ordnum=%s" % dd_h) 这句话中的参数引导时，如果换成%号，会出现数据库报错。所以一定要用，）

在创建zx_duibi.py

import xlrd,sys
import xdrlib
import os
from sql import dd_cz 

#打开Excel文件读取数据
def open_excel(file="1.xls"):
    try:
        data=xlrd.open_workbook(file)
        print("open sucess!")
        return data
    except Exception:
        print(str(e))

#根据索引获取Excel表格中的数据
#参数:file：Excel文件路径     colnameindex：表头列名所在行的索引
#by_index：表的索引

def excel_table_byindex(file="1.xls",colnameindex=0,by_index=0):
    #打开表
    data=open_excel(file)
    #获取工作表
    table=data.sheets()[by_index]
    nrows=table.nrows #行数
    #获取某一行数据
    colnames=table.row_values(colnameindex)
    list=[]
    #3跳转行数
    for rownum in range(1,nrows):
        #获取某一行的值
        row=table.row_values(rownum)
        if row:
            col_l={}
            col_l[colnames[3]]=row[3]
            list.append(col_l)
    return list

def main():
    tables=excel_table_byindex()
    #excel总数
    i=0
    #不存在的
    j=0
    for row in tables:
        #根据exl表中的数据，查询数据库中的数据
        ddh=dd_cz(row["列标题"])
        #列标题是否存在
        ddh_cz=ddh[0]

        if ddh_cz == 0:
            print("列标题：%s，不存在"%(row["列标题"]))
            j+=1
        else:
            print("存在:%s"%(row["列标题"]))
        i+=1
    print("总数为:%s" % i )
    print("不存在为:%s"%j)

if __name__=="__main__":
    main()

至此根据excel的数据和数据中的数据做对比的程序就ok了

参考文档：

mysql操作为：好人

excel操作为：好人