Python--数据库

Python–数据库

---

一、建立数据库：

import xlwt
import sqlite3

conn=sqlite3.connect('test.db')    
#打开数据库文件,程序会在.py文件同目录下生成一个test.db文件，那就是我们的数据库文件
print('Opened database successfully')    #打开成功则会打印这句话
c=conn.cursor()#获取游标
sql='''
create table company
(id int primary key not null,
name text not null,
age int not null,
address char(50),
salary real);
'''#sql具体类型可查看菜鸟教程的SQLite部分
c.execute(sql)#执行sql语句
conn.commit()#提交数据库操作
conn.close()#关闭数据库连接
print('成功建表')

运行后，使用SQLiteSpy打开test.db文件，可以看到数据库中的表如下图所示：

二、数据的写入：

在第一步数据库建立好以后，我们需要删除create table company的内容才能往数据库中写入内容，如过建立数据库和写入在同一个程序中同时运行的话，会出现no such table: company的警告，我们将程序改成这样：

import xlwt
import sqlite3

conn=sqlite3.connect('test.db')    
#打开数据库文件,程序会在.py文件同目录下生成一个test.db文件，那就是我们的数据库文件
print('Opened database successfully')    #打开成功则会打印这句话
c=conn.cursor()#获取游标
sql='''
insert into company(id,name,age,address,salary)

values(1,'张三',32,'成都',8888)
'''#sql具体类型可查看菜鸟教程的SQLite部分
c.execute(sql)#执行sql语句
conn.commit()#提交数据库操作
conn.close()#关闭数据库连接
print('成功建表')

其实就是将create table换成insert into company，但是程序不要重复运行，否则第二次运行的时候会出现错误，即UNIQUE constraint failed: company.id

打开test.db显示如下：

如果需要写入多行数据，代码如下：

import xlwt
import sqlite3

conn=sqlite3.connect('test.db')    
#打开数据库文件,程序会在.py文件同目录下生成一个test.db文件，那就是我们的数据库文件
print('Opened database successfully')    #打开成功则会打印这句话
c=conn.cursor()#获取游标
sql='''
insert into company(id,name,age,address,salary)
values(1,'张三',32,'成都',8888)
'''#sql具体类型可查看菜鸟教程的SQLite部分，1表示第一个数据
sql='''
insert into company(id,name,age,address,salary)
values(2,'李四',34,'北京',9999)
'''#写入多行数据
c.execute(sql)#执行sql语句
conn.commit()#提交数据库操作
conn.close()#关闭数据库连接
print('成功建表')

运行结果如下：

三、数据的查询：

代码：

import xlwt
import sqlite3

conn=sqlite3.connect('test.db')    
#打开数据库文件,程序会在.py文件同目录下生成一个test.db文件，那就是我们的数据库文件
print('Opened database successfully')    #打开成功则会打印这句话
c=conn.cursor()#获取游标
sql='select id,name,address,salary from company'
cursor=c.execute(sql)#查询时c.execute(sql)有一个返回值，即我们查询数据库得到的数据
for row in cursor:
      print('id=',row[0])
      print('name=',row[1])
      print('address=',row[2])
      print('salary=',row[3],'\n')
conn.close()#关闭数据库连接
print('查询完毕')

运行结果：