FateWithShellmiaoComment/FATE-Flow/python/fate_flow/entity/_data_cache.py

#
#  Copyright 2019 The FATE Authors. All Rights Reserved.
#
#  Licensed under the Apache License, Version 2.0 (the "License");
#  you may not use this file except in compliance with the License.
#  You may obtain a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
#  Unless required by applicable law or agreed to in writing, software
#  distributed under the License is distributed on an "AS IS" BASIS,
#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#  See the License for the specific language governing permissions and
#  limitations under the License.
#
import typing
from ._base import BaseEntity
from fate_arch.common import DTable


class DataCache(BaseEntity):
    def __init__(self, name: str, key: str = None, data: typing.Dict[str, DTable] = None, meta: dict = None, job_id: str = None, component_name: str = None, task_id: str = None, task_version: int = None):
        self._name: str = name
        self._key: str = key
        self._data: typing.Dict[str, DTable] = data if data else {}
        self._meta: dict = meta
        self._job_id = job_id
        self._component_name = component_name
        self._task_id: str = task_id
        self._task_version: int = task_version

    @property
    def name(self):
        return self._name

    @property
    def key(self):
        return self._key

    @key.setter
    def key(self, key: str):
        self._key = key

    @property
    def data(self):
        return self._data

    @property
    def meta(self):
        return self._meta

    @property
    def job_id(self):
        return self._job_id

    @job_id.setter
    def job_id(self, job_id: str):
        self._job_id = job_id

    @property
    def component_name(self):
        return self._component_name

    @component_name.setter
    def component_name(self, component_name: str):
        self._component_name = component_name

    @property
    def task_id(self):
        return self._task_id

    @task_id.setter
    def task_id(self, task_id: str):
        self._task_id = task_id

    @property
    def task_version(self):
        return self._task_version

    @task_version.setter
    def task_version(self, task_version: int):
        self._task_version = task_version