mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-13 07:18:25 +01:00
Alexander GuschinandOded Messer
45181504ea
* api call to add/write artifact * return dvcyaml path on add * create dir for dvc.yaml if needed * create dir for dvc.yaml if needed - fix * allow nested artifact names * make dvc get download artifacts * Apply suggestions from code review Co-authored-by: Oded Messer <[email protected]> * use feedback * remove 'dvc get' from this PR, keep only adding artifacts * allow first char of artifact name to be a number * add note about constants * fix parsing artifact name * remove 'parse_name' and write tests for 'add' * Apply suggestions from code review Co-authored-by: Oded Messer <[email protected]> * updates from comments * fix for studio * fix again * making check_name_format to raise an exception * Update dvc/repo/artifacts.py --------- Co-authored-by: Oded Messer <[email protected]>
59 lines
1.6 KiB
Python
59 lines
1.6 KiB
Python
from dataclasses import asdict, dataclass, field, fields
|
|
from typing import Any, ClassVar, Dict, List, Optional
|
|
|
|
from funcy import compact
|
|
from voluptuous import Required
|
|
|
|
|
|
@dataclass
|
|
class Annotation:
|
|
PARAM_DESC: ClassVar[str] = "desc"
|
|
PARAM_TYPE: ClassVar[str] = "type"
|
|
PARAM_LABELS: ClassVar[str] = "labels"
|
|
PARAM_META: ClassVar[str] = "meta"
|
|
|
|
desc: Optional[str] = None
|
|
type: Optional[str] = None # noqa: A003
|
|
labels: List[str] = field(default_factory=list)
|
|
meta: Dict[str, Any] = field(default_factory=dict)
|
|
|
|
def update(self, **kwargs) -> "Annotation":
|
|
for attr, value in kwargs.items():
|
|
if value and hasattr(self, attr):
|
|
setattr(self, attr, value)
|
|
return self
|
|
|
|
def to_dict(self) -> Dict[str, str]:
|
|
return compact(asdict(self))
|
|
|
|
|
|
@dataclass
|
|
class Artifact:
|
|
PARAM_PATH: ClassVar[str] = "path"
|
|
PARAM_DESC: ClassVar[str] = "desc"
|
|
PARAM_TYPE: ClassVar[str] = "type"
|
|
PARAM_LABELS: ClassVar[str] = "labels"
|
|
PARAM_META: ClassVar[str] = "meta"
|
|
|
|
path: str
|
|
desc: Optional[str] = None
|
|
type: Optional[str] = None # noqa: A003
|
|
labels: List[str] = field(default_factory=list)
|
|
meta: Dict[str, Any] = field(default_factory=dict)
|
|
|
|
def to_dict(self) -> Dict[str, str]:
|
|
return compact(asdict(self))
|
|
|
|
|
|
ANNOTATION_FIELDS = [field.name for field in fields(Annotation)]
|
|
ANNOTATION_SCHEMA = {
|
|
Annotation.PARAM_DESC: str,
|
|
Annotation.PARAM_TYPE: str,
|
|
Annotation.PARAM_LABELS: [str],
|
|
Annotation.PARAM_META: object,
|
|
}
|
|
ARTIFACT_SCHEMA = {
|
|
Required(Artifact.PARAM_PATH): str,
|
|
**ANNOTATION_SCHEMA, # type: ignore[arg-type]
|
|
}
|