- 安装pymysql: pip install pymysql
- 代码中导入包:import pymysql
- 连接数据库
import pymysql
con_sql = pymysql.connect(
host=HOSTNAME,
user=USERNAME,
password=PASSWORD,
charset='utf8mb4',
cursorclass=pymysql.cursors.DictCursor)
注意:
- charset='utf8mb4', # mysql字符格式
- cursorclass=pymysql.cursors.DictCursor) #可以使查询结果以字典的形式返回,方便我们操作和处理数据。
封装sql类
import pymysql
class MysqlDataDeal(object):
# 定义SQL查询函数
def SQLQuery(self, db, strsql):
global cursor
try:
cursor = db.cursor()
cursor.execute(strsql)
results = cursor.fetchall()
return results
except pymysql.Error as err:
print(err)
finally:
cursor.close()
db.close()
# 定义SQL命令函数(create、insert、delete、update)
def SQLCommand(self, db, strsql):
global cursor
try:
cursor = db.cursor()
cursor.execute(strsql)
# 事务提交
db.commit()
return True
except pymysql.Error as err:
print(err)
finally:
cursor.close()
db.close()
if __name__ == '__main__':
con_sql = MysqlDataDeal()
具体功能:数据库插入数据
class SQlWebjs(object):
def __init__(self):
self.command = MysqlDataDeal()
self.now_time = datetime.datetime.now(pytz.timezone('PRC')).strftime('%Y-%m-%d %H:%M:%S')
self.logger = logger()
#备注:PRC为北京时间
# 连接数据库
self.con_sql = pymysql.connect(host=HOSTNAME, user=USERNAME,
password=PASSWORD,
charset='utf8mb4', # mysql字符格式
cursorclass=pymysql.cursors.DictCursor)
# code_coverage_analysis_pc 表 插入数据
def insertRecord(self, login_role):
sql = """insert into uitest_live_server.code_coverage_analysis_pc (login_role) values('{}'""".format(login_role)
# print("sql=" + sql)
result = self.command.SQLCommand(self.con_sql, sql)
if result:
res = {'status': True, "msg": "数据:{},写入成功!".format(login_role)}
else:
res = {'status': False, "msg": "数据:{},写入失败!".format(login_role)}
return res