Source code for pybpodgui_api.models.subject.subject_base

# !/usr/bin/python3
# -*- coding: utf-8 -*-

import logging
import os
import uuid
from pybpodgui_api.models.session import Session
from pybpodgui_api.utils.generate_name import generate_name

logger = logging.getLogger(__name__)


class SubjectBase(object):

    def __init__(self, project):
        self._path = None
        self.uuid4 = uuid.uuid4()

        self.name = generate_name([x.name for x in project.subjects], "subject")

        self.project = project
        self.setup = None

        self.project += self

        self._sessions = []

    ##########################################################################
    ####### PROPERTIES #######################################################
    ##########################################################################

    def __add__(self, value):
        if isinstance(value, Session):
            self._sessions.append(value)
        return self

    def __sub__(self, obj):
        if isinstance(obj, Session) and obj in self._sessions:
            self._sessions.remove(obj)

    def remove(self):
        print('subject base removing session')

    def get_sessions(self):
        """
        Get all subject sessions

        :rtype: list(Session)
        """
        print('subject base get sessions')
        for exp in self.project.experiments:
            for setup in exp.setups:
                for session in setup.sessions:
                    if self in session.subjects:
                        yield session

        return None

    @property
    def name(self):
        """
        Get and set setup name

        :rtype: str
        """
        return self._name

    @name.setter
    def name(self, value):
        self._name = value

    @property
    def project(self):
        """
        Get and set project

        :rtype: str
        """
        return self._project

    @project.setter
    def project(self, value):
        self._project = value

    @property
    def setup(self):
        """
        Get and set project

        :rtype: str
        """
        return self._setup

    @setup.setter
    def setup(self, value):
        self._setup = value

    @property
    def path(self):
        """
        Get and set the path

        :rtype: str
        """
        if self.project.path is None:
            return None
        return os.path.join(self.project.path, 'subjects', self.name)

    ##########################################################################
    ####### FUNCTIONS ########################################################
    ##########################################################################

    def remove(self):
        """
        Remove the subject from the project
        """
        pass

    def __unicode__(self):
        return self.name

    def __str__(self):
        return self.__unicode__()