a factory with name mapping for js-ng

mappedfactory.py

"""
Redis client
"""


from jumpscale.core.base import Base, StoredFactory
from jumpscale.core.base import fields

class Project(Base):
    name = fields.String()
    total_tasks = fields.Integer()


def map_name(method):
    def inner(self, name, *args, **kwargs):
        if not name:
            return method(name, *args, **kwargs)
        
        if not name.startswith('__'):
            instance = method(self, self.get_identifier(name), *args, **kwargs)
            if hasattr(instance, "name"):
                instance.name = name
        else:
            instance = method(self, name, *args, **kwargs)
        
        return instance
    return inner
    
class CustomFactory(StoredFactory):
    def get_identifier(self, name):
        """returns a new unique name that can be used as an identifier"""
        hashed = j.data.hash.md5(name)
        # prefix this hashed value, so it will be hidden
        # and also always a str (md5 can start with a number)
        return f"__{hashed}"

    @map_name
    def find(self, name):
        return super().find(name)

    @map_name
    def new(self, name, *args, **kwargs):
        instance = super().new(name, *args, **kwargs)
        return instance
    
    @map_name
    def get(self, name, *args, **kwargs):
        return super().get(name, *args, **kwargs)

    def list_all(self):
        names = super().list_all()
        return [self.get(name).name for name in names]

class Account(Base):
    name = fields.String()
    projects = fields.Factory(Project, factory_type=CustomFactory)


accounts = CustomFactory(Account)