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文件,可以看到数据库中的表如下图所示:

python kv数据库 用python做数据库_python

二、数据的写入:

在第一步数据库建立好以后,我们需要删除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显示如下:

python kv数据库 用python做数据库_sqlite_02

如果需要写入多行数据,代码如下:

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('成功建表')

运行结果如下:

python kv数据库 用python做数据库_数据_03

三、数据的查询:

代码:

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('查询完毕')

运行结果:

python kv数据库 用python做数据库_数据库_04