# coding=utf-8
#author:        4N
#createtime:    2022/3/17
#email:         nheweijun@sina.com

from app.modules.auth.models import User

class UserAlias:
    def __init__(self):
        self.creator_dict = {}

    def get_alias(self,creator):
        alias = self.creator_dict.get(creator)
        if alias:
            return alias
        else:
            user = User.query.filter_by(username=creator).one_or_none()
            if user:
                self.creator_dict[creator] = user.displayname
            else:
                self.creator_dict[creator] = ""
            return self.creator_dict[creator]