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

一、创建数据库连接

  '''
  #!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()