# -*- coding: utf-8 -*- import os, sqlite3 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): '''返回指定分数区间的名字,按分数从低到高排序''' # 关键别忘记用try..except..finally 真的好用啊 try: con = sqlite3.connect(db_file) cursor = con.cursor() cursor.execute(r"select name from user where score between ? and ? order by score",(low,high)) # 获取查询数据 value = cursor.fetchall() return [v[0] for v in value] except: print('出错了') finally: #关闭游标 cursor.close() #提交事物 con.commit() #关闭数据库连接 con.close() # 测试: 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)
Sign in to make a reply
黑色镜框BlacKiller