feat: refactor handle function

This commit is contained in:
lanvent
2023-03-30 14:44:45 +08:00
parent 950a9f2ee0
commit 83136e3142
2 changed files with 115 additions and 98 deletions

View File

@@ -14,6 +14,15 @@ class Context:
self.type = type
self.content = content
self.kwargs = kwargs
def __contains__(self, key):
if key == 'type':
return self.type is not None
elif key == 'content':
return self.content is not None
else:
return key in self.kwargs
def __getitem__(self, key):
if key == 'type':
return self.type
@@ -21,6 +30,12 @@ class Context:
return self.content
else:
return self.kwargs[key]
def get(self, key, default=None):
try:
return self[key]
except KeyError:
return default
def __setitem__(self, key, value):
if key == 'type':