Finished tweaking 2FA

This commit is contained in:
Donald Zou
2025-06-03 23:37:43 +08:00
parent 2a1a885056
commit d775fb69e3
3 changed files with 23 additions and 7 deletions

View File

@@ -39,9 +39,22 @@ class DashboardClientsTOTP:
"ExpireTime": datetime.datetime.now() + datetime.timedelta(minutes=10)
})
)
return token
def RevokeToken(self, Token) -> bool:
try:
with self.engine.begin() as conn:
conn.execute(
self.dashboardClientsTOTPTable.update().values({
"ExpireTime": datetime.datetime.now()
}).where(
self.dashboardClientsTOTPTable.c.Token == Token
)
)
except Exception as e:
return False
return True
def GetTotp(self, token: str) -> tuple[bool, dict] or tuple[bool, None]:
with self.engine.connect() as conn:
totp = conn.execute(