Source code for proteinshake.transforms.transforms

"""
Abstract class for transforming a protein.
"""

class Transform:
    """ A callable object which accepts a protein dictionary and returns an updated version of it."""
    def __init__(self):
        pass

    def __call__(self, protein):
        """ Accepts a protein dict and returns a new protein dict

        Arguments
        ----------
        protein: dict
            A protein dictionary.

        """
        raise NotImplementedError

class IdentityTransform: """ Do nothing to the protein""" def __call__(self, protein): return protein