From 717f079bd28619ce59bacdedc7e648dfdb82a6b8 Mon Sep 17 00:00:00 2001 From: YiLi Date: Mon, 6 Nov 2023 04:50:51 -0600 Subject: [PATCH] fix: Use correct typehint for return values (#1454) Co-authored-by: lethe --- api/services/account_service.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/api/services/account_service.py b/api/services/account_service.py index 3bd7b802341752..58c315c4249b55 100644 --- a/api/services/account_service.py +++ b/api/services/account_service.py @@ -6,7 +6,7 @@ import uuid from datetime import datetime, timedelta from hashlib import sha256 -from typing import Optional +from typing import Optional, Dict, Any from werkzeug.exceptions import Forbidden, Unauthorized from flask import session, current_app @@ -24,7 +24,8 @@ from models.account import * from tasks.mail_invite_member_task import send_invite_member_mail_task -def _create_tenant_for_account(account): + +def _create_tenant_for_account(account) -> Tenant: tenant = TenantService.create_tenant(f"{account.name}'s Workspace") TenantService.create_tenant_member(tenant, account, role='owner') @@ -505,7 +506,7 @@ def revoke_token(cls, workspace_id: str, email: str, token: str): redis_client.delete(cls._get_invitation_token_key(token)) @classmethod - def get_invitation_if_token_valid(cls, workspace_id: str, email: str, token: str) -> Optional[Account]: + def get_invitation_if_token_valid(cls, workspace_id: str, email: str, token: str) -> Optional[Dict[str, Any]]: invitation_data = cls._get_invitation_by_token(token, workspace_id, email) if not invitation_data: return None @@ -539,7 +540,7 @@ def get_invitation_if_token_valid(cls, workspace_id: str, email: str, token: str } @classmethod - def _get_invitation_by_token(cls, token: str, workspace_id: str, email: str) -> Optional[str]: + def _get_invitation_by_token(cls, token: str, workspace_id: str, email: str) -> Optional[Dict[str, str]]: if workspace_id is not None and email is not None: email_hash = sha256(email.encode()).hexdigest() cache_key = f'member_invite_token:{workspace_id}, {email_hash}:{token}'