python和数据库_使用Python与数据库交互

#-*- coding: utf-8 -*-

"""Created on Sun Nov 18 19:25:01 2018

@author: wangm"""

#使用Python自带的sqlite3模块#创建数据库中的表、在表中插入数据、以及在输出中获取数据并对行进行计数#!/usr/bin/env python3#导入sqlite3模块

importsqlite3#创建一个代表数据库的连接对象，此处使用专用名词 ':memory:' 在内存中创建一个数据库#也可以使用其他名字，此时创建的数据库会被保存在你的工作目录，或者自己指定的路径下

con = sqlite3.connect(':memory:')#con = sqlite3.connect('mydatabase')#创建表sales，具有四个属性#此处query内有多行，使用 “”“

query = """create table sales

(customer varchar(20),

product varchar(40),

amount float, date DATE);"""

#执行query中的sql命令

con.execute(query)#将对数据库做的修改提交，即保存到数据库中

con.commit()#在表中插入四行数据

data = [('AAA', 'Apple', 10000.0, '2018-01-01'),

('BBB', 'Huawei', 5000.0, '2018-02-01'),

('CCC', 'Mi', 3000.0, '2018-03-01')]#此处statement只有一行，所以可以用 “#此处 ？ 为占位符，在connect对象的con.execute()或con.executemany()方法中，需要提供#一个包含四个值得元组，元组中的值会按位置替换到sql语句中#这种替换的方法还有一个好处是可以防止SQL注入攻击

statement = "insert into sales values(?, ?, ?, ?)"

#为data中的每条数据元组都执行statement中的命令，此处执行四次

con.executemany(statement, data)#注意此处不能用con.execute()方法#出错信息：ProgrammingError: Incorrect number of bindings supplied. The current statement uses 4, and there are 3 supplied.#con.execute(statement, data)

con.commit()#查询sales表#执行SQL语句"select * from sales"，并将结果赋给一个光标对象cursor

cursor = con.execute("select * from sales")#fetchall()取出SQL返回结果的所有行，并将这些行赋给rows

rows =cursor.fetchall()#rows 的类型 #print(type(rows))

#输出查询到的每一行，并对行计数#row_counter = 0#for row in rows:

#print(row)

#row_counter += 1#print('row_counter is %d' % (row_counter))

"""输出结果：

('AAA', 'Apple', 10000.0, '2018-01-01')

('BBB', 'Huawei', 5000.0, '2018-02-01')

('CCC', 'Mi', 3000.0, '2018-03-01')

row_counter is 3"""

#根据上述的这些对SQL的操作，和前面的关于对CSV文件、Excel文件的操作，就可以很简单的#将一个CSV文件或Excel文件存储到数据库中，成为数据库中的一张表#以CSV文件为例，即读取CSV文件每一行数据，使用insert，插入到数据库中

#除了上述的创建数据库、创建表、向表中插入数据、查询表中的数据之外#也可以在con.execute()中执行其他的插删改操作

#插入：

con.execute("insert into sales values(?, ?, ?, ?)", ('DDD', 'MeiZu', 2000.0, '2018-04-01'))

con.commit()

cursor= con.execute("select * from sales")

rows=cursor.fetchall()

row_counter=0for row inrows:print(row)

row_counter+= 1

print('row_counter is %d' %(row_counter))#修改

temp = [2500.0, '2018-05-01', 'DDD']

con.execute("update sales set amount=?, date=? where customer=?;", temp)

con.commit()

cursor= con.execute("select * from sales")

rows=cursor.fetchall()

row_counter=0for row inrows:print(row)

row_counter+= 1

print('row_counter is %d' %(row_counter))#删除

temp = ['CCC']

con.execute("delete from sales where customer=?;", temp)

con.commit()

cursor= con.execute("select * from sales")

rows=cursor.fetchall()

row_counter=0for row inrows:print(row)

row_counter+= 1

print('row_counter is %d' %(row_counter))"""输出结果：

('AAA', 'Apple', 10000.0, '2018-01-01')

('BBB', 'Huawei', 5000.0, '2018-02-01')

('CCC', 'Mi', 3000.0, '2018-03-01')

('DDD', 'MeiZu', 2000.0, '2018-04-01')

row_counter is 4

('AAA', 'Apple', 10000.0, '2018-01-01')

('BBB', 'Huawei', 5000.0, '2018-02-01')

('CCC', 'Mi', 3000.0, '2018-03-01')

('DDD', 'MeiZu', 2500.0, '2018-05-01')

row_counter is 4

('AAA', 'Apple', 10000.0, '2018-01-01')

('BBB', 'Huawei', 5000.0, '2018-02-01')

('DDD', 'MeiZu', 2500.0, '2018-05-01')

row_counter is 3"""