import sqlite3
table_name = 'user'
#数据库文件是当前目录下的mrsoft.db,如果不存在会创建该文件
conn = sqlite3.connect('mrsoft.db')
#创建一个游标对象
cursor = conn.cursor()
sql = '''SELECT tbl_name FROM sqlite_master WHERE type = 'table' '''
cursor.execute(sql)
values = cursor.fetchall()
tables = []
for v in values:
tables.append(v[0])
#如果表名不存在,建表
if table_name not in tables:
#使用游标对象执行sql语句
sql = '''CREATE TABLE '''+table_name +''' (
id INTEGER PRIMARY KEY AUTOINCREMENT,
name varchar(20),
price FLOAT,
weiht FLOAT
)'''
cursor.execute(sql)
print(table_name + ' 创建成功')
else:
# 使用insert语句。
sql ='insert into user (name) values("刘亦菲")'
cursor.execute(sql)
print(table_name + ' 已经存在')
#需要提交事务才能将数据写入数据库
conn.commit()
#关闭游标
cursor.close()
#关闭数据库连接
conn.close()