使用Python语言完成图书类别管理功能的开发,要求该功能至少包括:输出图书类别管理主菜单、添加图书类别信息、删除图书类别信息、修改图书类别信息、显示所有图书类别信息。要求将图书类别存储到booktype.csv文件中,文件格式如下:
时间: 2023-12-04 12:04:43 浏览: 25
以下是一个简单的 Python 代码示例,实现了图书类别管理功能的基本要求:
```python
import csv
# 定义图书类别管理主菜单
def print_menu():
print("欢迎使用图书类别管理系统")
print("1. 添加图书类别信息")
print("2. 删除图书类别信息")
print("3. 修改图书类别信息")
print("4. 显示所有图书类别信息")
print("0. 退出系统")
# 添加图书类别信息
def add_book_type():
with open('booktype.csv', 'a', newline='') as csvfile:
writer = csv.writer(csvfile)
type_id = input("请输入图书类别编号:")
type_name = input("请输入图书类别名称:")
writer.writerow([type_id, type_name])
print("添加成功!")
# 删除图书类别信息
def del_book_type():
type_id = input("请输入要删除的图书类别编号:")
with open('booktype.csv', 'r', newline='') as csvfile:
reader = csv.reader(csvfile)
rows = [row for row in reader if row[0] != type_id]
with open('booktype.csv', 'w', newline='') as csvfile:
writer = csv.writer(csvfile)
writer.writerows(rows)
print("删除成功!")
# 修改图书类别信息
def modify_book_type():
type_id = input("请输入要修改的图书类别编号:")
with open('booktype.csv', 'r', newline='') as csvfile:
reader = csv.reader(csvfile)
rows = [row for row in reader if row[0] != type_id]
if len(rows) == 0:
print("未找到该图书类别!")
return
type_name = input("请输入新的图书类别名称:")
rows.append([type_id, type_name])
with open('booktype.csv', 'w', newline='') as csvfile:
writer = csv.writer(csvfile)
writer.writerows(rows)
print("修改成功!")
# 显示所有图书类别信息
def show_all_book_type():
with open('booktype.csv', 'r', newline='') as csvfile:
reader = csv.reader(csvfile)
for row in reader:
print(row)
# 主程序入口
def main():
while True:
print_menu()
choice = input("请输入您的选择:")
if choice == '1':
add_book_type()
elif choice == '2':
del_book_type()
elif choice == '3':
modify_book_type()
elif choice == '4':
show_all_book_type()
elif choice == '0':
print("感谢使用图书类别管理系统,再见!")
break
else:
print("输入有误,请重新输入!")
if __name__ == '__main__':
main()
```
在这个示例中,我们使用 Python 内置的 csv 模块来读写 CSV 文件,实现了添加、删除、修改和显示图书类别信息的功能。每个功能都对应一个函数,通过用户输入选择要执行的功能。注意,由于我们需要在文件中读写数据,因此需要使用相应的文件读写操作。