Spaces:
Runtime error
Runtime error
from datetime import datetime, timedelta | |
from collections import defaultdict | |
class ExpiringMap: | |
def __init__(self): | |
self.data = defaultdict(dict) | |
def put(self, key, value, expire_time): | |
current_time = datetime.now() | |
expire_timestamp = current_time + timedelta(seconds=expire_time) | |
self.data[key] = {'value': value, 'expire_time': expire_timestamp} | |
def get(self, key): | |
current_time = datetime.now() | |
if key in self.data and self.data[key]['expire_time'] > current_time: | |
return self.data[key]['value'] | |
else: | |
return None | |
def remove(self, key): | |
if key in self.data: | |
del self.data[key] | |