python+mysql数据库的简单操作

时间:2022-06-06 15:28:46

最近接了一个任务,测试某项类似于收益情况报表的功能,因计算公式复杂,单纯手算过于复杂,所以想到写成脚本

根据python的分治原则,先整了几个函数用于实现计算逻辑,后发现数据输入过于繁琐,所以决定使用脚本直接取库,因为之前没有用python操作数据库的经验,所以写篇入门教程给自己,以备不时之需

需要引入的包是 MySQLdb

先需要获取数据库连接

 conn = MySQLdb.connect(host=localhost, user='root',
passwd='root', db='test',
port=3306)

然后获取cursor对象,并使用cursor的execute方法执行SQL语句,

     cursor = conn.cursor()
sql = "select nav from xx_nav WHERE exponent_code = '" + \
market + "' and nav_date >= '" + start_time + "' and nav_date <= '" + end_time + "'"
cursor.execute(sql)

然后使用fetchall获取取到的数据

 datas = cursor.fetchall()
## 此时取到数据是元组套元组
## ((第一行第一列, 第一行第二列,……),(第二行第一列, 第二行第二列, ……), ……)

将数据拼接处理为你想要的数据结构并套用处理逻辑

记得在合适的地方关闭数据库连接

conn.close()