修复 Bug: 文件选择索引;身份证转化为文本格式

main
Dawn_Ocean 2023-12-25 23:43:52 +08:00
parent 34bbfe236a
commit c7a5318ed9
3 changed files with 13 additions and 9 deletions

20
main.py
View File

@ -1,21 +1,25 @@
import pandas as pd
import time, datetime
import datetime
from math import floor
from copy import copy
import os
# Read & Process ID table
items = os.listdir(os.path.abspath('.'))
print("在当前目录共有以下文件:")
file_index = 0
items_copy = copy(items)
for item in items:
if item != "main.py" and item != "main.exe":
if item != "main.py" and item != "main.exe" and item[0] != '.':
file_index += 1
print(f"{file_index} - {item}")
else:
items_copy.remove(item)
id_index = int(input("请选择身份证列表(输入数字后回车)"))
duty_index = int(input("请选择值班排班表(输入数字后回车)"))
identities_path = r'./' + items[id_index - 1]
duty_path = r'./' + items[duty_index - 1]
identities_path = r'./' + items_copy[id_index - 1]
duty_path = r'./' + items_copy[duty_index - 1]
print("正在读取身份信息...")
identities = pd.read_excel(identities_path, sheet_name= "Sheet1")
@ -39,9 +43,9 @@ index = 0
time_fmt = "%H:%M"
for members in duty["人员"]:
time = duty["时间"][index].split('-')
time_end = datetime.datetime.strptime(time[1], time_fmt)
time_start = datetime.datetime.strptime(time[0], time_fmt)
period = duty["时间"][index].split('-')
time_end = datetime.datetime.strptime(period[1], time_fmt)
time_start = datetime.datetime.strptime(period[0], time_fmt)
hours = (time_end - time_start).seconds / 3600
members = members.split('')
@ -68,7 +72,7 @@ hour_path = r'./hour.xlsx'
id_list = []
for member in hours_dict.keys():
id_list.append(id_dict[member])
id_list.append(str(id_dict[member]))
df = pd.DataFrame({"姓名": hours_dict.keys(), "身份证号码": id_list, "荣誉时数值": hours_dict.values()})
df = df.set_index("姓名")

Binary file not shown.

Binary file not shown.