feat: 秘钥凭证管理

This commit is contained in:
martsforever
2025-09-24 15:31:31 +08:00
parent 7d1f804ff0
commit 4218fed8c9
6 changed files with 133 additions and 5 deletions
+34
View File
@@ -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()