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