mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-10 07:18:27 +01:00
Alexander Schepanovski
d80293c44d
Replaced Path* dataclasses with PathInfo/URLInfo/CloudURLInfo, which know how parse, serialize, stringify and join themselves. They also provide useful accessors like part of an url or pathlib.Path like .parent or .name. Along the way two families of bugs where fixed and should never occur in the future: - paths not matched as strings anymore, e.g. using str.startswith() - Output.path is eliminated as ambiguous Both families might had caused undefined surprising behavior when local and remote paths clash.
112 lines
3.1 KiB
Python
112 lines
3.1 KiB
Python
from __future__ import unicode_literals
|
|
|
|
import os
|
|
import psutil
|
|
import platform
|
|
import argparse
|
|
import logging
|
|
import uuid
|
|
|
|
from dvc.utils.compat import pathlib
|
|
from dvc.repo import Repo
|
|
from dvc.command.base import CmdBaseNoRepo, append_doc_link
|
|
from dvc.version import __version__
|
|
from dvc.exceptions import DvcException, NotDvcRepoError
|
|
from dvc.system import System
|
|
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class CmdVersion(CmdBaseNoRepo):
|
|
def run(self):
|
|
dvc_version = __version__
|
|
python_version = platform.python_version()
|
|
platform_type = platform.platform()
|
|
|
|
info = (
|
|
"DVC version: {dvc_version}\n"
|
|
"Python version: {python_version}\n"
|
|
"Platform: {platform_type}\n"
|
|
).format(
|
|
dvc_version=dvc_version,
|
|
python_version=python_version,
|
|
platform_type=platform_type,
|
|
)
|
|
|
|
try:
|
|
repo = Repo()
|
|
root_directory = repo.root_dir
|
|
|
|
info += (
|
|
"Cache: {cache}\n"
|
|
"Filesystem type (cache directory): {fs_cache}\n"
|
|
).format(
|
|
cache=self.get_linktype_support_info(repo),
|
|
fs_cache=self.get_fs_type(repo.cache.local.cache_dir),
|
|
)
|
|
|
|
except NotDvcRepoError:
|
|
root_directory = os.getcwd()
|
|
|
|
info += ("Filesystem type (workspace): {fs_root}").format(
|
|
fs_root=self.get_fs_type(os.path.abspath(root_directory))
|
|
)
|
|
logger.info(info)
|
|
return 0
|
|
|
|
@staticmethod
|
|
def get_fs_type(path):
|
|
partition = {
|
|
pathlib.Path(part.mountpoint): (part.fstype, part.device)
|
|
for part in psutil.disk_partitions()
|
|
}
|
|
for parent in pathlib.Path(path).parents:
|
|
if parent in partition:
|
|
return partition[parent]
|
|
return ("unkown", "none")
|
|
|
|
@staticmethod
|
|
def get_linktype_support_info(repo):
|
|
links = {
|
|
"reflink": System.reflink,
|
|
"hardlink": System.hardlink,
|
|
"symlink": System.symlink,
|
|
}
|
|
|
|
fname = "." + str(uuid.uuid4())
|
|
src = os.path.join(repo.cache.local.cache_dir, fname)
|
|
open(src, "w").close()
|
|
dst = os.path.join(repo.root_dir, fname)
|
|
|
|
cache = []
|
|
|
|
for name, link in links.items():
|
|
try:
|
|
link(src, dst)
|
|
os.unlink(dst)
|
|
supported = True
|
|
except DvcException:
|
|
supported = False
|
|
cache.append(
|
|
"{name} - {supported}".format(
|
|
name=name, supported=True if supported else False
|
|
)
|
|
)
|
|
os.remove(src)
|
|
|
|
return ", ".join(cache)
|
|
|
|
|
|
def add_parser(subparsers, parent_parser):
|
|
VERSION_HELP = "Show DVC version and system/environment informaion."
|
|
|
|
version_parser = subparsers.add_parser(
|
|
"version",
|
|
parents=[parent_parser],
|
|
description=append_doc_link(VERSION_HELP, "version"),
|
|
help=VERSION_HELP,
|
|
formatter_class=argparse.RawDescriptionHelpFormatter,
|
|
)
|
|
version_parser.set_defaults(func=CmdVersion)
|