def __eq__(self, other):
# Checking the UUID is mostly sufficient but requiring an exact type
# match makes it safer in case `other` is a subclass or a completely
# different type that happens to have a `.uuid` property. We want to
# ensure (as best as we can) that the UUIDs we are comparing are linked
# to the same type of QIIME 2 object.
return (type(self) == type(other) and self.uuid == other.uuid)
def __eq__(self, other):
# Checking the UUID is mostly sufficient but requiring an exact type
# match makes it safer in case `other` is a subclass or a completely
# different type that happens to have a `.uuid` property. We want to
# ensure (as best as we can) that the UUIDs we are comparing are linked
# to the same type of QIIME 2 object.
return (
type(self) == type(other) and
self.uuid == other.uuid
)
def import_data(cls, type, view, view_type=None):
type_, type = type, __builtins__['type']
is_format = False
if isinstance(type_, str):
type_ = qiime2.sdk.parse_type(type_)
if isinstance(view_type, str):
view_type = qiime2.sdk.parse_format(view_type)
is_format = True
if view_type is None:
if type(view) is str or isinstance(view, pathlib.PurePath):
is_format = True
pm = qiime2.sdk.PluginManager()
output_dir_fmt = pm.get_directory_format(type_)
if pathlib.Path(view).is_file():
if not issubclass(output_dir_fmt,
model.SingleFileDirectoryFormatBase):
raise qiime2.plugin.ValidationError(
"Importing %r requires a directory, not %s" %
(output_dir_fmt.__name__, view))
view_type = output_dir_fmt.file.format
else:
view_type = output_dir_fmt
else:
view_type = type(view)
format_ = None
md5sums = None
if is_format:
path = pathlib.Path(view)
if path.is_file():
md5sums = {path.name: util.md5sum(path)}
elif path.is_dir():
md5sums = util.md5sum_directory(path)
else:
raise qiime2.plugin.ValidationError(
"Path '%s' does not exist." % path)
format_ = view_type
provenance_capture = archive.ImportProvenanceCapture(format_, md5sums)
return cls._from_view(type_,
view,
view_type,
provenance_capture,
validate_level='max')
def import_data(cls, type, view, view_type=None):
type_, type = type, __builtins__['type']
is_format = False
if isinstance(type_, str):
type_ = qiime2.sdk.parse_type(type_)
if isinstance(view_type, str):
view_type = qiime2.sdk.parse_format(view_type)
is_format = True
if view_type is None:
if type(view) is str or isinstance(view, pathlib.PurePath):
is_format = True
pm = qiime2.sdk.PluginManager()
output_dir_fmt = pm.get_directory_format(type_)
if pathlib.Path(view).is_file():
if not issubclass(output_dir_fmt,
model.SingleFileDirectoryFormatBase):
raise qiime2.plugin.ValidationError(
"Importing %r requires a directory, not %s"
% (output_dir_fmt.__name__, view))
view_type = output_dir_fmt.file.format
else:
view_type = output_dir_fmt
else:
view_type = type(view)
format_ = None
md5sums = None
if is_format:
path = pathlib.Path(view)
if path.is_file():
md5sums = {path.name: util.md5sum(path)}
elif path.is_dir():
md5sums = util.md5sum_directory(path)
else:
raise qiime2.plugin.ValidationError(
"Path '%s' does not exist." % path)
format_ = view_type
provenance_capture = archive.ImportProvenanceCapture(format_, md5sums)
return cls._from_view(type_, view, view_type, provenance_capture)
def load(cls, filepath):
"""Factory for loading Artifacts and Visualizations."""
archiver = archive.Archiver.load(filepath)
if Artifact._is_valid_type(archiver.type):
result = Artifact.__new__(Artifact)
elif Visualization._is_valid_type(archiver.type):
result = Visualization.__new__(Visualization)
else:
raise TypeError(
"Cannot load filepath %r into an Artifact or Visualization "
"because type %r is not supported." %
(filepath, archiver.type))
if type(result) is not cls and cls is not Result:
raise TypeError(
"Attempting to load %s with `%s.load`. Use `%s.load` instead."
% (type(result).__name__, cls.__name__, type(result).__name__))
result._archiver = archiver
return result
def load(cls, filepath):
"""Factory for loading Artifacts and Visualizations."""
archiver = archive.Archiver.load(filepath)
if Artifact._is_valid_type(archiver.type):
result = Artifact.__new__(Artifact)
elif Visualization._is_valid_type(archiver.type):
result = Visualization.__new__(Visualization)
else:
raise TypeError(
"Cannot load filepath %r into an Artifact or Visualization "
"because type %r is not supported."
% (filepath, archiver.type))
if type(result) is not cls and cls is not Result:
raise TypeError(
"Attempting to load %s with `%s.load`. Use `%s.load` instead."
% (type(result).__name__, cls.__name__,
type(result).__name__))
result._archiver = archiver
return result
def _alias(self, provenance_capture):
def clone_original(into):
# directory is empty, this function is meant to fix that, so we
# can rmdir so that copytree is happy
into.rmdir()
shutil.copytree(str(self._archiver.data_dir), str(into),
copy_function=os.link) # Use hardlinks
cls = type(self)
alias = cls.__new__(cls)
alias._archiver = archive.Archiver.from_data(
self.type, self.format, clone_original, provenance_capture)
return alias
def _alias(self, provenance_capture):
def clone_original(into):
# directory is empty, this function is meant to fix that, so we
# can rmdir so that copytree is happy
into.rmdir()
shutil.copytree(str(self._archiver.data_dir), str(into),
copy_function=os.link) # Use hardlinks
cls = type(self)
alias = cls.__new__(cls)
alias._archiver = archive.Archiver.from_data(
self.type, self.format, clone_original, provenance_capture)
return alias
