实训案例好友管理系统代码
时间: 2023-08-06 20:19:10 浏览: 95
以下是一个简单的好友管理系统的Python代码:
```python
class Friend:
def __init__(self, name, age, gender):
self.name = name
self.age = age
self.gender = gender
def __repr__(self):
return f"{self.name}, {self.age}, {self.gender}"
class FriendManager:
def __init__(self):
self.friend_list = []
def add_friend(self, friend):
self.friend_list.append(friend)
def search_friend(self, keyword):
result = []
for friend in self.friend_list:
if keyword.lower() in friend.name.lower():
result.append(friend)
return result
def remove_friend(self, friend):
if friend in self.friend_list:
self.friend_list.remove(friend)
def get_all_friends(self):
return self.friend_list
if __name__ == "__main__":
manager = FriendManager()
while True:
print("1. 添加好友")
print("2. 查找好友")
print("3. 删除好友")
print("4. 显示所有好友")
print("5. 退出程序")
choice = input("请选择操作:")
if choice == "1":
name = input("请输入好友姓名:")
age = input("请输入好友年龄:")
gender = input("请输入好友性别:")
friend = Friend(name, age, gender)
manager.add_friend(friend)
print("好友添加成功!")
elif choice == "2":
keyword = input("请输入查找关键字:")
result = manager.search_friend(keyword)
if result:
for friend in result:
print(friend)
else:
print("未找到匹配的好友。")
elif choice == "3":
name = input("请输入要删除的好友姓名:")
remove_list = []
for friend in manager.get_all_friends():
if friend.name == name:
remove_list.append(friend)
if len(remove_list) == 0:
print("未找到匹配的好友。")
elif len(remove_list) == 1:
manager.remove_friend(remove_list[0])
print("好友删除成功!")
else:
print("找到多个匹配的好友,请选择要删除的好友:")
for i, friend in enumerate(remove_list):
print(f"{i+1}. {friend}")
choice = int(input()) - 1
manager.remove_friend(remove_list[choice])
print("好友删除成功!")
elif choice == "4":
friends = manager.get_all_friends()
if friends:
for friend in friends:
print(friend)
else:
print("好友列表为空。")
elif choice == "5":
print("程序已退出。")
break
else:
print("无效的选项,请重新输入。")
```
该代码包含了Friend和FriendManager两个类,Friend类表示一个好友,FriendManager类表示好友管理器,提供了添加、查找、删除和显示所有好友的功能。
运行程序后,用户可以通过菜单选项进行相应的操作,包括添加好友、查找好友、删除好友和显示所有好友。用户输入菜单选项后,程序会调用相应的方法来处理用户请求,并在屏幕上显示相应的结果。
阅读全文