matteoferla/Fragmenstein

__all__ = ['pyrosetta']

from importlib import util, machinery
import sys
import warnings


class AttributeFilledMock:
    """
    This allows:

    .. code-block:: python
        mock = AttributeFilledMock()
        mock.me
        mock.me.again.over()
    """

    warning_msg = 'PyRosetta is not installed! Yet it has been called. ' + \
                  'The mock object taking its place does nothing.' + \
                  'This is bound to raise an error'
    __spec__ = machinery.ModuleSpec('pyrosetta', None)
    __signature__ = None

    def __getattr__(self, attr: str):
        return self

    def __call__(self, *args, **kargs):
        warnings.warn('This call does nothing as PyRosetta is not installed',
                      category=RuntimeWarning)
        return self


# ======================================================================================================================

if util.find_spec('pyrosetta'):
    import pyrosetta
else:
    warnings.warn('PyRosetta is not installed. A mock object is loaded. Any Igor calls will fail.',
                  category=RuntimeWarning)
    pyrosetta = AttributeFilledMock()
    sys.modules['pyrosetta'] = pyrosetta