|
|
import os
|
|
|
|
|
|
|
|
|
# 新增图书信息
|
|
|
def Add():
|
|
|
id_book = int(input('请输入编号:'))
|
|
|
name_book = input('请输入书名:')
|
|
|
author_book = input('请输入作者:')
|
|
|
type_book = input('请输入种类:')
|
|
|
num_book = int(input('请输入数量:'))
|
|
|
surplus_book = int(input('请输入剩余:'))
|
|
|
book_information = {'编号': id_book, '书名': name_book, '作者': author_book, '种类': type_book, '数量': num_book,
|
|
|
'剩余': surplus_book}
|
|
|
with open('图书信息.txt', 'a', encoding='utf-8') as f: # 图书信息以字典形式存储，增加信息就用a追加到文件中
|
|
|
if os.path.getsize('图书信息.txt') != 0:
|
|
|
f.write('\n') # 换行\n保证数据在文件中按行存储，这样在查询时用f.readlines()才能正确查询
|
|
|
f.write(str(book_information))
|
|
|
|
|
|
|
|
|
# 删除图书信息
|
|
|
def Delete():
|
|
|
del_num = int(input('请输入你要删除图书的编号:'))
|
|
|
with open('图书信息.txt', 'r', encoding='utf-8') as f: # 把文件信息全部提取出来
|
|
|
book_inf = f.readlines()
|
|
|
for i in range(len(book_inf)): # 循环查找想要编号的那一项
|
|
|
if eval(book_inf[i])['编号'] == del_num:
|
|
|
book_inf.pop(i) # 找到后把那一项删掉
|
|
|
break
|
|
|
with open('图书信息.txt', 'w', encoding='utf-8') as f: # 把删掉一项信息后的book_inf覆盖存储到文件中
|
|
|
for temp in book_inf:
|
|
|
f.write(str(temp)) # 存储必须以str类型，否则就报错
|
|
|
|
|
|
|
|
|
# 修改图书信息
|
|
|
def Alter():
|
|
|
alt_num = int(input('请输入你要修改图书的编号:'))
|
|
|
with open('图书信息.txt', 'r', encoding='utf-8') as f: # 取所有图书信息
|
|
|
books_inf = f.readlines()
|
|
|
for temp in books_inf: # 循环遍历出哪一项是想要修改的
|
|
|
b = eval(temp)
|
|
|
if b['编号'] == alt_num:
|
|
|
print(b) # 列出所选编号的图书信息，好看着修改
|
|
|
books_inf.remove(temp)
|
|
|
b['编号'] = int(input('请输入编号:'))
|
|
|
b['书名'] = input('请输入书名:')
|
|
|
b['作者'] = input('请输入作者:')
|
|
|
b['种类'] = input('请输入种类:')
|
|
|
b['数量'] = int(input('请输入数量:'))
|
|
|
b['剩余'] = int(input('请输入剩余:'))
|
|
|
break
|
|
|
with open('图书信息.txt', 'w', encoding='utf-8') as f:
|
|
|
f.write(str(b) + '\n')
|
|
|
with open('图书信息.txt', 'a', encoding='utf-8') as f:
|
|
|
for temp in books_inf:
|
|
|
f.write(str(temp))
|
|
|
|
|
|
|
|
|
# 查询图书信息
|
|
|
def Search():
|
|
|
sea_num = int(input('请输入你要查找图书的编号:'))
|
|
|
with open('图书信息.txt', 'r', encoding='utf-8') as f: # 取所有图书信息
|
|
|
books_inf = f.readlines()
|
|
|
for temp in books_inf: # 循环遍历出哪一项是想要查找的
|
|
|
b = eval(temp)
|
|
|
if b['编号'] == sea_num:
|
|
|
print('编号'.center(5, ' '), '书名'.center(15, ' '), '作者'.center(15, ' '), '种类'.center(15, ' '),
|
|
|
'数量'.center(5, ' '),
|
|
|
'剩余'.center(5, ' '))
|
|
|
format_data = '{:^5}\t{:^15}\t{:^10}\t{:^10}\t{:^5}\t{:^5}'
|
|
|
print(format_data.format(b.get('编号'), b.get('书名'), b.get('作者'), b.get('种类'), b.get('数量'),
|
|
|
b.get('剩余')))
|
|
|
|
|
|
|
|
|
# 展示所有图书信息
|
|
|
def Show():
|
|
|
with open('图书信息.txt', 'r', encoding='utf-8') as f: # 取所有图书信息
|
|
|
books_inf = f.readlines()
|
|
|
print('编号'.center(5, ' '), '书名'.center(15, ' '), '作者'.center(15, ' '), '种类'.center(15, ' '),
|
|
|
'数量'.center(5, ' '),
|
|
|
'剩余'.center(5, ' '))
|
|
|
format_data = '{:^5}\t{:^15}\t{:^10}\t{:^10}\t{:^5}\t{:^5}'
|
|
|
for temp in books_inf:
|
|
|
b = eval(temp)
|
|
|
print(format_data.format(b.get('编号'), b.get('书名'), b.get('作者'), b.get('种类'), b.get('数量'),
|
|
|
b.get('剩余')))
|
|
|
|
|
|
|
|
|
# 选择菜单
|
|
|
def Menu():
|
|
|
print('-----------------------------功能菜单-----------------------------')
|
|
|
print('\t\t\t\t1.新增图书信息')
|
|
|
print('\t\t\t\t2.删除图书信息')
|
|
|
print('\t\t\t\t3.修改图书信息')
|
|
|
print('\t\t\t\t4.查询图书信息')
|
|
|
print('\t\t\t\t5.展示所有图书信息')
|
|
|
print('\t\t\t\t0.退出')
|
|
|
print('----------------------------------------------------------------')
|
|
|
|
|
|
|
|
|
# 主程序
|
|
|
def main():
|
|
|
print('=========================资料室图书管理系统=========================')
|
|
|
while True:
|
|
|
Menu()
|
|
|
select = eval(input('请选择功能:'))
|
|
|
if select == 1:
|
|
|
Add()
|
|
|
print('新增信息完成！')
|
|
|
print('\n')
|
|
|
elif select == 2:
|
|
|
Delete()
|
|
|
print('删除信息完成！')
|
|
|
print('\n')
|
|
|
elif select == 3:
|
|
|
Alter()
|
|
|
print('修改信息完成！')
|
|
|
print('\n')
|
|
|
elif select == 4:
|
|
|
Search()
|
|
|
print('\n')
|
|
|
elif select == 5:
|
|
|
Show()
|
|
|
print('\n')
|
|
|
elif select == 0:
|
|
|
print('谢谢使用!')
|
|
|
break
|
|
|
else:
|
|
|
print("输入错误！请重新输入！")
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
main()
|
