mirror of
https://github.com/zhayujie/chatgpt-on-wechat.git
synced 2026-01-19 01:21:01 +08:00
43 lines
1.1 KiB
Python
43 lines
1.1 KiB
Python
from datetime import datetime, timedelta
|
|
|
|
|
|
class ExpiredDict(dict):
|
|
def __init__(self, expires_in_seconds):
|
|
super().__init__()
|
|
self.expires_in_seconds = expires_in_seconds
|
|
|
|
def __getitem__(self, key):
|
|
value, expiry_time = super().__getitem__(key)
|
|
if datetime.now() > expiry_time:
|
|
del self[key]
|
|
raise KeyError("expired {}".format(key))
|
|
self.__setitem__(key, value)
|
|
return value
|
|
|
|
def __setitem__(self, key, value):
|
|
expiry_time = datetime.now() + timedelta(seconds=self.expires_in_seconds)
|
|
super().__setitem__(key, (value, expiry_time))
|
|
|
|
def get(self, key, default=None):
|
|
try:
|
|
return self[key]
|
|
except KeyError:
|
|
return default
|
|
|
|
def __contains__(self, key):
|
|
try:
|
|
self[key]
|
|
return True
|
|
except KeyError:
|
|
return False
|
|
|
|
def keys(self):
|
|
keys = list(super().keys())
|
|
return [key for key in keys if key in self]
|
|
|
|
def items(self):
|
|
return [(key, self[key]) for key in self.keys()]
|
|
|
|
def __iter__(self):
|
|
return self.keys().__iter__()
|