mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-02 07:18:25 +01:00
Peter Rowlands (변기호)
b88b3b81c6
* GitTree: add stat() function * BaseTree: make walk_files a base tree method * WorkingTree: add stat() method (wrapper for os.stat) * utils.fs: support non-working tree for get_mtime_and_size * utils.fs: add copy_fobj_to_file * utils: support using tree.open in file_md5
95 lines
2.4 KiB
Python
95 lines
2.4 KiB
Python
import os
|
|
import stat
|
|
|
|
|
|
class BaseTree:
|
|
"""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
|
|
"""
|
|
|
|
def walk_files(self, top):
|
|
for root, _, files in self.walk(top):
|
|
for file in files:
|
|
yield os.path.join(root, file)
|
|
|
|
|
|
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."""
|
|
if "b" in mode:
|
|
encoding = None
|
|
return open(path, mode=mode, encoding=encoding)
|
|
|
|
def exists(self, path):
|
|
"""Test whether a path exists."""
|
|
return os.path.lexists(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
|
|
"""
|
|
|
|
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 isexec(self, path):
|
|
mode = os.stat(path).st_mode
|
|
return mode & (stat.S_IXUSR | stat.S_IXGRP | stat.S_IXOTH)
|
|
|
|
@staticmethod
|
|
def stat(path):
|
|
return os.stat(path)
|
|
|
|
|
|
def is_working_tree(tree):
|
|
return isinstance(tree, WorkingTree) or isinstance(
|
|
getattr(tree, "tree", None), WorkingTree
|
|
)
|