feat: 秘钥凭证管理
This commit is contained in:
@@ -0,0 +1,34 @@
|
||||
from enum import Enum
|
||||
|
||||
|
||||
class ApiSecretStatus(Enum):
|
||||
valid = "valid" # 秘钥有效
|
||||
invalid = "invalid" # 秘钥无效
|
||||
not_exist = "not_exist" # 秘钥不存在
|
||||
|
||||
|
||||
class ApiSecretUtils():
|
||||
def __init__(self):
|
||||
self.cache = {}
|
||||
|
||||
# 验证秘钥状态
|
||||
async def verify_secret(self, secret: str) -> ApiSecretStatus:
|
||||
api_secret_status = self.cache.get(secret, None)
|
||||
print("verify_secret", self.cache)
|
||||
return api_secret_status or ApiSecretStatus.not_exist
|
||||
|
||||
# 保存秘钥状态
|
||||
async def save_secret(self, secret: str, status: ApiSecretStatus):
|
||||
print("save_secret", self.cache)
|
||||
self.cache[secret] = status
|
||||
|
||||
# 移除秘钥缓存
|
||||
async def remove_secret(self, secret: str):
|
||||
# self.cache.pop(secret, None)
|
||||
if secret in self.cache:
|
||||
del self.cache[secret]
|
||||
print("remove_secret", secret)
|
||||
print(await self.verify_secret(secret))
|
||||
|
||||
|
||||
api_secret_utils = ApiSecretUtils()
|
||||
Reference in New Issue
Block a user