# -*- coding: utf-8 -*-
#
# Copyright 2017-2020 - Swiss Data Science Center (SDSC)
# A partnership between École Polytechnique Fédérale de Lausanne (EPFL) and
# Eidgenössische Technische Hochschule Zürich (ETHZ).
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""Model objects representing datasets."""
import datetime
import os
import pathlib
import re
import urllib
import uuid
from functools import partial
from pathlib import Path
import attr
from attr.validators import instance_of
from renku.core import errors
from renku.core.models.entities import Entity
from renku.core.models.provenance.agents import Person
from renku.core.models.refs import LinkReference
from renku.core.utils.datetime8601 import parse_date
from renku.core.utils.doi import extract_doi, is_doi
from . import jsonld as jsonld
NoneType = type(None)
_path_attr = partial(
jsonld.ib,
converter=Path,
)
@jsonld.s(
type='schema:URL',
context={
'schema': 'http://schema.org/',
},
)
class Url:
"""Represents a schema URL reference."""
url = jsonld.ib(default=None, kw_only=True, context='schema:url')
_id = jsonld.ib(kw_only=True, context='@id')
@_id.default
def default_id(self):
"""Define default value for id field."""
if self.url:
parsed_result = urllib.parse.urlparse(self.url)
id_ = urllib.parse.ParseResult('', *parsed_result[1:]).geturl()
else:
id_ = str(uuid.uuid4())
return '_:URL@{0}'.format(id_)
def _convert_creators(value):
"""Convert creators."""
if isinstance(value, dict): # compatibility with previous versions
return [Person.from_jsonld(value)]
if isinstance(value, list):
return [Person.from_jsonld(v) for v in value]
@attr.s
class CreatorMixin:
"""Mixin for handling creators container."""
creator = jsonld.container.list(
Person,
kw_only=True,
context='schema:creator',
converter=_convert_creators
)
@property
def creators_csv(self):
"""Comma-separated list of creators associated with dataset."""
return ', '.join(creator.name for creator in self.creator)
def _extract_doi(value):
"""Return either a string or the doi part of a URL."""
value = str(value)
if is_doi(value):
return extract_doi(value)
return value
@jsonld.s(
type='schema:PublicationEvent',
context={'schema': 'http://schema.org/'},
frozen=True,
slots=True,
)
class DatasetTag(object):
"""Represents a Tag of an instance of a dataset."""
client = attr.ib(default=None, kw_only=True)
name = jsonld.ib(
default=None,
kw_only=True,
validator=instance_of(str),
context='schema:name'
)
description = jsonld.ib(
default=None,
kw_only=True,
validator=instance_of(str),
context='schema:description'
)
commit = jsonld.ib(
default=None,
kw_only=True,
validator=instance_of(str),
context='schema:location'
)
created = jsonld.ib(
converter=parse_date, context='schema:startDate', kw_only=True
)
dataset = jsonld.ib(context='schema:about', default=None, kw_only=True)
_id = jsonld.ib(kw_only=True, context='@id')
@created.default
def _now(self):
"""Define default value for datetime fields."""
return datetime.datetime.now(datetime.timezone.utc)
@_id.default
def default_id(self):
"""Define default value for id field."""
return '_:{0}@{1}'.format(self.name, self.commit)
@jsonld.s(
type='schema:Language',
context={'schema': 'http://schema.org/'},
slots=True,
)
class Language:
"""Represent a language of an object."""
alternate_name = jsonld.ib(
default=None, kw_only=True, context='schema:alternateName'
)
name = jsonld.ib(default=None, kw_only=True, context='schema:name')
def convert_filename_path(p):
"""Return name of the file."""
if p:
return Path(p).name
def convert_based_on(v):
"""Convert based_on to DatasetFile."""
if v:
return DatasetFile.from_jsonld(v)
[docs]@jsonld.s(
type='schema:DigitalDocument',
slots=True,
context={
'schema': 'http://schema.org/',
}
)
class DatasetFile(Entity, CreatorMixin):
"""Represent a file in a dataset."""
added = jsonld.ib(
converter=parse_date, context='schema:dateCreated', kw_only=True
)
checksum = attr.ib(default=None, kw_only=True)
filename = attr.ib(kw_only=True, converter=convert_filename_path)
name = jsonld.ib(context='schema:name', kw_only=True, default=None)
filesize = attr.ib(default=None, kw_only=True)
filetype = attr.ib(default=None, kw_only=True)
url = jsonld.ib(default=None, context='schema:url', kw_only=True)
based_on = jsonld.ib(
default=None,
context='schema:isBasedOn',
kw_only=True,
converter=convert_based_on
)
@added.default
def _now(self):
"""Define default value for datetime fields."""
return datetime.datetime.now(datetime.timezone.utc)
[docs] @filename.default
def default_filename(self):
"""Generate default filename based on path."""
if self.path:
return Path(self.path).name
@property
def full_path(self):
"""Return full path in the current reference frame."""
path = Path(self.path)
if self.client:
return (self.client.path / path).resolve()
return path.resolve()
@property
def size_in_mb(self):
"""Return file size in megabytes."""
return self.filesize * 1e-6
def __attrs_post_init__(self):
"""Set the property "name" after initialization."""
super().__attrs_post_init__()
if not self.name:
self.name = self.filename
parsed_id = urllib.parse.urlparse(self._id)
if not parsed_id.scheme:
self._id = 'file://{}'.format(self._id)
def _convert_dataset_files(value):
"""Convert dataset files."""
coll = value
if isinstance(coll, dict): # compatibility with previous versions
if any([key.startswith('@') for key in coll.keys()]):
return [DatasetFile.from_jsonld(coll)]
else:
coll = value.values()
return [DatasetFile.from_jsonld(v) for v in coll]
def _convert_dataset_tags(value):
"""Convert dataset tags."""
if isinstance(value, dict): # compatibility with previous versions
value = [value]
return [DatasetTag.from_jsonld(v) for v in value]
def _convert_language(obj):
"""Convert language object."""
if isinstance(obj, dict):
language = Language.from_jsonld(obj)
return language
def _convert_keyword(keywords):
"""Convert keywords collection."""
if isinstance(keywords, list):
return keywords
if isinstance(keywords, dict):
return keywords.keys()
[docs]@jsonld.s(
type='schema:Dataset',
context={
'schema': 'http://schema.org/',
},
)
class Dataset(Entity, CreatorMixin):
"""Repesent a dataset."""
SUPPORTED_SCHEMES = ('', 'file', 'http', 'https', 'git+https', 'git+ssh')
EDITABLE_FIELDS = [
'creator', 'date_published', 'description', 'in_language', 'keywords',
'license', 'name', 'url', 'version', 'created', 'files'
]
_id = jsonld.ib(default=None, context='@id', kw_only=True)
_label = jsonld.ib(default=None, context='rdfs:label', kw_only=True)
date_published = jsonld.ib(
default=None, context='schema:datePublished', kw_only=True
)
description = jsonld.ib(
default=None, context='schema:description', kw_only=True
)
identifier = jsonld.ib(
default=attr.Factory(uuid.uuid4),
context='schema:identifier',
kw_only=True,
converter=_extract_doi
)
in_language = jsonld.ib(
type=Language,
default=None,
converter=_convert_language,
context='schema:inLanguage',
kw_only=True
)
keywords = jsonld.container.list(
str,
converter=_convert_keyword,
context='schema:keywords',
kw_only=True
)
based_on = jsonld.ib(
default=None, context='schema:isBasedOn', kw_only=True
)
license = jsonld.ib(default=None, context='schema:license', kw_only=True)
name = jsonld.ib(
default=None, type=str, context='schema:name', kw_only=True
)
url = jsonld.ib(default=None, context='schema:url', kw_only=True)
version = jsonld.ib(default=None, context='schema:version', kw_only=True)
created = jsonld.ib(
converter=parse_date, context='schema:dateCreated', kw_only=True
)
files = jsonld.container.list(
DatasetFile,
default=None,
converter=_convert_dataset_files,
context='schema:hasPart',
kw_only=True
)
tags = jsonld.container.list(
DatasetTag,
default=None,
converter=_convert_dataset_tags,
context={
'@id': 'schema:subjectOf',
},
kw_only=True
)
same_as = jsonld.ib(
context='schema:sameAs', default=None, kw_only=True, type=Url
)
short_name = jsonld.ib(
default=None, context='schema:alternateName', kw_only=True
)
@created.default
def _now(self):
"""Define default value for datetime fields."""
return datetime.datetime.now(datetime.timezone.utc)
[docs] @short_name.validator
def short_name_validator(self, attribute, value):
"""Validate short_name."""
# short_name might have been scaped and have '%' in it
if value and not is_dataset_short_name_valid(value):
raise errors.ParameterError(
'Invalid "short_name": {}'.format(value)
)
@property
def uid(self):
"""UUID part of identifier."""
if is_doi(self.identifier):
return self.identifier
return self.identifier.split('/')[-1]
@property
def short_id(self):
"""Shorter version of identifier."""
if is_doi(self.identifier):
return self.identifier
return str(self.uid)[:8]
@property
def creators_csv(self):
"""Comma-separated list of creators associated with dataset."""
return ', '.join(creator.name for creator in self.creator)
@property
def tags_csv(self):
"""Comma-separated list of tags associated with dataset."""
return ','.join(tag.name for tag in self.tags)
@property
def editable(self):
"""Subset of attributes which user can edit."""
obj = self.asjsonld()
data = {field_: obj.pop(field_) for field_ in self.EDITABLE_FIELDS}
return data
@property
def datadir(self):
"""Directory where dataset files are stored."""
if self.client:
return Path(self.client.datadir) / self.short_name
return ''
[docs] def contains_any(self, files):
"""Check if files are already within a dataset."""
for file_ in files:
if self.find_file(file_['path']):
return True
return False
[docs] def find_file(self, filename, return_index=False):
"""Find a file in files container."""
for index, file_ in enumerate(self.files):
if str(file_.path) == str(filename):
if return_index:
return index
return file_
[docs] def update_files(self, files):
"""Update files with collection of DatasetFile objects."""
to_insert = []
for new_file in files:
existing_file = self.find_file(new_file.path)
if existing_file is None:
to_insert.append(new_file)
else:
existing_file.commit = new_file.commit
existing_file._label = new_file._label
existing_file.based_on = new_file.based_on
self.files += to_insert
[docs] def rename_files(self, rename):
"""Rename files using the path mapping function."""
files = []
for file_ in self.files:
new_path = rename(file_.path)
new_file = attr.evolve(file_, path=new_path)
if not self.find_file(new_file.path):
files.append(new_file)
else:
raise FileExistsError
renamed = attr.evolve(self, files=files)
setattr(renamed, '__reference__', self.__reference__)
if self.__source__:
setattr(renamed, '__source__', self.__source__.copy())
return renamed
[docs] def unlink_file(self, file_path):
"""Unlink a file from dataset.
:param file_path: Relative path used as key inside files container.
"""
index = self.find_file(file_path, return_index=True)
return self.files.pop(index)
def __attrs_post_init__(self):
"""Post-Init hook."""
from urllib.parse import quote
super().__attrs_post_init__()
# Determine the hostname for the resource URIs.
# If RENKU_DOMAIN is set, it overrides the host from remote.
# Default is localhost.
host = 'localhost'
if self.client:
host = self.client.remote.get('host') or host
host = os.environ.get('RENKU_DOMAIN') or host
# always set the id by the identifier
self._id = urllib.parse.urljoin(
'https://{host}'.format(host=host),
pathlib.posixpath.join(
'/datasets', quote(self.identifier, safe='')
)
)
if not self.url:
# set the url for non-imported datasets
self.url = self._id
# if `date_published` is set, we are probably dealing with
# an imported dataset so `created` is not needed
if self.date_published:
self.created = None
self._label = self.identifier
if not self.path:
self.path = str(
self.client.renku_datasets_path /
quote(str(self.uid), safe='')
)
if self.files:
for dataset_file in self.files:
file_exists = Path(dataset_file.path).exists()
if dataset_file.client is None and file_exists:
client, _, _ = self.client.resolve_in_submodules(
self.client.find_previous_commit(
dataset_file.path, revision='HEAD'
),
dataset_file.path,
)
dataset_file.client = client
try:
if self.client:
self.commit = self.client.find_previous_commit(
self.path, revision=self.commit or 'HEAD'
)
except KeyError:
# if with_dataset is used, the dataset is not committed yet
pass
if not self.short_name:
self.short_name = generate_default_short_name(
self.name, self.version
)
def is_dataset_short_name_valid(short_name):
"""A valid short_name is a valid Git reference name with no /."""
# TODO make short_name an RFC 3986 compatible and migrate old projects
return (
short_name and
LinkReference.check_ref_format(short_name, no_slashes=True) and
'/' not in short_name
)
def generate_default_short_name(dataset_name, dataset_version):
"""Get dataset short_name."""
# For compatibility with older versions use name as short_name
# if it is valid; otherwise, use encoded name
if is_dataset_short_name_valid(dataset_name):
return dataset_name
name = re.sub(r'\s+', ' ', dataset_name)
name = name.lower()[:24]
def to_unix(el):
"""Parse string to unix friendly name."""
parsed_ = re.sub('[^a-zA-Z0-9]', '', re.sub(r'\s+', ' ', el))
parsed_ = re.sub(' .+', '.', parsed_.lower())
return parsed_
short_name = [to_unix(el) for el in name.split()]
short_name = [el for el in short_name if el]
if dataset_version:
version = to_unix(dataset_version)
name = '{0}_{1}'.format('_'.join(short_name), version)
return name
return '_'.join(short_name)