encrypter.py 561 B

123456789101112131415161718192021
  1. import base64
  2. from extensions.ext_database import db
  3. from libs import rsa
  4. from models.account import Tenant
  5. def obfuscated_token(token: str):
  6. return token[:6] + '*' * (len(token) - 8) + token[-2:]
  7. def encrypt_token(tenant_id: str, token: str):
  8. tenant = db.session.query(Tenant).filter(Tenant.id == tenant_id).first()
  9. encrypted_token = rsa.encrypt(token, tenant.encrypt_public_key)
  10. return base64.b64encode(encrypted_token).decode()
  11. def decrypt_token(tenant_id: str, token: str):
  12. return rsa.decrypt(base64.b64decode(token), tenant_id)