mirror of
https://github.com/langgenius/dify.git
synced 2024-11-16 19:59:50 +08:00
21 lines
561 B
Python
21 lines
561 B
Python
|
import base64
|
||
|
|
|
||
|
|
from extensions.ext_database import db
|
||
|
|
from libs import rsa
|
||
|
|
|
||
|
|
from models.account import Tenant
|
||
|
|
|
||
|
|
|
||
|
|
def obfuscated_token(token: str):
|
||
|
|
return token[:6] + '*' * (len(token) - 8) + token[-2:]
|
||
|
|
|
||
|
|
|
||
|
|
def encrypt_token(tenant_id: str, token: str):
|
||
|
|
tenant = db.session.query(Tenant).filter(Tenant.id == tenant_id).first()
|
||
|
|
encrypted_token = rsa.encrypt(token, tenant.encrypt_public_key)
|
||
|
|
return base64.b64encode(encrypted_token).decode()
|
||
|
|
|
||
|
|
|
||
|
|
def decrypt_token(tenant_id: str, token: str):
|
||
|
|
return rsa.decrypt(base64.b64decode(token), tenant_id)
|