|

楼主 |
发表于 2020-8-14 10:02:14
|
显示全部楼层
- import os.path
- import openpyxl
- import pandas as pd
- import xlrd
- import xlwt
- from xlutils.copy import copy
- # 创建excel表格
- def create():
- wb = xlwt.Workbook('信息表.xls')
- sh1 = wb.add_sheet('Sheet1')
- # 添加信息
- sh1.write(0, 0, '身份证件号')
- sh1.write(0, 1, '姓名')
- sh1.write(0, 2, '房间号')
- sh1.write(0, 3, '房费')
- sh1.write(0, 4, '押金')
- # 保存
- wb.save('信息表.xls')
- # 查询住户信息
- def Inquire():
- Ch1 = input("请输入身份证件号:")
- readbook = xlrd.open_workbook('信息表.xls')
- sh1 = readbook.sheet_by_index(0)
- # 循环信息表的所有行
- for row in sh1.get_rows():
- ID = row[0] # 身份证所在的列
- product_value = ID.value
- # 判断输入身份证的信息
- if product_value == Ch1:
- if product_value != '身份证件号': # 排除第一行
- price_column = row[0] # 身份证所在的列
- ID_card = price_column.value
- price_column = row[1] # 姓名所在的列
- name = price_column.value
- price_column = row[2] # 房间号所在的列
- room_number = price_column.value
- price_column = row[3] # 房费所在的列
- Room_rate = price_column.value
- price_column = row[4] # 押金所在的列
- deposit = price_column.value
- # 打印
- print(f"身份证件号:{ID_card}姓名:{name}房间号:{room_number}房费:{Room_rate}押金:{deposit}")
- # 修改住户信息
- def modify():
- return
- # 添加住户
- def Add_to():
- # 打开excel表格
- readbook = xlrd.open_workbook('信息表.xls')
- wb = copy(readbook)
- # 获取最后一行并加一行
- excel = pd.read_excel(readbook)
- line = len(excel) + 1
- # 选取第一个表单
- sh1 = wb.get_sheet(0)
- # 输入数据
- num = input("身份证件号:")
- name = input("姓名:")
- num2 = input("房间号:")
- num3 = input("房费:")
- num4 = input("押金:")
- # 导入到excel表格里
- sh1.write(line, 0, num)
- sh1.write(line, 1, name)
- sh1.write(line, 2, num2)
- sh1.write(line, 3, num3)
- sh1.write(line, 4, num4)
- # 保存
- wb.save('信息表.xls')
- # 删除住户
- def delete():
- readbook = openpyxl.load_workbook('信息表.xls')
- wb = copy(readbook)
- ws = wb['Sheet1']
- # 获取最后一行并加一行
- ws.delete_rows()
- wb.save('信息表.xls')
- # 选择
- def select():
- for i in range(2):
- Ch = input("请选择:")
- if Ch == "1":
- Inquire()
- break
- elif Ch == "2":
- modify()
- break
- elif Ch == "3":
- Add_to()
- break
- elif Ch == "4":
- delete()
- break
- elif Ch == "5":
- exit(1)
- else:
- print("输入错误")
- # 判断是否有信息表
- if not os.path.exists('信息表.xls'):
- create()
- print("创建成功")
- Add_to()
- else:
- print("表格已存在")
- print("1:查询住户信息 2:修改住户信息 3:添加住户 4:删除住户 5:退出")
- select()
复制代码
要不以为自己很牛逼似的,说得好像我没有弄就在这里哔哔赖赖,代码给你了有种就把完全补完不然就不要在这哔哔赖赖fw
|
|