import sqlite3
# 连接数据库
conn = sqlite3.connect('textbook.db')
cursor = conn.cursor()
# 创建教材表
cursor.execute('''
CREATE TABLE IF NOT EXISTS textbook (
id INTEGER PRIMARY KEY AUTOINCREMENT,
name TEXT NOT NULL,
author TEXT,
year INT,
quantity INT
)
''')
print("教材表已成功创建！")
def add_textbook(name, author, year, quantity):
conn = sqlite3.connect('textbook.db')
cursor = conn.cursor()
cursor.execute('''
INSERT INTO textbook (name, author, year, quantity)
VALUES (?, ?, ?, ?)
''', (name, author, year, quantity))
conn.commit()
print(f"教材《{name}》已成功添加！")
def query_textbook_by_name(name):
conn = sqlite3.connect('textbook.db')
cursor = conn.cursor()
cursor.execute('SELECT * FROM textbook WHERE name LIKE ?', ('%' + name + '%',))
results = cursor.fetchall()
if results:
for row in results:
print(row)
else:
print("未找到相关教材。")
def main():
while True:
print("\n教材管理系统")
print("1. 添加教材")
print("2. 查询教材")
print("3. 退出")
choice = input("请选择操作：")
if choice == '1':
name = input("请输入教材名称：")
author = input("请输入作者：")
year = int(input("请输入出版年份："))
quantity = int(input("请输入库存数量："))
add_textbook(name, author, year, quantity)
elif choice == '2':
name = input("请输入要查询的教材名称：")
query_textbook_by_name(name)
elif choice == '3':
print("感谢使用教材管理系统！")
break
else:
print("无效输入，请重新选择。")
if __name__ == "__main__":
main()