mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-12 07:18:23 +01:00
83 lines
2.1 KiB
Python
83 lines
2.1 KiB
Python
import os
|
|
|
|
from dvc.compat import fspath
|
|
|
|
|
|
class BaseTree(object):
|
|
"""Abstract class to represent access to files"""
|
|
|
|
@property
|
|
def tree_root(self):
|
|
pass
|
|
|
|
def open(self, path, mode="r", encoding="utf-8"):
|
|
"""Open file and return a stream."""
|
|
|
|
def exists(self, path):
|
|
"""Test whether a path exists."""
|
|
|
|
def isdir(self, path):
|
|
"""Return true if the pathname refers to an existing directory."""
|
|
|
|
def isfile(self, path):
|
|
"""Test whether a path is a regular file"""
|
|
|
|
def walk(self, top, topdown=True):
|
|
"""Directory tree generator.
|
|
|
|
See `os.walk` for the docs. Differences:
|
|
- no support for symlinks
|
|
- it could raise exceptions, there is no onerror argument
|
|
"""
|
|
|
|
|
|
class WorkingTree(BaseTree):
|
|
"""Proxies the repo file access methods to working tree files"""
|
|
|
|
def __init__(self, repo_root=os.getcwd()):
|
|
self.repo_root = repo_root
|
|
|
|
@property
|
|
def tree_root(self):
|
|
return self.repo_root
|
|
|
|
def open(self, path, mode="r", encoding="utf-8"):
|
|
"""Open file and return a stream."""
|
|
return open(path, mode=mode, encoding=encoding)
|
|
|
|
def exists(self, path):
|
|
"""Test whether a path exists."""
|
|
return os.path.exists(path)
|
|
|
|
def isdir(self, path):
|
|
"""Return true if the pathname refers to an existing directory."""
|
|
return os.path.isdir(path)
|
|
|
|
def isfile(self, path):
|
|
"""Test whether a path is a regular file"""
|
|
return os.path.isfile(path)
|
|
|
|
def walk(self, top, topdown=True):
|
|
"""Directory tree generator.
|
|
|
|
See `os.walk` for the docs. Differences:
|
|
- no support for symlinks
|
|
- it could raise exceptions, there is no onerror argument
|
|
"""
|
|
|
|
top = fspath(top)
|
|
|
|
def onerror(e):
|
|
raise e
|
|
|
|
for root, dirs, files in os.walk(
|
|
top, topdown=topdown, onerror=onerror
|
|
):
|
|
yield os.path.normpath(root), dirs, files
|
|
|
|
|
|
def is_working_tree(tree):
|
|
return isinstance(tree, WorkingTree) or isinstance(
|
|
getattr(tree, "tree", None), WorkingTree
|
|
)
|