欢迎访问宙启技术站
智能推送

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中实现用户之间的关注和取消关注功能。你可以根据自己的需要进行扩展和修改。