Python+Selenium+Mysql（动态获取数据，数据库交互）

Posted 2020-09-25

一、创建数据库连接

#!coding:utf-8
import pymysql
‘‘‘
Python3之后不再支持MySQLdb的方式进行访问mysql数据库；
可以采用pymysql的方式

连接方式：
    1、导包
        import pymysql
    2、打开数据库连接
        conn = pymysql.connect(host=‘10.*.*.*‘,user=‘root‘,password=‘123456‘,db=‘self_dev‘,charset=‘utf8‘,cursorclass=pymysql.cursors.DictCursor)
        备注：其中cursorclass=pymysql.cursors.DictCursor 可有可无，配置的是每个字段的展示方式，按照字典的形式进行展示（方便通过列名进行访问），默认元组形式。
    3、使用cursor()方法获取操作游标 
        cur = connection.cursor()
    4、SQL 查询语句
        sql = "SELECT * FROM table t where t.name=‘政协‘"
    5、执行SQL语句
        cur.execute(sql)
    6、获取所有记录列表
        rows = cur.fetchall()
    7、输出

    8、关闭数据库连接
        connection.close()

‘‘‘
class dbClection:
    connection = pymysql.connect(host=‘10.*.*.*‘,
        user=‘root‘,
        password=‘123456‘,
        db=‘self_dev‘,
        charset=‘utf8‘,
        cursorclass=pymysql.cursors.DictCursor)
    def getOrgid(self):
        cur = self.connection.cursor()

        sql = "SELECT t.id FROM org t where t.name=‘自动化添加(行政区域)‘ and t.status=‘1‘"
        try:
            cur.execute(sql)
            results = cur.fetchall()

            for row in results:
                org_id = row[‘id‘]
                return org_id
                #print(org_id)
        except:
            print(‘Error:unable to fetch data‘)

        self.connection.close()

‘‘‘
if __name__ == ‘__main__‘:
    db = dbClection()
    res = db.getOrgid()
    print(res)
‘‘‘

二、引用上述创建的dbClection下的getOrgid（）方法

#!coding:utf-8
from selenium import webdriver
from time import sleep
import unittest
from selenium.webdriver.common.action_chains import ActionChains
from selenium.webdriver.support.ui import Select
from mySql import dbClection


----------主体省略----------------

def test_add_fenwei(self):
        u‘‘‘登录‘‘‘
        username = ‘org-admin‘
        password = ‘123456‘
        self.login(username,password) 
        sleep(2)
        #进入组织管理界面
        self.dr.get(self.org_url)
        sleep(2)
        db = dbClection()
        xzjd_org = db.getOrgid()
        # print(xzjd_org)
        self.by_xpath(‘//*[@id="%s"]/i‘%xzjd_org).click()
        sleep(2)
        self.dr.find_element_by_partial_link_text(‘自动化添加(地域)‘).click()
        sleep(2)
        self.by_id(‘addDept‘).click()
        sleep(3)
        self.by_id(‘orgName_add‘).send_keys(‘自动化添加(分类)‘)
        sleep(2)
        Select(self.by_id("govDeptType_add")).select_by_visible_text("分类")
        sleep(2)
        self.by_id(‘saveDeptAdd‘).click()

　　备注：变量在使用在页面定位中时，需要进行%格式化，否则无法传入。
　　　　　　例如：
　      db = dbClection()

        xzjd_org = db.getOrgid()
        # print(xzjd_org)
        self.by_xpath(‘//*[@id="%s"]/i‘%xzjd_org).click()