mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-13 07:18:25 +01:00
53 lines
1.3 KiB
Python
53 lines
1.3 KiB
Python
from collections import OrderedDict
|
|
from dataclasses import dataclass, field
|
|
from typing import Dict, Optional
|
|
|
|
HASH_DIR_SUFFIX = ".dir"
|
|
|
|
DirInfo = Dict[str, str]
|
|
|
|
|
|
@dataclass
|
|
class HashInfo:
|
|
PARAM_SIZE = "size"
|
|
PARAM_NFILES = "nfiles"
|
|
|
|
name: Optional[str]
|
|
value: Optional[str]
|
|
dir_info: Optional[DirInfo] = field(default=None, compare=False)
|
|
size: Optional[int] = field(default=None, compare=False)
|
|
nfiles: Optional[int] = field(default=None, compare=False)
|
|
|
|
def __bool__(self):
|
|
return bool(self.value)
|
|
|
|
@classmethod
|
|
def from_dict(cls, d):
|
|
_d = d.copy() if d else {}
|
|
size = _d.pop(cls.PARAM_SIZE, None)
|
|
nfiles = _d.pop(cls.PARAM_NFILES, None)
|
|
|
|
if not _d:
|
|
return cls(None, None)
|
|
|
|
((name, value),) = _d.items()
|
|
return cls(name, value, size=size, nfiles=nfiles)
|
|
|
|
def to_dict(self):
|
|
ret = OrderedDict()
|
|
if not self:
|
|
return ret
|
|
|
|
ret[self.name] = self.value
|
|
if self.size is not None:
|
|
ret[self.PARAM_SIZE] = self.size
|
|
if self.nfiles is not None:
|
|
ret[self.PARAM_NFILES] = self.nfiles
|
|
return ret
|
|
|
|
@property
|
|
def isdir(self):
|
|
if not self:
|
|
return False
|
|
return self.value.endswith(HASH_DIR_SUFFIX)
|