Design Twitter LeetCode solution in Python

1
class Twitter:
2

3
    def __init__(self):
4
        self.tweets = []
5
        self.friends = defaultdict(list)
6

7
    def postTweet(self, userId: int, tweetId: int) -> None:
8
        self.tweets.append([userId, tweetId])
9

10
    def getNewsFeed(self, userId: int) -> List[int]:
11
        try:
12
            tweets = [
13
                x[1] for x in self.tweets if x[0] in [userId] + self.friends[userId]
14
            ]
15
            return list(reversed(tweets))[:10]
16
        except:
17
            return
18

19
    def follow(self, followerId: int, followeeId: int) -> None:
20
        self.friends[followerId].append(followeeId)
21

22
    def unfollow(self, followerId: int, followeeId: int) -> None:
23
        try:
24
            self.friends[followerId].remove(followeeId)
25
        except:
26
            pass