SNS 맞팔 하는 코드 | 객체 지향 프로그래밍 시리즈 004

별스타그램에 취직한 Jane은 User 클래스에 "팔로우" 기능을 추가하라는 지시를 받았습니다.

팔로우 기능은 크게 2개의 동작을 해야 합니다.

1. "내가 팔로우하는 사람" 목록에 그 사람을 추가하는 동작과
2. 상대방의 "나를 팔로우하는 사람" 목록에 나를 추가하는 동작

팔로우 기능은 follow 메소드로 구현하려고 하는데요. 팔로우 기능을 만드는 김에

1. follow 메소드 말고도
2. 유저가 팔로우하는 사람 수를 알려주는 num_following 메소드와
3. 유저를 팔로우하는 사람 수를 알려주는 num_followers 메소드도

추가해봅시다.

User 클래스에 이 메소드들을 모두 추가하고 나서 코드를 실행하면 아래와 같은 실행 결과가 나와야 합니다.

 

class User:
    # 인스턴스 변수 설정
    def __init__(self, name, email, password):
        self.name = name
        self.email = email
        self.password = password

        self.following_list = []    # 이 유저가 팔로우하는 유저 리스트
        self.followers_list = []    # 이 유저를 팔로우하는 유저 리스트

    # 팔로우
    def follow(self, another_user):
        self.following_list.append(another_user)
        another_user.followers_list.append(self)
    # 내가 몇 명을 팔로우하는지 리턴
    def num_following(self):
        return len(self.following_list)

    # 나를 몇 명이 팔로우하는지 리턴
    def num_followers(self):
        return len(self.followers_list)

# 유저들 생성
user1 = User("Young", "young@.kr", "123456")
user2 = User("Yoonsoo", "yoonsoo@.kr", "abcdef")
user3 = User("Taeho", "taeho@.kr", "123abc")
user4 = User("Lisa", "lisa@.kr", "abc123")

# 유저마다 서로 관심 있는 유저를 팔로우
user1.follow(user2)
user1.follow(user3)
user2.follow(user1)
user2.follow(user3)
user2.follow(user4)
user4.follow(user1)

# 유저 이름, 자신의 팔로워 수, 자신이 팔로우하는 사람 수를 출력합니다
print(user1.name, user1.num_followers(), user1.num_following())
print(user2.name, user2.num_followers(), user2.num_following())
print(user3.name, user3.num_followers(), user3.num_following())
print(user4.name, user4.num_followers(), user4.num_following())