mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-09 15:28:26 +01:00
Saugat Pachhai
6c33fbb617
* introduce index * migrate tests to use index * migrate checkout/diff to use index.outs * test for index * fix typings * add few more tests
35 lines
809 B
Python
35 lines
809 B
Python
import logging
|
|
from typing import TYPE_CHECKING, Iterator, Union
|
|
|
|
from .tree import Tree
|
|
|
|
if TYPE_CHECKING:
|
|
from dvc.hash_info import HashInfo
|
|
|
|
from .db.base import ObjectDB
|
|
from .file import HashFile
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def check(odb: "ObjectDB", obj: "HashFile", **kwargs):
|
|
if isinstance(obj, Tree):
|
|
for _, entry in obj:
|
|
odb.check(entry.hash_info, **kwargs)
|
|
|
|
odb.check(obj.hash_info, **kwargs)
|
|
|
|
|
|
def load(odb: "ObjectDB", hash_info: "HashInfo") -> "HashFile":
|
|
if hash_info.isdir:
|
|
return Tree.load(odb, hash_info)
|
|
return odb.get(hash_info)
|
|
|
|
|
|
def iterobjs(
|
|
obj: Union["Tree", "HashFile"]
|
|
) -> Iterator[Union["Tree", "HashFile"]]:
|
|
if isinstance(obj, Tree):
|
|
yield from (entry_obj for _, entry_obj in obj)
|
|
yield obj
|