马上注册,结交更多好友,享用更多功能^_^
您需要 登录 才可以下载或查看,没有账号?立即注册
x
import os.path
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():
Inquire()
Ch1 = input("再次确认修改住户信息(是/否):")
while True:
if Ch1 == "是":
# 是的话就修改
break
elif Ch1 == "否":
# 不是的话就退出
break
# 添加住户
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():
Inquire()
Ch1 = input("再次删除住户信息(是/否):")
while True:
if Ch1 == "是":
# 是的话就删除
break
elif Ch1 == "否":
# 不是的话就退出
break
# 选择
def select():
for i in range(3):
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()
如何弄原理应该怎么弄,弄了半天还是没有头绪 |