voice_leave_of_absence_Helper / simulated_data.py
ohytic6's picture
Update simulated_data.py
e5a9d08 verified
import json
from datetime import datetime, timedelta
import random
# 定義員工名單(八名中文、兩名英文)和對應員工編號
employees = [
{"姓名": "陳大明", "員工編號": "001"},
{"姓名": "李小華", "員工編號": "002"},
{"姓名": "王小美", "員工編號": "003"},
{"姓名": "林大志", "員工編號": "004"},
{"姓名": "黃佳玲", "員工編號": "005"},
{"姓名": "趙明華", "員工編號": "006"},
{"姓名": "劉大強", "員工編號": "007"},
{"姓名": "張小英", "員工編號": "008"},
{"姓名": "John Doe", "員工編號": "009"},
{"姓名": "Jane Smith", "員工編號": "010"}
]
# 定義上班狀態
work_status_options = ["已上班", "未上班未請假", "未上班已請假"]
weekend_status = "休假"
# 設定日期範圍,2024年11月1日至11月30日
start_date = datetime(2024, 11, 1)
end_date = datetime(2024, 11, 30)
# 建立請假狀態表
attendance_data = []
# 生成每個日期和員工的上班狀態
current_date = start_date
while current_date <= end_date:
for employee in employees:
if current_date.weekday() >= 5: # 週六日設為休假
status = weekend_status
elif current_date <= datetime(2024, 11, 12): # 11月12日及以前隨機分布上班狀態
status = random.choice(work_status_options)
else: # 11月13日以後保持工作日空白
status = "" if current_date.weekday() < 5 else weekend_status
attendance_data.append({
"日期": current_date.strftime("%Y-%m-%d"),
"姓名": employee["姓名"],
"員工編號": employee["員工編號"],
"當日上班狀態": status
})
current_date += timedelta(days=1)
# 儲存為 JSON 檔案
json_path = '/mnt/data/attendance_data_november_2024.json'
with open(json_path, 'w', encoding='utf-8') as f:
json.dump(attendance_data, f, ensure_ascii=False, indent=4)