mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-06 23:09:55 +01:00
Alexei Tenitski
235d4c9a94
* #1618 Custom message when cache and remote are in sync * Output the correct remote name during dvc status * Simplify logic
84 lines
2.5 KiB
Python
84 lines
2.5 KiB
Python
import logging
|
|
|
|
from dvc.command.data_sync import CmdDataBase
|
|
from dvc.exceptions import DvcException
|
|
from dvc.utils import format_link
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class CmdDataStatus(CmdDataBase):
|
|
STATUS_LEN = 20
|
|
STATUS_INDENT = "\t"
|
|
UP_TO_DATE_MSG = "Data and pipelines are up to date."
|
|
IN_SYNC_MSG = "Cache and remote '{remote}' are in sync."
|
|
EMPTY_PROJECT_MSG = (
|
|
"There are no data or pipelines tracked in this project yet.\n"
|
|
"See {link} to get started!"
|
|
).format(link=format_link("https://dvc.org/doc/start"))
|
|
|
|
def _normalize(self, s):
|
|
s += ":"
|
|
assert len(s) < self.STATUS_LEN
|
|
return s + (self.STATUS_LEN - len(s)) * " "
|
|
|
|
def _show(self, status, indent=0):
|
|
ind = indent * self.STATUS_INDENT
|
|
|
|
if isinstance(status, str):
|
|
logger.info(f"{ind}{status}")
|
|
return
|
|
|
|
if isinstance(status, list):
|
|
for entry in status:
|
|
self._show(entry, indent)
|
|
return
|
|
|
|
assert isinstance(status, dict)
|
|
|
|
for key, value in status.items():
|
|
if isinstance(value, str):
|
|
logger.info("{}{}{}".format(ind, self._normalize(value), key))
|
|
elif value:
|
|
logger.info(f"{ind}{key}:")
|
|
self._show(value, indent + 1)
|
|
|
|
def run(self):
|
|
indent = 1 if self.args.cloud else 0
|
|
try:
|
|
st = self.repo.status(
|
|
targets=self.args.targets,
|
|
jobs=self.args.jobs,
|
|
cloud=self.args.cloud,
|
|
remote=self.args.remote,
|
|
all_branches=self.args.all_branches,
|
|
all_tags=self.args.all_tags,
|
|
all_commits=self.args.all_commits,
|
|
with_deps=self.args.with_deps,
|
|
recursive=self.args.recursive,
|
|
)
|
|
|
|
if self.args.quiet:
|
|
return bool(st)
|
|
|
|
if self.args.show_json:
|
|
import json
|
|
|
|
logger.info(json.dumps(st))
|
|
elif st:
|
|
self._show(st, indent)
|
|
elif not self.repo.stages:
|
|
logger.info(self.EMPTY_PROJECT_MSG)
|
|
elif self.args.cloud or self.args.remote:
|
|
remote = self.args.remote or self.repo.config["core"].get(
|
|
"remote"
|
|
)
|
|
logger.info(self.IN_SYNC_MSG.format(remote=remote))
|
|
else:
|
|
logger.info(self.UP_TO_DATE_MSG)
|
|
|
|
except DvcException:
|
|
logger.exception("")
|
|
return 1
|
|
return 0
|