mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-07 23:09:21 +01:00
Alexander Guschin
32bebc0246
* add artifacts section * adding and removing artifacts * delete code for add/remove; add tests * fix pre-commit * remove extra piece of writing to dvc.yaml * remove extra piece * simplify kwargs * allow same names for artifacts in different dvc.yaml files * set ID as path; don't return dvc.yaml(s) with no artifacts * fix PR feedback * issue warning at incorrect artifact name * issue warning at incorrect artifact name * reference right schema * add r to regexp * add tests for regexp * set artifacts for SingleStageFile * fix windows test * make path required; simplify regexp and remove slash from it * allow to use gto's artifacts.yaml * remove extra piece of code for migration from gto
47 lines
1.2 KiB
Python
47 lines
1.2 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(Annotation):
|
|
PARAM_PATH: ClassVar[str] = "path"
|
|
path: Optional[str] = None
|
|
|
|
|
|
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]
|
|
}
|