import abc
import copy
from uuid import uuid4
from immuneML.data_model.DatasetItem import DatasetItem
from immuneML.util.NumpyHelper import NumpyHelper
[docs]
class Receptor(DatasetItem):
FIELDS = {}
def __init__(self, metadata: dict = None, identifier: str = None):
self.metadata = metadata
self.identifier = identifier if identifier is not None else uuid4().hex
[docs]
@abc.abstractmethod
def get_chains(self):
pass
[docs]
@classmethod
@abc.abstractmethod
def create_from_record(cls, record):
pass
[docs]
@classmethod
@abc.abstractmethod
def get_record_names(cls):
pass
[docs]
def clone(self):
copied_element = copy.deepcopy(self)
copied_element.identifier = str(uuid4().hex)
return copied_element
[docs]
def get_chain(self, chain: str):
return getattr(self, chain)
[docs]
def get_record(self):
chains = self.get_chains()
record = self.get_chain(chains[0]).get_record() + self.get_chain(chains[1]).get_record() \
+ [NumpyHelper.get_numpy_representation(getattr(self, name)) for name in self.FIELDS if name not in chains]
return record
[docs]
def get_attribute(self, name: str):
if hasattr(self, name):
return getattr(self, name)
if name in self.metadata.keys():
return self.metadata[name]
else:
raise KeyError(f"Receptor {self.identifier} does not have attribute {name}.")