import pymysql class MysqlDBUtil(): # 字典输出 def __init__(self, host, user, password, database, charset): # host = "localhost" # 指定访问的服务器,本地服务器指定“localhost”,远程服务器指定服务器ip地址 # port = 3306 # 指定端口号,范围在0-65535 # user = "root" # 用户名 # password = "123456" # 密码 # database = "sys" # 数据库的名字 # charset = "utf8" # 数据库的编码方式 # 建立连接 self.connect = pymysql.Connect(host=host, port=3306, user=user, passwd=password, db=database, charset=charset) # 数据库的查询 def query(self , sql): cursor = None res = None try: cursor = self.connect.cursor() cursor.execute(sql) res = cursor.fetchall() except Exception as e: print(str(e)) finally: cursor.close() return res # 数据库的增删改 def uid(self , sql): cursor = None try: cursor = self.connect.cursor() cursor.execute(sql) self.connect.commit() except Exception as e: print(str(e)) self.connect.rollback() finally: cursor.close() #数据库的关闭 def sql_close(self): self.connect.close() import pytest import allure import logging from utils.mysql_database_connection_utils import MysqlDBUtil import utils.mysql_database_connection_utils class TestChaxun(): def mysql_connect(self): self.conn = MysqlDBUtil("localhost", "root", "123456", "sys", "utf8") def teardown_class(self): self.conn.sql_close() def test1(self): data = self.conn.query("SELECT * from stu WHERE name = '张三' ") print(data)
时间: 2024-02-14 18:07:17 浏览: 88
这段代码中的问题可能是在TestChaxun类中的mysql_connect方法中,没有使用self关键字来定义conn属性。应该将mysql_connect方法中的 "self.conn" 更改为 "self.db",因为在类的初始化方法中已经定义了 "self.connect"。修改后的代码如下:
```
class TestChaxun():
def mysql_connect(self):
self.db = MysqlDBUtil("localhost", "root", "123456", "sys", "utf8")
def teardown_class(self):
self.db.sql_close()
def test1(self):
data = self.db.query("SELECT * from stu WHERE name = '张三' ")
print(data)
```
另外,你需要确保在运行测试之前已经安装了pytest和allure模块,并且将utils.mysql_database_connection_utils模块添加到PYTHONPATH中,以便测试代码能够正确地导入MysqlDBUtil类。
相关问题
解释每一句import pymysql class MyMysql: def __init__(self): self.connect = pymysql.connect( host=Mysql服务IP地址, port=Mysql数据库端口, user=数据库用户名, password=数据库密码, database=数据库名, charset='utf8' ) self.cursor = self.connect.cursor()import random from tools.config import MyMysql class UaPool: def __init__(self): database = MyMysql() db = database.connect cur = database.cursor sql = 'select useragent from ua_pool where types = %s and popularity != %s' cur.execute(sql, ('Windows', 'Uncommon')) data_d = cur.fetchall() desktop_list = [] for data_d_per in data_d: desktop_list.append(data_d_per[0]) self.desktop = desktop_list cur.execute(sql, ('Android', 'Uncommon')) data_m = cur.fetchall() mobile_list = [] for data_m_per in data_m: mobile_list.append(data_m_per[0]) self.mobile = mobile_list cur.close() db.close() def choose_ua(self, platform='desktop'): if platform == 'desktop': return random.choice(self.desktop) elif platform == 'mobile': return random.choice(self.mobile) else: return random.choice(self.desktop) if __name__ == '__main__': up = UaPool() print(up.choose_ua())
这段代码主要实现了从MySQL数据库中读取用户代理(User-Agent)池,并随机选择一个用户代理。具体解释如下:
1. `import pymysql`:导入 PyMySQL 模块,用于连接和操作 MySQL 数据库。
2. `class MyMysql:`:定义一个 MyMysql 类,用于连接 MySQL 数据库并返回连接对象和游标对象。
3. `def __init__(self):`:定义 MyMysql 类的构造函数,初始化 MySQL 数据库连接和游标对象。
4. `self.connect = pymysql.connect(...)`:连接 MySQL 数据库,需要指定数据库的 IP 地址、端口、用户名、密码、数据库名以及编码方式。
5. `self.cursor = self.connect.cursor()`:创建游标对象,用于执行 SQL 语句。
6. `from tools.config import MyMysql`:从 tools.config 模块中导入 MyMysql 类,用于连接 MySQL 数据库。
7. `class UaPool:`:定义一个 UaPool 类,用于从 MySQL 数据库中读取用户代理池。
8. `def __init__(self):`:定义 UaPool 类的构造函数,初始化 MySQL 数据库连接和游标对象,并读取用户代理池。
9. `database = MyMysql()`:创建 MyMysql 类的实例,用于连接 MySQL 数据库。
10. `db = database.connect`:获取 MySQL 数据库连接对象。
11. `cur = database.cursor`:获取 MySQL 数据库游标对象。
12. `sql = 'select useragent from ua_pool where types = %s and popularity != %s'`:定义 SQL 查询语句,用于查询指定类型和流行程度的用户代理。
13. `cur.execute(sql, ('Windows', 'Uncommon'))`:执行 SQL 查询语句,并传入参数,用于查询 Windows 桌面平台上不常见的用户代理。
14. `data_d = cur.fetchall()`:获取查询结果集,即 Windows 桌面平台上不常见的用户代理列表。
15. `desktop_list = []`:创建一个空列表,用于存储 Windows 桌面平台上不常见的用户代理。
16. `for data_d_per in data_d:`:遍历查询结果集,获取每个用户代理。
17. `desktop_list.append(data_d_per[0])`:将每个用户代理添加到 desktop_list 列表中。
18. `self.desktop = desktop_list`:将 desktop_list 列表赋值给 UaPool 类的 desktop 属性,表示 Windows 桌面平台上不常见的用户代理池。
19. `cur.execute(sql, ('Android', 'Uncommon'))`:执行 SQL 查询语句,并传入参数,用于查询 Android 移动平台上不常见的用户代理。
20. `data_m = cur.fetchall()`:获取查询结果集,即 Android 移动平台上不常见的用户代理列表。
21. `mobile_list = []`:创建一个空列表,用于存储 Android 移动平台上不常见的用户代理。
22. `for data_m_per in data_m:`:遍历查询结果集,获取每个用户代理。
23. `mobile_list.append(data_m_per[0])`:将每个用户代理添加到 mobile_list 列表中。
24. `self.mobile = mobile_list`:将 mobile_list 列表赋值给 UaPool 类的 mobile 属性,表示 Android 移动平台上不常见的用户代理池。
25. `cur.close()`:关闭游标对象。
26. `db.close()`:关闭数据库连接对象。
27. `def choose_ua(self, platform='desktop'):`:定义 choose_ua() 方法,用于随机选择一个用户代理。
28. `if platform == 'desktop':`:如果指定平台是桌面平台。
29. `return random.choice(self.desktop)`:从 Windows 桌面平台上不常见的用户代理池中随机选择一个用户代理并返回。
30. `elif platform == 'mobile':`:如果指定平台是移动平台。
31. `return random.choice(self.mobile)`:从 Android 移动平台上不常见的用户代理池中随机选择一个用户代理并返回。
32. `else:`:如果没有指定平台或者指定平台不是桌面平台或移动平台。
33. `return random.choice(self.desktop)`:从 Windows 桌面平台上不常见的用户代理池中随机选择一个用户代理并返回。
34. `if __name__ == '__main__':`:如果当前脚本作为主程序运行。
35. `up = UaPool()`:创建 UaPool 类的实例。
36. `print(up.choose_ua())`:随机选择一个用户代理并打印。
from PyQt5.QtWidgets import QApplication, QPushButton from PyQt5 import uic from dao.datab import dataInfo import pymysql class Stats: def __init__(self): # 从文件中加载UI定义 # 从 UI 定义中动态 创建一个相应的窗口对象 # 注意:里面的控件对象也成为窗口对象的属性了 # 比如 self.ui.button , self.ui.textEdit super().__init__() self.ui = uic.loadUi("Form - untitled.ui") #print(self.ui.__dict__) hostaddr=self.ui.hostaddr.text() username=self.ui.username.text() password=self.ui.password.text() database=self.ui.database.text() tablename=self.ui.tablename.text() # 信号和槽 self.ui.login.clicked.connect(self.handleCalc) def handleCalc(self): print(hostaddr) # dataInfo.connect(hostaddr=self.hostaddr, username=self.username, password=self.password, database=self.database, # table=self.tablename) con = pymysql.connect(host=hostaddr, password=password, port=3306, user=username, charset='utf8', database=database, table=tablename) cur = con.cursor() statement = "select * from danxuanti where id=1" data = cur.execute(statement) print(data) app = QApplication([]) stats = Stats() stats.ui.show() app.exec_()给这段代码改错
这段代码有一些问题,我来给您进行修改:
```python
from PyQt5.QtWidgets import QApplication
from PyQt5.uic import loadUi
import pymysql
class Stats:
def __init__(self):
# 从文件中加载UI定义
# 从 UI 定义中动态创建一个相应的窗口对象
# 注意:里面的控件对象也成为窗口对象的属性了
# 比如 self.ui.button , self.ui.textEdit
self.ui = loadUi("Form - untitled.ui")
# 信号和槽
self.ui.login.clicked.connect(self.handleCalc)
def handleCalc(self):
hostaddr = self.ui.hostaddr.text()
username = self.ui.username.text()
password = self.ui.password.text()
database = self.ui.database.text()
tablename = self.ui.tablename.text()
# 连接数据库
con = pymysql.connect(host=hostaddr,
user=username,
password=password,
database=database,
charset='utf8mb4',
cursorclass=pymysql.cursors.DictCursor)
cur = con.cursor()
statement = "select * from {table} where id=1".format(table=tablename)
cur.execute(statement)
data = cur.fetchone()
print(data)
con.close()
if __name__ == '__main__':
app = QApplication([])
stats = Stats()
stats.ui.show()
app.exec_()
```
修改的内容如下:
1. `from PyQt5.QtWidgets import QApplication, QPushButton`这一行代码中多了一个`QPushButton`,但在代码中并没有使用到,因此可以删除。
2. `from PyQt5 import uic`改为`from PyQt5.uic import loadUi`,因为`uic`模块中只有`loadUi`函数是我们需要用到的。
3. `from dao.datab import dataInfo`这一行代码不需要,可以删除。
4. 在`__init__`方法中,`super().__init__()`这一行代码并没有什么作用,可以删除。
5. 在`handleCalc`方法中,需要修改连接数据库的代码。原来的代码中使用了一个`dataInfo`对象来进行数据库连接,但在代码中并没有定义这个对象,因此需要改为使用`pymysql`模块来进行连接。此外,还需要注意设置`charset`参数为`utf8mb4`,避免出现中文乱码问题,同时设置`cursorclass`参数为`pymysql.cursors.DictCursor`,可以使查询结果以字典的形式返回,方便后续的数据处理。
6. `con.close()`要在`cur.fetchone()`语句执行后关闭,避免出现异常。
7. 在`if __name__ == '__main__':`语句块中,`app.exec()`应改为`app.exec_()`。
这样修改后,应该可以正常运行了。
阅读全文