import pymysql
"""pymysql的封装"""
class Mysql:
def __init__(self):
"""初始化方法"""
self.connect_obj = self.get_connect() # mysql连接对象
self.cursor_obj = self.get_cursor() # 游标对象
def get_connect(self):
"""
获取数据库的连接对象
:return: 返回连接对象
"""
conn = pymysql.connect(
host='127.0.0.1',
port=3306,
user='root',
password='uslee541',
database='python',
charset='utf8',
)
return conn
def get_cursor(self):
"""
通过数据库连接对象获取游标对象
:return: 返回游标对象
"""
cursor = self.connect_obj.cursor(cursor=pymysql.cursors.DictCursor) # 游标对象
return cursor
def select_all_method(self, sql):
"""
查询数据库全部的内容
:param sql: sql语句
:return: 返回sql查询的内容
"""
self.cursor_obj.execute(sql)
data = self.cursor_obj.fetchall()
return data
def select_one_method(self, sql):
"""
查询数据库单条数据内容
:param sql: sql语句
:return: 返回执行的sql语句的内容
"""
self.cursor_obj.execute(sql)
data = self.cursor_obj.fetchone()
return data
def commit_method(self, sql):
"""
提交sql语句到数据库
:param sql: 要执行的sql语句
:return:
"""
self.cursor_obj.execute(sql)
self.connect_obj.commit() # 提交执行的sql语句
def __del__(self):
"""程序结束,关闭游标,连接对象"""
self.cursor_obj.close() # 关闭游标对象
self.connect_obj.close() # 关闭连接对象
mysql = Mysql()
if __name__ == '__main__':
res = mysql.select_one_method("select * from student")
print(res)
代码封装为类,可以直接调用