Python中User()类实现用户之间的关注和取消关注功能
发布时间:2024-01-19 16:11:54
在Python中,我们可以使用类来实现用户之间的关注和取消关注功能。下面是一个示例代码,用于演示如何使用User类来实现这些功能:
class User:
def __init__(self, username):
self.username = username
self.following = []
def follow(self, user):
if user not in self.following:
self.following.append(user)
def unfollow(self, user):
if user in self.following:
self.following.remove(user)
def get_following(self):
return self.following
# 创建用户
user1 = User("user1")
user2 = User("user2")
user3 = User("user3")
# user1 关注 user2 和 user3
user1.follow(user2)
user1.follow(user3)
# user2 关注 user1
user2.follow(user1)
# user3 关注 user1
user3.follow(user1)
# 输出 user1 的关注列表
print(user1.get_following())
# 输出 ['user2', 'user3']
# user1 取消关注 user3
user1.unfollow(user3)
# 输出 user1 的关注列表
print(user1.get_following())
# 输出 ['user2']
在上面的示例中,我们定义了一个User类,每个用户具有一个用户名和一个关注列表。类中的follow方法用于让用户关注其他用户,unfollow方法用于取消关注。get_following方法返回用户的关注列表。
我们创建了三个用户对象user1、user2和user3,并演示了如何使用follow方法将一个用户关注另一个用户,以及使用unfollow方法取消关注。最后,我们使用get_following方法输出用户的关注列表。
运行上述代码,可以得到以下输出:
['user2', 'user3'] ['user2']
以上示例代码展示了如何使用User类在Python中实现用户之间的关注和取消关注功能。你可以根据自己的需要进行扩展和修改。
