开发Python图书管理系统使用csv
时间: 2023-12-25 16:44:22 浏览: 67
基于Python的图书管理系统
5星 · 资源好评率100%
好的,使用CSV文件来存储图书管理系统的数据是一种非常方便和简单的方法。以下是一个简单的Python图书管理系统的示例代码,它使用CSV文件来存储和管理图书信息:
```python
import csv
# 定义CSV文件路径
CSV_FILE_PATH = 'books.csv'
# 读取CSV文件中的数据
def read_csv():
with open(CSV_FILE_PATH, 'r') as file:
reader = csv.reader(file)
books = []
for row in reader:
books.append(row)
return books
# 写入数据到CSV文件中
def write_csv(books):
with open(CSV_FILE_PATH, 'w', newline='') as file:
writer = csv.writer(file)
for book in books:
writer.writerow(book)
# 添加一本书
def add_book():
title = input('请输入书名:')
author = input('请输入作者:')
pub_year = input('请输入出版年份:')
books = read_csv()
books.append([title, author, pub_year])
write_csv(books)
# 删除一本书
def delete_book():
title = input('请输入要删除的书名:')
books = read_csv()
for book in books:
if book[0] == title:
books.remove(book)
break
write_csv(books)
# 修改一本书的信息
def update_book():
title = input('请输入要修改的书名:')
books = read_csv()
for book in books:
if book[0] == title:
book[1] = input('请输入作者:')
book[2] = input('请输入出版年份:')
break
write_csv(books)
# 查找一本书
def search_book():
title = input('请输入要查找的书名:')
books = read_csv()
for book in books:
if book[0] == title:
print('书名:', book[0])
print('作者:', book[1])
print('出版年份:', book[2])
break
# 显示所有书籍
def show_books():
books = read_csv()
for book in books:
print('书名:', book[0])
print('作者:', book[1])
print('出版年份:', book[2])
# 主函数
def main():
while True:
print('1.添加书籍')
print('2.删除书籍')
print('3.修改书籍信息')
print('4.查找书籍')
print('5.显示所有书籍')
print('6.退出')
choice = int(input('请输入选项:'))
if choice == 1:
add_book()
elif choice == 2:
delete_book()
elif choice == 3:
update_book()
elif choice == 4:
search_book()
elif choice == 5:
show_books()
elif choice == 6:
break
else:
print('无效选项,请重新输入!')
if __name__ == '__main__':
main()
```
在这个示例中,我们使用了Python的csv模块来读取和写入CSV文件中的数据。在主函数中,我们通过不同的选项来执行不同的操作,比如添加一本书、删除一本书、修改一本书的信息、查找一本书、显示所有书籍等。每个选项中都会涉及到读取和写入CSV文件的操作。
阅读全文