class _SourceRelation():
    """
    Mixin for classes that have a Source relation property
    """

    def _get_source(self):
        if self.source_id is None:
            return None
        from tdw_catalog.source import Source
        return Source.get(self._client, self.organization_id, self.source_id)


class _CredentialRelation():
    """
    Mixin for classes that have a Credential relation property
    """

    def _get_credential(self):
        if self.credential_id is None:
            return None
        from tdw_catalog.credential import Credential
        return Credential.get(self._client, self.organization_id,
                              self.credential_id)


class _ConnectionRelation():
    """
    Mixin for classes that have a Connection relation property
    """

    def _get_connection(self):
        if self.connection_id is None:
            return None
        from tdw_catalog.connection import _Connection
        return _Connection.get(self._client, self.connection_id)


class _OrganizationRelation():
    """
    Mixin for classes that have an Organization relation property
    """

    def _get_organization(self):
        if self.organization_id is None:
            return None
        import tdw_catalog.organization as organization
        return organization.Organization.get(self._client,
                                             self.organization_id)


class _TeamRelation():
    """
    Mixin for classes that have a Team relation property
    """

    def _get_team(self):
        if self.team_id is None:
            return None
        import tdw_catalog.team as team
        return team.Team.get(self._client, self.organization_id, self.team_id)


class _DatasetLineageRelationshipRelation():
    """
    Mixin for classes that have a DatasetLineageRelationship relation property
    """

    def _get_dataset_lineage_relationship(self):
        import tdw_catalog.lineage.dataset_relationship as dataset_relationship
        return dataset_relationship.DatasetLineageRelationship.get(
            self._client, self.dataset_lineage_relationship_id)