leetcode.com/python/359_Logger_Rate_Limiter.py

from collections import defaultdict
class Logger(object):

    def __init__(self):
        """
        Initialize your data structure here.
        """
        self.messageTimeMap = defaultdict(list)

    def shouldPrintMessage(self, timestamp, message):
        """
        Returns true if the message should be printed in the given timestamp, otherwise returns false.
        If this method returns false, the message will not be printed.
        The timestamp is in seconds granularity.
        :type timestamp: int
        :type message: str
        :rtype: bool
        """

        if len(self.messageTimeMap[message]) == 0:
            self.messageTimeMap[message].append(timestamp)
            return True
        else:
            timeDiff = abs(timestamp - self.messageTimeMap[message][-1])
            if timeDiff >= 10:
                self.messageTimeMap[message].append(timestamp)
                return True
            else:
                return False


# Your Logger object will be instantiated and called as such:
# obj = Logger()
# param_1 = obj.shouldPrintMessage(timestamp,message)