http://blog.csdn.net/fdipzone/article/details/52705507

转载自 http://blog.csdn.net/HeatDeath/article/details/65633896

我使用的是 Python3.6,所以选择 mysqlclient 来操作 MySQL

安装mysqlclient

要想使 python 可以操作 mysql 就需要 MySQLdb 驱动,它是 python 操作 mysql 必不可少的模块。
使用pip安装

  1. pip install mysqlclient1

测试

测试非常简单,检查 MySQLdb 模块是否可以正常导入。

  1. >>> import MySQLdb1

没有报错提示MySQLdb模块找不到,说明安装OK

python 操作mysql数据库基础

  1. #coding=utf-8
  2. import MySQLdb
  3. #connect() 方法用于创建数据库的连接,里面可以指定参数:用户名,密码,主机等信息。
  4. #这只是连接到了数据库,要想操作数据库需要创建游标。
  5. conn= MySQLdb.connect(
  6. host='localhost',
  7. port = 3306,
  8. user='root',
  9. passwd='123456',
  10. db ='test',
  11. )
  12. #通过获取到的数据库连接conn下的cursor()方法来创建游标。
  13. cur = conn.cursor()
  14. #创建数据表,通过游标cur 操作execute()方法可以写入纯sql语句。通过execute()方法中写如sql语句来对数据进行操作
  15. cur.execute("create table student(id int ,name varchar(20),class varchar(30),age varchar(10))")
  16. #插入一条数据
  17. cur.execute("insert into student values('2','Tom','3 year 2 class','9')")
  18. #修改查询条件的数据
  19. cur.execute("update student set class='3 year 1 class' where name = 'Tom'")
  20. #删除查询条件的数据
  21. cur.execute("delete from student where age='9'")
  22. #cur.close() 关闭游标
  23. cur.close()
  24. #conn.commit()方法在提交事物,在向数据库插入一条数据时必须要有这个方法,否则数据不会被真正的插入。
  25. conn.commit()
  26. #conn.close()关闭数据库连接
  27. conn.close()

插入数据

通过上面execute()方法中写入纯的sql语句来插入数据并不方便。如:

  1. cur.execute("insert into student values('2','Tom','3 year 2 class','9')")

我要想插入新的数据,必须要对这条语句中的值做修改。我们可以做如下修改:

  1. #coding=utf-8
  2. import MySQLdb
  3. conn= MySQLdb.connect(
  4. host='localhost',
  5. port = 3306,
  6. user='root',
  7. passwd='123456',
  8. db ='test',
  9. )
  10. cur = conn.cursor()
  11. #插入一条数据
  12. sqli="insert into student values(%s,%s,%s,%s)"
  13. cur.execute(sqli,('3','Huhu','2 year 1 class','7'))
  14. cur.close()
  15. conn.commit()
  16. conn.close()

假如要一次向数据表中插入多条值呢?

  1. #coding=utf-8
  2. import MySQLdb
  3. conn= MySQLdb.connect(
  4. host='localhost',
  5. port = 3306,
  6. user='root',
  7. passwd='123456',
  8. db ='test',
  9. )
  10. cur = conn.cursor()
  11. #一次插入多条记录
  12. sqli="insert into student values(%s,%s,%s,%s)"
  13. cur.executemany(sqli,[
  14. ('3','Tom','1 year 1 class','6'),
  15. ('3','Jack','2 year 1 class','7'),
  16. ('3','Yaheng','2 year 2 class','7'),
  17. ])
  18. cur.close()
  19. conn.commit()
  20. conn.close()

executemany()方法可以一次插入多条值,执行单挑sql语句,但是重复执行参数列表里的参数,返回值为受影响的行数。
http://www.cnblogs.com/fnng/p/3565912.html


完整代码,从 CSV 导入 mysql

用到的库 https://mysqlclient.readthedocs.io/user_guide.html#connection-objects

  1. #coding=utf-8
  2. import MySQLdb
  3. import csv
  4. import os
  5. import codecs
  6. def Read_CSV():
  7. guojia = [x for x in os.listdir(".") if os.path.isfile(x) and os.path.splitext(x)[1]=='.csv']
  8. for csvname in guojia:
  9. filename = os.path.join(os.getcwd(), csvname)
  10. with open(csvname, 'r', encoding='utf-8') as f:
  11. reader = csv.reader(f)
  12. for item in reader:
  13. i = tuple(item)
  14. print(i)
  15. save_to_mysql(i)
  16. def save_to_mysql(i):
  17. conn= MySQLdb.connect(
  18. host='localhost',
  19. port = 3306,
  20. user='root',
  21. passwd='nihao52cons',
  22. db ='QQ_QUN',
  23. charset= 'utf8mb4'
  24. )
  25. cur = conn.cursor()
  26. # cur.execute("insert into student values('群名称', '群号', '群人数', '群上限', '群主', '地域', '分类', '标签', '群简介')")
  27. sqli = "insert into xuexi values(%s,%s,%s,%s,%s,%s,%s,%s,%s)"
  28. cur.executemany(sqli,[i])
  29. cur.close()
  30. conn.commit()
  31. conn.close()
  32. def main():
  33. Read_CSV()
  34. if __name__ == '__main__':
  35. main()

上面的代码有一个问题,

群人数应该是 INT,因为群名称需要排序。 所以我要从上面的CSV 读取里删除第一行。