在本篇博客中,我将介绍如何使用Python实现一个简单的抽签程序,并将数据存储到文件中。我们将使用JSON格式来存储活动和参与者的信息,并添加控制台命令交互功能。
准备工作
首先,我们需要导入必要的模块:
import json
import random
接着,我们定义了两个空的字典来存储活动数据和参与者数据:
activities = {} # 活动数据
participants = {} # 参与者数据
加载和保存数据
我们编写了两个函数来加载和保存数据:
def load_data():
global activities, participants
try:
with open('data.json', 'r') as file:
data = json.load(file)
activities = data.get('activities', {})
participants = data.get('participants', {})
except FileNotFoundError:
print("未找到数据文件,将从头开始。")
def save_data():
data = {'activities': activities, 'participants': participants}
with open('data.json', 'w') as file:
json.dump(data, file, ensure_ascii=False, indent=4)
主要功能
我们实现了以下功能来管理活动和参与者:
添加活动
#添加活动
def add_activity(activity_id, activity_name):
activities[activity_id] = {'name': activity_name, 'participants': []}
save_data()
删除活动
# 删除活动
def delete_activity(activity_id):
if activity_id in activities:
del activities[activity_id]
save_data()
添加参与者
# 添加参与者
def add_participant(activity_id, participant_id, participant_name):
if activity_id in activities:
activities[activity_id]['participants'].append({'id': participant_id, 'name': participant_name})
save_data()
删除参与者
# 删除参与者
def delete_participant(activity_id, participant_id):
if activity_id in activities:
activities[activity_id]['participants'] = [p for p in activities[activity_id]['participants'] if p['id'] != participant_id]
save_data()
查看参与人数和排序
# 查看参与人数和排序
def view_participants(activity_id):
if activity_id in activities:
participants = activities[activity_id]['participants']
print(f"参与人数: {len(participants)}")
print("参与者列表:")
for participant in participants:
print(f"{participant['id']}: {participant['name']}")
打乱参与者顺序
# 打乱参与者顺序
def shuffle_participants(activity_id):
if activity_id in activities:
random.shuffle(activities[activity_id]['participants'])
save_data()
print("参与者顺序已打乱。")
查看所有活动
# 查看所有活动
def view_all_activities():
print("所有活动:")
for activity_id, activity_data in activities.items():
print(f"{activity_id}: {activity_data['name']}")
控制台命令交互
我们使用一个循环来提供控制台命令交互功能,用户可以通过输入数字选择不同的操作。例如,用户可以添加活动、添加参与者、查看参与人数和排序等。
# 控制台命令交互
if __name__ == "__main__":
load_data()
while True:
print("\n可用命令:")
print("1. 添加活动")
print("2. 删除活动")
print("3. 添加参与者")
print("4. 删除参与者")
print("5. 查看参与人数和排序")
print("6. 打乱参与者顺序")
print("7. 查看所有活动")
print("8. 退出")
choice = input("请输入您的选择: ")
if choice == '1':
activity_id = input("请输入活动ID: ")
activity_name = input("请输入活动名称: ")
add_activity(activity_id, activity_name)
elif choice == '2':
activity_id = input("请输入要删除的活动ID: ")
delete_activity(activity_id)
elif choice == '3':
activity_id = input("请输入活动ID: ")
participant_id = input("请输入参与者ID: ")
participant_name = input("请输入参与者名称: ")
add_participant(activity_id, participant_id, participant_name)
elif choice == '4':
activity_id = input("请输入活动ID: ")
participant_id = input("请输入要删除的参与者ID: ")
delete_participant(activity_id, participant_id)
elif choice == '5':
activity_id = input("请输入活动ID: ")
view_participants(activity_id)
elif choice == '6':
activity_id = input("请输入活动ID: ")
shuffle_participants(activity_id)
elif choice == '7':
view_all_activities()
elif choice == '8':
save_data()
print("数据已保存。程序退出。")
break
else:
print("无效选择,请重试。")
总结
通过这个简单的抽签程序示例,我们展示了如何使用Python管理活动和参与者数据,并实现基本的功能。您可以根据自己的需求扩展和修改这段代码,添加更多功能或优化现有功能。
希望这篇博客能帮助您理解如何在Python中实现抽签程序并存储数据。如果您有任何问题或建议,请随时留言反馈!