使用SQLite

使用SQLite

在Sqlite中根据分数段查找指定的名字：

#还了解了

1、__file__ 默认变量会返回当前文件的文件路径（绝对路径)

2、for 列表推导式 

3、assert用法

 

 

# -*- coding: utf-8 -*-
'''
use SQLite3
返回指定分数区间的名单，按分数从低到高排序

'''
import os, sqlite3
# print("__file__ = "+__file__)
db_file = os.path.join(os.path.dirname(__file__), 'test.db')
if os.path.isfile(db_file):
    os.remove(db_file)

# 初始数据:
conn = sqlite3.connect(db_file)
cursor = conn.cursor()
cursor.execute('create table user(id varchar(20) primary key, name varchar(20), score int)')
cursor.execute(r"insert into user values ('A-001', 'Adam', 95)")
cursor.execute(r"insert into user values ('A-002', 'Bart', 62)")
cursor.execute(r"insert into user values ('A-003', 'Lisa', 78)")
cursor.close()
conn.commit()
conn.close()

def get_score_in(low, high):
    ' 返回指定分数区间的名字，按分数从低到高排序 '
    conn = sqlite3.connect(db_file)
    cursor = conn.cursor()
    cursor.execute("select name from user where score between ? and ? order  by score",(low,high))
    values = cursor.fetchall()
    #返回的数据如下：[('Bart',), ('Lisa',), ('Adam',)]
    #使用列表推导式，生成一个新的列表，并返回
    #分开写可以写成 x = [ a[0] for a in values ] ; return x
    cursor.close()
    conn.close()
    return [ a[0] for a in values ]
# 测试:
#assert 两边相等，如果 不相等，则抛出异常，异常为 , 后面内容，在这里则是执行方法，并展示 方法返回的内容
assert get_score_in(80, 95) == ['Adam'], get_score_in(80, 95)
assert get_score_in(60, 80) == ['Bart', 'Lisa'], get_score_in(60, 80)
assert get_score_in(60, 100) == ['Bart', 'Lisa', 'Adam'], get_score_in(60, 100)

print('Pass')