<reponame>robinandeer/puzzle
# -*- coding: utf-8 -*-
from sqlalchemy import Column, ForeignKey, Integer, String, UniqueConstraint
from sqlalchemy.orm import relationship

from .models import BASE


class CaseGenelistLink(BASE):

    """Link between case and gene list."""

    __tablename__ = 'case_genelist_link'
    __table_args__ = (UniqueConstraint('case_id', 'genelist_id',
                                       name='_case_genelist_uc'),)

    id = Column(Integer, primary_key=True)
    case_id = Column(Integer, ForeignKey('case.id'))
    genelist_id = Column(Integer, ForeignKey('gene_list.id'))


class GeneList(BASE):

    """Represent a list of gene identifiers."""

    __tablename__ = "gene_list"

    id = Column(Integer, primary_key=True)
    list_id = Column(String(32), nullable=False, unique=True)
    # comma separated list of gene ids
    _gene_ids = Column(String(1024))

    cases = relationship('Case', secondary='case_genelist_link',
                         backref='gene_lists')

    @property
    def gene_ids(self):
        """Return a list of gene ids."""
        return self._gene_ids.split(',') if self._gene_ids else []

    @gene_ids.setter
    def gene_ids(self, value):
        self._gene_ids = ','.join(value)

    def delete_gene(self, *gene_ids):
        """Delete one or more gene ids form the list."""
        self.gene_ids = [gene_id for gene_id in self.gene_ids
                         if gene_id not in gene_ids]

    def __repr__(self):
        return "PhenotypeTerm(list_id={this.list_id})".format(this=self)
