python 基础教程 13数据库支持

数据库应用程序示例：

基于USDA的营养数据实验室提供的数据：https://www.ars.usda.gov/nutrientdata 点击下面的链接





点击download链接， 下载标题为Abbreviated下方的ASCII链接所指向的ASCCII格式的zip文件，取里面的ABBREV.txt文本文件，该文件数据每一行都有一个数据记录，字段以脱字符（^）进行分割， 数字字段直接包含数字，而文本字段包括波浪号（~）括起来的字符串值

将数据导入数据库：

#coding: utf-8
'''
Created on 2017年8月20日

@author: Leoi
'''
#importdata.py

import sqlite3

#convert将每一行分解为多个字段
def convert(value):
if value.startswith('~'):
return value.strip('~')
if not value:
value = '0'
return float(value)

conn  = sqlite3.connect('food.db')    #创建一个数据库，不存在则生成
curs = conn.cursor()    #创建一个光标conne.cursor()

try:
curs.execute('''
CREATE TABLE food(
id       TEXT        PRIMARY KEY,
desc     TEXT,
water    FLOAT,
kcal     FLOAT,
protein  FLOAT,
fat      FLOAT,
ash      FLOAT,
carbs    FLOAT,
fiber    FLOAT,
sugar    FLOAT
)

''')

except:
pass

query = 'INSERT INTO food VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)'

for line in open(r'C:\Users\Leoi\Desktop\python code\ABBREV.txt'):
fields = line.split('^')
vals = [convert(f) for f in fields[:10]]  #此处修改为10，因为每个表有10个数据
curs.execute(query, vals)  #执行sql的INSERT语句将文本字段中的值插入到数据库中

conn.commit()
conn.close()

对数据库进行查询：

#coding: utf-8
#food_query.py

import sqlite3, sys

conn = sqlite3.connect('food.db')
curs = conn.cursor()  #游标，用于执行查询和检查结果

query = 'SELECT * FROM food WHERE %s' % 'kcal <= 100 AND fiber >=10'
print query
curs.execute(query)
names = [f[0] for f in curs.description]
for row in curs.fetchall():
for pair in zip(names, row):
print '%s: %s' % pair
print

将书上的sys.argv[1]改为了kcal <= 100 AND fiber >=10或其他sql查询条件，不用命令行的形式运行，该程序为一次性程序。