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 08:07:17 浏览: 97
这段代码中的问题可能是在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类。
相关问题
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_()`。
这样修改后,应该可以正常运行了。
阅读全文