当前位置：Gxlcms >
数据库问题 >
Python+Selenium+Mysql（动态获取数据，数据库交互）
Python+Selenium+Mysql（动态获取数据，数据库交互）
时间：2021-07-01 10:21:17
帮助过：22人阅读
!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()
Python+Selenium+Mysql（动态获取数据，数据库交互）
标签：cti har common turn partial class user 元组 语句
