Create hook-asyncssh.py

Fixes #7577

.codecov.yml

@@ -1,10 +0,0 @@
-comment: false
-coverage:
-  status:
-    project:
-      default:
-        target: 90
-        threshold: 10%
-    patch:
-      default:
-        threshold: 25%

.coveragerc

@@ -1,16 +0,0 @@
-[run]
-branch = True
-source = dvc
-omit =
-    *__main__.py
-
-[report]
-exclude_lines =
-    if __name__ == .__main__.:
-    if typing.TYPE_CHECKING:
-    if TYPE_CHECKING:
-    pass
-    raise NotImplementedError
-    # pragma: no cover
-    #  pragma: no cover
-    \.\.\.$

.dvcignore

@@ -1,4 +1,3 @@
 /tests
 /dvc
 /.github
-/bin

.flake8

@@ -0,0 +1,12 @@
+[flake8]
+ignore=
+    E203, # Whitespace before ':'
+    E266, # Too many leading '#' for block comment
+    W503, # Line break occurred before a binary operator
+    P1,  # unindexed parameters in the str.format, see:
+    # https://pypi.org/project/flake8-string-format/
+max_line_length = 79
+max-complexity = 15
+select = B,C,E,F,W,T4,B902,T,P
+show_source = true
+count = true

.git_archival.txt

@@ -1 +1,4 @@
+node: $Format:%H$
+node-date: $Format:%cI$
+describe-name: $Format:%(describe:tags=true)$
 ref-names: $Format:%D$

.github/CODEOWNERS

@@ -1 +0,0 @@
-*       @iterative/dvc

.github/FUNDING.yml

@@ -1,3 +0,0 @@
-# These are supported funding model platforms
-
-patreon: DVCorg

.github/ISSUE_TEMPLATE/bug_report.md

@@ -8,7 +8,7 @@ about: Create a bug report to help us improve DVC
 <!--
 ## Issue name
 
-Issue names must follow the pattern `command: description` where the command is the dvc command that you are trying to run. The description should describe the consequence of the bug. 
+Issue names must follow the pattern `command: description` where the command is the dvc command that you are trying to run. The description should describe the consequence of the bug.
 
 Example: `repro: doesn't detect input changes`
 -->

.github/ISSUE_TEMPLATE/feature_request.md

@@ -2,4 +2,3 @@
 name: "\U0001F680 Feature Request"
 about: Suggest an idea for this project
 ---
-

.github/codecov.yml

@@ -0,0 +1,12 @@
+codecov:
+  notify:
+    wait_for_ci: true
+coverage:
+  status:
+    project:
+      default:
+        target: auto
+        threshold: 2%
+    patch: off
+github_checks:
+  annotations: false

.github/mergify.yml

@@ -1,9 +0,0 @@
-pull_request_rules:
-  - name: backport patches to 1.11
-    conditions:
-      - base=main
-      - label=backport-1.11
-    actions:
-      backport:
-        branches:
-          - "1.11"

.github/release-drafter.yml

@@ -1,49 +0,0 @@
-# Config for https://github.com/apps/release-drafter
-name-template: "$RESOLVED_VERSION 🦉 "
-tag-template: "$RESOLVED_VERSION"
-categories:
-  - title: "🚀 New Features and Enhancements "
-    labels:
-      - "feature"
-      - "enhancement"
-      - "ui"
-  - title: "💣 Breaking Changes "
-    labels:
-      - "breaking-change"
-  - title: "🏇 Optimizations "
-    labels:
-      - "optimize"
-      - "performance"
-  - title: "🐛 Bug Fixes "
-    labels:
-      - "bugfix"
-      - "bug"
-  - title: "🔨 Maintenance "
-    labels:
-      - "maintenance"
-      - "refactoring"
-      - "chore"
-      - "build"
-      - "ci"
-change-template: "- $TITLE (#$NUMBER) @$AUTHOR"
-version-resolver:
-  major:
-    labels:
-      - "major"
-  minor:
-    labels:
-      - "minor"
-  patch:
-    labels:
-      - "patch"
-  default: patch
-exclude-labels:
-  - "skip-changelog"
-template: |
-  > Refer to https://dvc.org/doc/install for installation instructions.
-
-  ## Changes
-
-  $CHANGES
-
-  Thanks again to $CONTRIBUTORS for the contributions! 🎉

.github/release.yml

@@ -0,0 +1,16 @@
+changelog:
+  exclude:
+    labels: ["skip-changelog"]
+  categories:
+    - title: 🛠 Breaking Changes
+      labels: ["breaking-change"]
+    - title: 🚀 New Features and Enhancements
+      labels: ["feature", "enhancement", "ui"]
+    - title: ⚡ Optimizations
+      labels: ["optimize", "performance"]
+    - title: 🐛 Bug Fixes
+      labels: ["bugfix", "bug"]
+    - title: 🔨 Maintenance
+      labels: ["maintenance", "refactoring", "chore", "build", "ci"]
+    - title: Other Changes
+      labels: ["*"]

.github/workflows/benchmarks.yaml

@@ -1,23 +1,18 @@
 name: benchmarks
 on: [pull_request, workflow_dispatch]
 
+concurrency:
+  group: ${{ github.workflow }}-${{ github.head_ref || github.run_id }}
+  cancel-in-progress: true
+
 jobs:
   bench:
     name: run benchmarks
-    runs-on: ${{ matrix.os }}
-    strategy:
-      fail-fast: false
-      matrix:
-        os: [ubuntu-18.04]
+    runs-on: ubuntu-latest
     steps:
-      - name: Cancel Previous Runs
-        uses: styfle/[email protected]
+      - uses: actions/setup-python@v4
         with:
-          access_token: ${{ github.token }}
-      - uses: actions/setup-python@v3
-        with:
-            python-version: 3.7
+            python-version: "3.11"
       - uses: iterative/dvc-bench@main
         with:
             pytest_options: "-k 'test_init or test_help'"
-

.github/workflows/codeql.yml

@@ -0,0 +1,39 @@
+name: "CodeQL"
+
+on:
+  push:
+    branches: ["main"]
+  pull_request:
+    branches: ["main"]
+  schedule:
+    - cron: '32 19 * * 2'
+  workflow_dispatch:
+
+permissions:
+  contents: read
+
+jobs:
+  analyze:
+    name: Analyze
+    runs-on: ubuntu-latest
+    permissions:
+      actions: read
+      contents: read
+      security-events: write
+
+    strategy:
+      fail-fast: false
+      matrix:
+        language: ['python']
+
+    steps:
+    - name: Checkout repository
+      uses: actions/checkout@v3
+
+    - name: Initialize CodeQL
+      uses: github/codeql-action/init@v2
+      with:
+        languages: ${{ matrix.language }}
+
+    - name: Perform CodeQL Analysis
+      uses: github/codeql-action/analyze@v2

.github/workflows/packages.yaml

@@ -6,152 +6,107 @@ on:
   release:
     types: [released, prereleased]
   workflow_dispatch:
+
+permissions:  # added using https://github.com/step-security/secure-workflows
+  contents: read
+
 jobs:
-  deb:
-    runs-on: ubuntu-18.04
+  binary:
+    permissions:
+      contents: write  # for actions/upload-release-asset to upload release asset
+    strategy:
+      matrix:
+        include:
+         - {pkg: "deb",  os: "ubuntu-18.04", asset: "dvc_${{ github.event.release.tag_name }}_amd64.deb"}
+         - {pkg: "rpm", os: "ubuntu-18.04", asset: "dvc-${{ github.event.release.tag_name }}-1.x86_64.rpm"}
+         - {pkg: "osxpkg", os: "macos-10.15", asset: "dvc-${{ github.event.release.tag_name }}.pkg"}
+         - {pkg: "exe",  os: "windows-2019", asset: "dvc-${{ github.event.release.tag_name }}.exe"}
+
+    name: ${{ matrix.pkg }}
+    runs-on: ${{ matrix.os }}
     steps:
     - uses: actions/checkout@v3
       with:
           fetch-depth: 0
-    - name: Set up Python 3.9
-      uses: actions/setup-python@v3
+
+    - name: Set up Python 3.10
+      uses: actions/setup-python@v4
       with:
-        python-version: 3.9
-    - name: Set up Ruby 2.4
-      uses: actions/[email protected]
+        python-version: "3.10"
+        cache: pip
+        cache-dependency-path: |
+          pyproject.toml
+          scripts/build-requirements.txt
+
+    - name: Set up Ruby 2.6
+      uses: ruby/setup-ruby@v1
+      if: matrix.pkg != 'exe'
       with:
-        ruby-version: 2.4
-    - name: Install deps
-      run: |
-        pip install .[all]
-        pip install -r scripts/build-requirements.txt
-        gem install --no-document fpm
-    - name: Build deb
-      run: python scripts/build.py deb
-    - name: Publish deb
-      if: github.event_name == 'release'
-      uses: actions/[email protected]
-      env:
-        GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
-      with:
-        upload_url: ${{ github.event.release.upload_url }}
-        asset_path: scripts/fpm/dvc_${{ github.event.release.tag_name }}_amd64.deb
-        asset_name: dvc_${{ github.event.release.tag_name }}_amd64.deb
-        asset_content_type: binary/octet-stream
-  rpm:
-    runs-on: ubuntu-18.04
-    steps:
-    - uses: actions/checkout@v3
-      with:
-        fetch-depth: 0
-    - name: Set up Python 3.9
-      uses: actions/setup-python@v3
-      with:
-        python-version: 3.9
-    - name: Set up Ruby 2.4
-      uses: actions/[email protected]
-      with:
-        ruby-version: 2.4
-    - name: Install deps
-      run: |
-        pip install .[all]
-        pip install -r scripts/build-requirements.txt
-        gem install --no-document fpm
-    - name: Build rpm
-      run: python scripts/build.py rpm
-    - name: Publish rpm
-      if: github.event_name == 'release'
-      uses: actions/[email protected]
-      env:
-        GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
-      with:
-        upload_url: ${{ github.event.release.upload_url }}
-        asset_path: scripts/fpm/dvc-${{ github.event.release.tag_name }}-1.x86_64.rpm
-        asset_name: dvc-${{ github.event.release.tag_name }}-1.x86_64.rpm
-        asset_content_type: binary/octet-stream
-  osxpkg:
-    runs-on: macos-10.15
-    steps:
-    - uses: actions/checkout@v3
-      with:
-        fetch-depth: 0
-    - name: Set up Python 3.9
-      uses: actions/setup-python@v3
-      with:
-        python-version: 3.9
-    - name: Set up Ruby 2.4
-      uses: actions/[email protected]
-      with:
-        ruby-version: 2.4
-    - name: Install deps
-      run: |
-        # available lxml (webdav dependency) wheels are built with an old
-        # SDK, which breaks osxpkg notarization. Building from scratch
-        # instead.
-        pip install .[all] --no-binary lxml
-        pip install -r scripts/build-requirements.txt
-        gem install --no-document fpm
-    - name: Build
-      run: python scripts/build.py osxpkg
-    - name: Publish
-      if: github.event_name == 'release'
-      uses: actions/[email protected]
-      env:
-        GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
-      with:
-        upload_url: ${{ github.event.release.upload_url }}
-        asset_path: scripts/fpm/dvc-${{ github.event.release.tag_name }}.pkg
-        asset_name: dvc-${{ github.event.release.tag_name }}.pkg
-        asset_content_type: binary/octet-stream
-  exe:
-    runs-on: windows-2019
-    steps:
-    - uses: actions/checkout@v3
-      with:
-        fetch-depth: 0
-    - name: Set up Python 3.9
-      uses: actions/setup-python@v3
-      with:
-        python-version: 3.9
+        ruby-version: '2.6'
+
+    - name: Install fpm
+      if: matrix.pkg != 'exe'
+      run: gem install --no-document fpm
+
     - name: Install deps
       run: |
+        pip install --upgrade pip wheel setuptools
         pip install .[all]
         pip install -r scripts/build-requirements.txt
+
     - name: Pull images
-      run: |
-        dvc pull
-    - name: Build
-      run: python scripts\build.py exe
-      shell: cmd
-    - name: Publish
+      run: dvc pull
+
+    - name: Build ${{ matrix.pkg }}
+      run: python scripts/build.py ${{ matrix.pkg }}
+
+    - name: Publish ${{ matrix.pkg }}
       if: github.event_name == 'release'
+      uses: actions/[email protected]
       env:
         GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
-      uses: actions/[email protected]
       with:
         upload_url: ${{ github.event.release.upload_url }}
-        asset_path: scripts/innosetup/dvc-${{ github.event.release.tag_name }}.exe
-        asset_name: dvc-${{ github.event.release.tag_name }}.exe
+        asset_path: scripts/${{ matrix.pkg == 'exe' && 'innosetup' || 'fpm' }}/${{ matrix.asset }}
+        asset_name: ${{ matrix.asset }}
         asset_content_type: binary/octet-stream
+
   pip:
-    runs-on: ubuntu-18.04
+    runs-on: ubuntu-20.04
     steps:
     - uses: actions/checkout@v3
       with:
         fetch-depth: 0
-    - name: Set up Python 3.7
-      uses: actions/setup-python@v3
+
+    - name: Set up Python 3.8
+      uses: actions/setup-python@v4
       with:
-        python-version: 3.7
+        python-version: 3.8
+
     - name: Install
-      run: pip install ".[all,tests]"
-    - name: Build packages
+      run: pip install --upgrade pip wheel
+
+    - name: Force version for Test PyPI uploads
+      if: ${{ !startsWith(github.ref, 'refs/tags') }}
       run: |
-        pip install -U pip
-        ./scripts/build_package.sh
+        pip install setuptools_scm
+        echo version=$(python -m setuptools_scm | awk -F+ '{print $1}' | tail -1) >> $GITHUB_ENV
+
+    - name: Build packages
+      run: ./scripts/build_package.sh
+      env:
+        SETUPTOOLS_SCM_PRETEND_VERSION: ${{ env.version }}
+
     - name: Publish packages to PyPI
       if: github.event_name == 'release'
-      uses: pypa/gh-action-pypi-publish@master
+      uses: pypa/gh-action-pypi-publish@release/v1
       with:
-        user: __token__
         password: ${{ secrets.PYPI_TOKEN }}
+
+    - name: Publish to Test PyPI
+      if: ${{ github.event_name == 'release' || (github.event_name == 'push' && github.ref == 'refs/heads/main') }}
+      uses: pypa/gh-action-pypi-publish@release/v1
+      with:
+        password: ${{ secrets.TEST_PYPI_TOKEN }}
+        repository_url: https://test.pypi.org/legacy/
+        skip_existing: true

.github/workflows/tests.yaml

@@ -1,89 +1,94 @@
 name: Tests
 
 on:
-  pull_request: {}
-  push: {}
+  push:
+    branches: [main]
+  pull_request:
   schedule:
     - cron: '5 1 * * *'  # every day at 01:05
+  workflow_dispatch:
 
 env:
   DVC_TEST: "true"
   HOMEBREW_NO_AUTO_UPDATE: 1
   SHELL: /bin/bash
 
+concurrency:
+  group: ${{ github.workflow }}-${{ github.head_ref || github.run_id }}
+  cancel-in-progress: true
+
+permissions:  # added using https://github.com/step-security/secure-workflows
+  contents: read
+
 jobs:
   lint:
     timeout-minutes: 10
     runs-on: ubuntu-latest
     steps:
-    - name: Cancel Previous Runs
-      uses: styfle/[email protected]
-      with:
-        access_token: ${{ github.token }}
     - uses: actions/checkout@v3
       with:
         fetch-depth: 0
     - name: Set up Python 3.8
-      uses: actions/setup-python@v3
+      uses: actions/setup-python@v4
       with:
         python-version: 3.8
+        cache: pip
+        cache-dependency-path: pyproject.toml
     - name: Install requirements
       run: |
-        pip install --upgrade pip setuptools wheel
-        pip install collective.checkdocs==0.2
-        pip install ".[dev]" pre-commit
-    - name: Check README
-      run: python setup.py checkdocs
-    - uses: pre-commit/[email protected]
+        pip install --upgrade pip wheel
+        pip install ".[dev]"
+    - uses: pre-commit/[email protected]
   tests:
     timeout-minutes: 45
     runs-on: ${{ matrix.os }}
     strategy:
       fail-fast: false
       matrix:
-        os: [ubuntu-20.04, windows-latest, macos-latest]
-        pyv: ["3.7", "3.8", "3.9", "3.10"]
+        os: [ubuntu-20.04, windows-latest]
+        pyv: ["3.8", "3.9", "3.10", "3.11"]
+        pytest-filter:
+        - "import or plot or live or experiment"
+        - "not (import or plot or live or experiment)"
+        include:
+        - {os: macos-latest, pyv: "3.8"}
+        - {os: macos-latest, pyv: "3.9"}
+        - {os: macos-latest, pyv: "3.10"}
+        - {os: macos-latest, pyv: "3.11"}
+
     steps:
     - uses: actions/checkout@v3
       with:
         fetch-depth: 0
     - name: Set up Python
-      uses: actions/setup-python@v3
+      uses: actions/setup-python@v4
       with:
         python-version: ${{ matrix.pyv }}
-    - name: get pip cache dir
-      id: pip-cache-dir
-      run: |
-        echo "::set-output name=dir::$(pip cache dir)"
-    - name: set pip cache
-      id: pip-cache
-      uses: actions/cache@v3
-      with:
-        path: ${{ steps.pip-cache-dir.outputs.dir }}
-        key: ${{ runner.os }}-pip-${{ hashFiles('setup.cfg') }}
-        restore-keys: |
-          ${{ runner.os }}-pip-
+        cache: pip
+        cache-dependency-path: pyproject.toml
     - name: install
-      if: steps.cache.pip-cache-dir.cache-hit != 'true'
       run: |
-        pip install --upgrade pip setuptools wheel
+        pip install --upgrade pip wheel
         pip install -e ".[dev]"
     - name: run tests
-      timeout-minutes: 40
-      run: >-
-        python -m tests -n=auto
-        --cov-report=xml --cov-report=term
-        ${{ env.extra_test_args }}
+      timeout-minutes: 20
+      run: pytest -nauto --durations 100 --cov --cov-report=xml --cov-report=term -k "${{ matrix.pytest-filter }}"
     - name: upload coverage report
       uses: codecov/codecov-action@v3
       with:
         file: ./coverage.xml
         fail_ci_if_error: false
+  check:
+    if: always()
+    needs: [lint, tests]
+    runs-on: ubuntu-latest
+    steps:
+      - uses: re-actors/alls-green@release/v1
+        with:
+          jobs: ${{ toJSON(needs) }}
   notify:
     if: github.ref == 'refs/heads/main' && failure()
-    needs:
-      - lint
-      - tests
+    needs: [lint, tests]
     runs-on: ubuntu-latest
     steps:
     - name: Slack Notification

.pre-commit-config.yaml

@@ -1,39 +1,62 @@
+default_language_version:
+  python: python3
+
 ci:
   skip: [mypy, pylint, dvc-pre-commit]
 
 repos:
-  - hooks:
-      - id: black
-        language_version: python3
-    repo: https://github.com/ambv/black
-    rev: 22.3.0
   - repo: https://github.com/pre-commit/pre-commit-hooks
-    rev: v4.2.0
+    rev: v4.3.0
     hooks:
+      - id: check-added-large-files
+      - id: check-case-conflict
+      - id: check-executables-have-shebangs
+      - id: check-json
+      - id: check-merge-conflict
+        exclude: "tests/func/utils/test_strict_yaml.py"
+        args: ['--assume-in-merge']
       - id: check-toml
       - id: check-yaml
-  - hooks:
-      - id: codespell
-        args: 
-          - --ignore-words-list
-          - ba,datas,fo,uptodate,cachable
-    repo: https://github.com/codespell-project/codespell
-    rev: v2.1.0
-  - hooks:
-      - id: isort
-        language_version: python3
-    repo: https://github.com/timothycrosley/isort
+      - id: debug-statements
+        exclude: "dvc/_debug.py"
+      - id: end-of-file-fixer
+      - id: mixed-line-ending
+      - id: sort-simple-yaml
+      - id: trailing-whitespace
+  - repo: https://github.com/lovesegfault/beautysh
+    rev: v6.2.1
+    hooks:
+      - id: beautysh
+        args: ["-i", "2"]
+  - repo: https://github.com/asottile/pyupgrade
+    rev: v3.2.0
+    hooks:
+      - id: pyupgrade
+        args:
+          - --py38-plus
+  - repo: https://github.com/psf/black
+    rev: 22.10.0
+    hooks:
+      - id: black
+  - repo: https://github.com/PyCQA/isort
     rev: 5.10.1
-  - hooks:
+    hooks:
+      - id: isort
+  - repo: https://github.com/codespell-project/codespell
+    rev: v2.2.2
+    hooks:
+      - id: codespell
+        additional_dependencies: ["tomli"]
+  - repo: https://github.com/PyCQA/flake8
+    rev: 5.0.4
+    hooks:
       - id: flake8
-        language_version: python3
         additional_dependencies:
+          - flake8-broken-line
           - flake8-bugbear
           - flake8-comprehensions
           - flake8-debugger
           - flake8-string-format
-    repo: https://gitlab.com/pycqa/flake8
-    rev: 3.9.2
   - repo: local
     hooks:
     - id: mypy
@@ -47,14 +70,6 @@ repos:
       entry: pylint
       language: system
       types: [python]
-  - hooks:
-      - args:
-          - -i
-          - "2"
-        id: beautysh
-        language_version: python3
-    repo: https://github.com/lovesegfault/beautysh
-    rev: v6.2.1
   - hooks:
       - args:
           - git-hook

CODE_OF_CONDUCT.md

@@ -74,4 +74,3 @@ available at https://www.contributor-covenant.org/version/1/4/code-of-conduct.ht
 
 For answers to common questions about this code of conduct, see
 https://www.contributor-covenant.org/faq
-

MANIFEST.in

@@ -1,3 +0,0 @@
-include LICENSE
-recursive-include requirements *.txt
-recursive-include dvc/repo/plots/templates *.json

README.rst

@@ -3,97 +3,114 @@
 `Website <https://dvc.org>`_
 • `Docs <https://dvc.org/doc>`_
 • `Blog <http://blog.dataversioncontrol.com>`_
-• `Twitter <https://twitter.com/DVCorg>`_
-• `Chat (Community & Support) <https://dvc.org/chat>`_
 • `Tutorial <https://dvc.org/doc/get-started>`_
-• `Mailing List <https://sweedom.us10.list-manage.com/subscribe/post?u=a08bf93caae4063c4e6a351f6&id=24c0ecc49a>`_
+• `Related Technologies <https://dvc.org/doc/user-guide/related-technologies>`_
+• `How DVC works`_
+• `VS Code Extension`_
+• `Installation`_
+• `Contributing`_
+• `Community and Support`_
 
-|CI| |Maintainability| |Coverage| |Donate| |DOI|
+|CI| |Python Version| |Coverage| |VS Code| |DOI|
 
-|PyPI| |Packages| |Brew| |Conda| |Choco| |Snap|
+|PyPI| |PyPI Downloads| |Packages| |Brew| |Conda| |Choco| |Snap|
 
 |
 
-**Data Version Control** or **DVC** is an **open-source** tool for data science and machine
-learning projects. Key features:
+**Data Version Control** or **DVC** is a command line tool and `VS Code Extension`_ to help you develop reproducible machine learning projects:
 
-#. Simple **command line** Git-like experience. Does not require installing and maintaining
-   any databases. Does not depend on any proprietary online services.
+#. **Version** your data and models.
+   Store them in your cloud storage but keep their version info in your Git repo.
 
-#. Management and versioning of **datasets** and **machine learning
-   models**. Data can be saved in S3, Google cloud, Azure, Alibaba 
-   cloud, SSH server, HDFS, or even local HDD RAID.
+#. **Iterate** fast with lightweight pipelines.
+   When you make changes, only run the steps impacted by those changes.
 
-#. Makes projects **reproducible** and **shareable**; helping to answer questions about how
-   a model was built.
+#. **Track** experiments in your local Git repo (no servers needed).
 
-#. Helps manage experiments with Git tags/branches and **metrics** tracking.
+#. **Compare** any data, code, parameters, model, or performance plots.
 
-**DVC** aims to replace spreadsheet and document sharing tools (such as Excel or Google Docs)
-frequently used as both knowledge repositories and team ledgers.
-DVC also replaces both ad-hoc scripts to track, move, and deploy different model versions
-and ad-hoc data file suffixes and prefixes.
-
-.. contents:: **Contents**
-  :backlinks: none
-
-How DVC works
-=============
-
-We encourage you to read our `Get Started <https://dvc.org/doc/get-started>`_ guide to better understand what DVC
-is and how it can fit your scenarios.
-
-The easiest (but not perfect!) *analogy* to describe it: DVC is Git (or Git-LFS to be precise) & Makefiles
-made right and tailored specifically for ML and Data Science scenarios.
-
-#. ``Git/Git-LFS`` part - DVC helps store and share data artifacts and models, connecting them with a Git repository.
-#. ``Makefile``\ s part - DVC describes how one data or model artifact was built from other data and code.
-
-DVC usually runs along with Git. Git is used as usual to store and version code (including DVC meta-files). DVC helps
-to store data and model files seamlessly out of Git, while preserving almost the same user experience as if they
-were stored in Git itself. To store and share the data cache, DVC supports multiple remotes - any cloud (S3, Azure,
-Google Cloud, etc) or any on-premise network storage (via SSH, for example).
-
-|Flowchart|
-
-The DVC pipelines (computational graph) feature connects code and data together. It is possible to explicitly
-specify all steps required to produce a model: input dependencies including data, commands to run,
-and output information to be saved. See the quick start section below or
-the `Get Started <https://dvc.org/doc/get-started>`_ tutorial to learn more.
+#. **Share** experiments and automatically reproduce anyone's experiment.
 
 Quick start
 ===========
 
-Please read `Get Started <https://dvc.org/doc/get-started>`_ guide for a full version. Common workflow commands include:
+    Please read our `Command Reference <https://dvc.org/doc/command-reference>`_ for a complete list.
 
-+-----------------------------------+----------------------------------------------------------------------------+
-| Step                              | Command                                                                    |
-+===================================+============================================================================+
-| Track data                        | | ``$ git add train.py``                                                   |
-|                                   | | ``$ dvc add images.zip``                                                 |
-+-----------------------------------+----------------------------------------------------------------------------+
-| Connect code and data by commands | | ``$ dvc run -n prepare -d images.zip -o images/ unzip -q images.zip``    |
-|                                   | | ``$ dvc run -n train -d images/ -d train.py -o model.p python train.py`` |
-+-----------------------------------+----------------------------------------------------------------------------+
-| Make changes and reproduce        | | ``$ vi train.py``                                                        |
-|                                   | | ``$ dvc repro model.p.dvc``                                              |
-+-----------------------------------+----------------------------------------------------------------------------+
-| Share code                        | | ``$ git add .``                                                          |
-|                                   | | ``$ git commit -m 'The baseline model'``                                 |
-|                                   | | ``$ git push``                                                           |
-+-----------------------------------+----------------------------------------------------------------------------+
-| Share data and ML models          | | ``$ dvc remote add myremote -d s3://mybucket/image_cnn``                 |
-|                                   | | ``$ dvc push``                                                           |
-+-----------------------------------+----------------------------------------------------------------------------+
+A common CLI workflow includes:
+
+
++-----------------------------------+----------------------------------------------------------------------------------------------------+
+| Task                              | Terminal                                                                                           |
++===================================+====================================================================================================+
+| Track data                        | | ``$ git add train.py params.yaml``                                                               |
+|                                   | | ``$ dvc add images/``                                                                            |
++-----------------------------------+----------------------------------------------------------------------------------------------------+
+| Connect code and data             | | ``$ dvc stage add -n featurize -d images/ -o features/ python featurize.py``                     |
+|                                   | | ``$ dvc stage add -n train -d features/ -d train.py -o model.p -M metrics.json python train.py`` |
++-----------------------------------+----------------------------------------------------------------------------------------------------+
+| Make changes and experiment       | | ``$ dvc exp run -n exp-baseline``                                                                |
+|                                   | | ``$ vi train.py``                                                                                |
+|                                   | | ``$ dvc exp run -n exp-code-change``                                                             |
++-----------------------------------+----------------------------------------------------------------------------------------------------+
+| Compare and select experiments    | | ``$ dvc exp show``                                                                               |
+|                                   | | ``$ dvc exp apply exp-baseline``                                                                 |
++-----------------------------------+----------------------------------------------------------------------------------------------------+
+| Share code                        | | ``$ git add .``                                                                                  |
+|                                   | | ``$ git commit -m 'The baseline model'``                                                         |
+|                                   | | ``$ git push``                                                                                   |
++-----------------------------------+----------------------------------------------------------------------------------------------------+
+| Share data and ML models          | | ``$ dvc remote add myremote -d s3://mybucket/image_cnn``                                         |
+|                                   | | ``$ dvc push``                                                                                   |
++-----------------------------------+----------------------------------------------------------------------------------------------------+
+
+How DVC works
+=============
+
+    We encourage you to read our `Get Started
+    <https://dvc.org/doc/get-started>`_ docs to better understand what DVC
+    does and how it can fit your scenarios.
+
+The closest *analogies* to describe the main DVC features are these:
+
+#. **Git for data**: Store and share data artifacts (like Git-LFS but without a server) and models, connecting them with a Git repository. Data management meets GitOps!
+#. **Makefiles** for ML: Describes how data or model artifacts are built from other data and code in a standard format. Now you can version your data pipelines with Git.
+#. Local **experiment tracking**: Turn your machine into an ML experiment management platform, and collaborate with others using existing Git hosting (Github, Gitlab, etc.).
+
+Git is employed as usual to store and version code (including DVC meta-files as placeholders for data).
+DVC `stores data and model files <https://dvc.org/doc/start/data-management>`_ seamlessly in a cache outside of Git, while preserving almost the same user experience as if they were in the repo.
+To share and back up the *data cache*, DVC supports multiple remote storage platforms - any cloud (S3, Azure, Google Cloud, etc.) or on-premise network storage (via SSH, for example).
+
+|Flowchart|
+
+`DVC pipelines <https://dvc.org/doc/start/data-management/data-pipelines>`_ (computational graphs) connect code and data together.
+They specify all steps required to produce a model: input dependencies including code, data, commands to run; and output information to be saved.
+
+Last but not least, `DVC Experiment Versioning <https://dvc.org/doc/start/experiments>`_ lets you prepare and run a large number of experiments.
+Their results can be filtered and compared based on hyperparameters and metrics, and visualized with multiple plots.
+
+.. _`VS Code Extension`:
+
+Visual Studio Code Extension
+============================
+
+|VS Code|
+
+To use DVC as a GUI right from your VS Code IDE, install the `DVC Extension <https://marketplace.visualstudio.com/items?itemName=Iterative.dvc>`_ from the Marketplace.
+It currently features experiment tracking and data management, and more features (data pipeline support, etc.) are coming soon!
+
+|VS Code Extension Overview|
+
+    Note: You'll have to install core DVC on your system separately (as detailed
+    below). The Extension will guide you if needed.
 
 Installation
 ============
 
-There are four options to install DVC: ``pip``, Homebrew, Conda (Anaconda) or an OS-specific package.
+There are several ways to install DVC: in VS Code; using ``snap``, ``choco``, ``brew``, ``conda``, ``pip``; or with an OS-specific package.
 Full instructions are `available here <https://dvc.org/doc/get-started/install>`_.
 
-Snap (Snapcraft/Linux)
-----------------------
+Snapcraft (Linux)
+-----------------
 
 |Snap|
 
@@ -102,11 +119,10 @@ Snap (Snapcraft/Linux)
    snap install dvc --classic
 
 This corresponds to the latest tagged release.
-Add ``--beta`` for the latest tagged release candidate,
-or ``--edge`` for the latest ``main`` version.
+Add ``--beta`` for the latest tagged release candidate, or ``--edge`` for the latest ``main`` version.
 
-Choco (Chocolatey/Windows)
---------------------------
+Chocolatey (Windows)
+--------------------
 
 |Choco|
 
@@ -114,8 +130,8 @@ Choco (Chocolatey/Windows)
 
    choco install dvc
 
-Brew (Homebrew/Mac OS)
-----------------------
+Brew (mac OS)
+-------------
 
 |Brew|
 
@@ -123,8 +139,8 @@ Brew (Homebrew/Mac OS)
 
    brew install dvc
 
-Conda (Anaconda)
-----------------
+Anaconda (Any platform)
+-----------------------
 
 |Conda|
 
@@ -133,11 +149,10 @@ Conda (Anaconda)
    conda install -c conda-forge mamba # installs much faster than conda
    mamba install -c conda-forge dvc
 
-Depending on the remote storage type you plan to use to keep and share your data, you might need to
-install optional dependencies: `dvc-s3`, `dvc-azure`, `dvc-gdrive`, `dvc-gs`, `dvc-oss`, `dvc-ssh`.
+Depending on the remote storage type you plan to use to keep and share your data, you might need to install optional dependencies: `dvc-s3`, `dvc-azure`, `dvc-gdrive`, `dvc-gs`, `dvc-oss`, `dvc-ssh`.
 
-pip (PyPI)
-----------
+PyPI (Python)
+-------------
 
 |PyPI|
 
@@ -145,10 +160,8 @@ pip (PyPI)
 
    pip install dvc
 
-Depending on the remote storage type you plan to use to keep and share your data, you might need to specify
-one of the optional dependencies: ``s3``, ``gs``, ``azure``, ``oss``, ``ssh``. Or ``all`` to include them all.
-The command should look like this: ``pip install dvc[s3]`` (in this case AWS S3 dependencies such as ``boto3``
-will be installed automatically).
+Depending on the remote storage type you plan to use to keep and share your data, you might need to specify one of the optional dependencies: ``s3``, ``gs``, ``azure``, ``oss``, ``ssh``. Or ``all`` to include them all.
+The command should look like this: ``pip install 'dvc[s3]'`` (in this case AWS S3 dependencies such as ``boto3`` will be installed automatically).
 
 To install the development version, run:
 
@@ -156,88 +169,58 @@ To install the development version, run:
 
    pip install git+git://github.com/iterative/dvc
 
-Package
--------
+Package (Platform-specific)
+---------------------------
 
 |Packages|
 
-Self-contained packages for Linux, Windows, and Mac are available. The latest version of the packages
-can be found on the GitHub `releases page <https://github.com/iterative/dvc/releases>`_.
+Self-contained packages for Linux, Windows, and Mac are available.
+The latest version of the packages can be found on the GitHub `releases page <https://github.com/iterative/dvc/releases>`_.
 
 Ubuntu / Debian (deb)
 ^^^^^^^^^^^^^^^^^^^^^
 .. code-block:: bash
 
    sudo wget https://dvc.org/deb/dvc.list -O /etc/apt/sources.list.d/dvc.list
-   sudo apt-get update
-   sudo apt-get install dvc
+   wget -qO - https://dvc.org/deb/iterative.asc | sudo apt-key add -
+   sudo apt update
+   sudo apt install dvc
 
 Fedora / CentOS (rpm)
 ^^^^^^^^^^^^^^^^^^^^^
 .. code-block:: bash
 
    sudo wget https://dvc.org/rpm/dvc.repo -O /etc/yum.repos.d/dvc.repo
+   sudo rpm --import https://dvc.org/rpm/iterative.asc
    sudo yum update
    sudo yum install dvc
 
-Comparison to related technologies
-==================================
-
-#. Data Engineering tools such as `AirFlow <https://airflow.apache.org/>`_,
-   `Luigi <https://github.com/spotify/luigi>`_, and others - in DVC data,
-   model and ML pipelines represent a single ML project focused on data
-   scientists' experience.  Data engineering tools orchestrate multiple data
-   projects and focus on efficient execution. A DVC project can be used from
-   existing data pipelines as a single execution step.
-
-#. `Git-annex <https://git-annex.branchable.com/>`_ - DVC uses the idea of storing the content of large files (which should
-   not be in a Git repository) in a local key-value store, and uses file hardlinks/symlinks instead of
-   copying/duplicating files.
-
-#. `Git-LFS <https://git-lfs.github.com/>`_ - DVC is compatible with many
-   remote storage services (S3, Google Cloud, Azure, SSH, etc). DVC also
-   uses reflinks or hardlinks to avoid copy operations on checkouts; thus
-   handling large data files much more efficiently.
-
-#. Makefile (and analogues including ad-hoc scripts) - DVC tracks
-   dependencies (in a directed acyclic graph).
-
-#. `Workflow Management Systems <https://en.wikipedia.org/wiki/Workflow_management_system>`_ - DVC is a workflow
-   management system designed specifically to manage machine learning experiments. DVC is built on top of Git.
-
-#. `DAGsHub <https://dagshub.com/>`_ - online service to host DVC
-   projects.  It provides a useful UI around DVC repositories and integrates
-   other tools.
-
-#. `DVC Studio <https://studio.iterative.ai/>`_ - official online
-   platform for DVC projects.  It can be used to manage data and models, run
-   and track experiments, and visualize and share results.  Also, it
-   integrates with `CML (CI/CD for ML) <https://cml.dev/>`__ for training
-   models in the cloud or Kubernetes.
-
-
 Contributing
 ============
 
-|Maintainability| |Donate|
+|Maintainability|
 
-Contributions are welcome! Please see our `Contributing Guide <https://dvc.org/doc/user-guide/contributing/core>`_ for more
-details. Thanks to all our contributors!
+Contributions are welcome!
+Please see our `Contributing Guide <https://dvc.org/doc/user-guide/contributing/core>`_ for more details.
+Thanks to all our contributors!
 
 |Contribs|
 
-Mailing List
-============
+Community and Support
+=====================
 
-Want to stay up to date? Want to help improve DVC by participating in our occasional polls? Subscribe to our `mailing list <https://sweedom.us10.list-manage.com/subscribe/post?u=a08bf93caae4063c4e6a351f6&id=24c0ecc49a>`_. No spam, really low traffic.
+* `Twitter <https://twitter.com/DVCorg>`_
+* `Forum <https://discuss.dvc.org/>`_
+* `Discord Chat <https://dvc.org/chat>`_
+* `Email <mailto:[email protected]>`_
+* `Mailing List <https://sweedom.us10.list-manage.com/subscribe/post?u=a08bf93caae4063c4e6a351f6&id=24c0ecc49a>`_
 
 Copyright
 =========
 
 This project is distributed under the Apache license version 2.0 (see the LICENSE file in the project root).
 
-By submitting a pull request to this project, you agree to license your contribution under the Apache license version
-2.0 to this project.
+By submitting a pull request to this project, you agree to license your contribution under the Apache license version 2.0 to this project.
 
 Citation
 ========
@@ -254,6 +237,9 @@ Barrak, A., Eghan, E.E. and Adams, B. `On the Co-evolution of ML Pipelines and S
    :target: https://dvc.org
    :alt: DVC logo
 
+.. |VS Code Extension Overview| image:: https://raw.githubusercontent.com/iterative/vscode-dvc/main/extension/docs/overview.gif
+   :alt: DVC Extension for VS Code
+
 .. |CI| image:: https://github.com/iterative/dvc/workflows/Tests/badge.svg?branch=main
    :target: https://github.com/iterative/dvc/actions
    :alt: GHA Tests
@@ -262,14 +248,14 @@ Barrak, A., Eghan, E.E. and Adams, B. `On the Co-evolution of ML Pipelines and S
    :target: https://codeclimate.com/github/iterative/dvc
    :alt: Code Climate
 
+.. |Python Version| image:: https://img.shields.io/pypi/pyversions/dvc
+   :target: https://pypi.org/project/dvc
+   :alt: Python Version
+
 .. |Coverage| image:: https://codecov.io/gh/iterative/dvc/branch/main/graph/badge.svg
    :target: https://codecov.io/gh/iterative/dvc
    :alt: Codecov
 
-.. |Donate| image:: https://img.shields.io/badge/patreon-donate-green.svg?logo=patreon
-   :target: https://www.patreon.com/DVCorg/overview
-   :alt: Donate
-
 .. |Snap| image:: https://img.shields.io/badge/snap-install-82BEA0.svg?logo=snapcraft
    :target: https://snapcraft.io/dvc
    :alt: Snapcraft
@@ -290,6 +276,10 @@ Barrak, A., Eghan, E.E. and Adams, B. `On the Co-evolution of ML Pipelines and S
    :target: https://pypi.org/project/dvc
    :alt: PyPI
 
+.. |PyPI Downloads| image:: https://img.shields.io/pypi/dm/dvc.svg?color=blue&label=Downloads&logo=pypi&logoColor=gold
+   :target: https://pypi.org/project/dvc
+   :alt: PyPI Downloads
+
 .. |Packages| image:: https://img.shields.io/github/v/release/iterative/dvc?label=deb|pkg|rpm|exe&logo=GitHub
    :target: https://github.com/iterative/dvc/releases/latest
    :alt: deb|pkg|rpm|exe
@@ -305,3 +295,7 @@ Barrak, A., Eghan, E.E. and Adams, B. `On the Co-evolution of ML Pipelines and S
 .. |Contribs| image:: https://contrib.rocks/image?repo=iterative/dvc
    :target: https://github.com/iterative/dvc/graphs/contributors
    :alt: Contributors
+
+.. |VS Code| image:: https://img.shields.io/visual-studio-marketplace/v/Iterative.dvc?color=blue&label=VSCode&logo=visualstudiocode&logoColor=blue
+   :target: https://marketplace.visualstudio.com/items?itemName=Iterative.dvc
+   :alt: VS Code Extension

bin/dvc

@@ -1,3 +0,0 @@
-#!/bin/bash
-
-PYTHONPATH=$DVC_HOME python -m dvc $@

bin/dvc.bat

@@ -1,3 +0,0 @@
-@echo OFF
-set PYTHONPATH=%DVC_HOME%
-python -m dvc %*

dvc/__init__.py

@@ -4,6 +4,6 @@ DVC
 Make your data science projects reproducible and shareable.
 """
 import dvc.logger
-from dvc.version import __version__  # noqa: F401
+from dvc.version import __version__, version_tuple  # noqa: F401
 
 dvc.logger.setup()

dvc/_debug.py

@@ -1,16 +1,14 @@
 from contextlib import ExitStack, contextmanager
 from datetime import datetime
-from typing import TYPE_CHECKING, Callable, Union
+from typing import TYPE_CHECKING, Callable, Optional, Union
 
 if TYPE_CHECKING:
     from argparse import Namespace
+    from types import FrameType
 
 
 @contextmanager
-def viztracer_profile(
-    path: Union[Callable[[], str], str],
-    max_stack_depth: int = -1,
-):
+def viztracer_profile(path: Union[Callable[[], str], str], depth: int = -1):
     try:
         import viztracer  # pylint: disable=import-error
     except ImportError:
@@ -18,7 +16,7 @@ def viztracer_profile(
         yield
         return
 
-    tracer = viztracer.VizTracer(max_stack_depth=max_stack_depth)
+    tracer = viztracer.VizTracer(max_stack_depth=depth)
 
     tracer.start()
     yield
@@ -29,7 +27,9 @@ def viztracer_profile(
 
 @contextmanager
 def yappi_profile(
-    path: Union[Callable[[], str], str] = None, wall_clock: bool = True
+    path: Union[Callable[[], str], str] = None,
+    wall_clock: Optional[bool] = True,
+    separate_threads: Optional[bool] = False,
 ):
     try:
         import yappi  # pylint: disable=import-error
@@ -44,14 +44,26 @@ def yappi_profile(
     yield
     yappi.stop()
 
-    # pylint:disable=no-member
-    if path:
-        stats = yappi.get_func_stats()
-        fpath = path() if callable(path) else path
-        stats.save(fpath, "callgrind")
+    threads = yappi.get_thread_stats()
+    stats = {}
+    if separate_threads:
+        for thread in threads:
+            ctx_id = thread.id
+            stats[ctx_id] = yappi.get_func_stats(ctx_id=ctx_id)
     else:
-        yappi.get_func_stats().print_all()
-        yappi.get_thread_stats().print_all()
+        stats[None] = yappi.get_func_stats()
+
+    fpath = path() if callable(path) else path
+    for ctx_id, st in stats.items():
+        if fpath:
+            out = f"{fpath}-{ctx_id}" if ctx_id is not None else fpath
+            st.save(out, type="callgrind")
+        else:
+            if ctx_id is not None:
+                print(f"\nThread {ctx_id}")
+            st.print_all()  # pylint:disable=no-member
+            if ctx_id is None:
+                threads.print_all()  # pylint:disable=no-member
 
     yappi.clear_stats()
 
@@ -109,6 +121,38 @@ def debug():
         raise  # prevent from jumping ahead
 
 
+def _sigshow(_, frame: Optional["FrameType"]) -> None:
+    from shutil import get_terminal_size
+    from traceback import format_stack
+
+    from dvc.ui import ui
+
+    lines = "\u2015" * get_terminal_size().columns
+    ui.error_write(lines, "\n", *format_stack(frame), lines, sep="")
+
+
+@contextmanager
+def show_stack():
+    r"""Show stack trace on SIGQUIT (Ctrl-\) or SIGINFO (Ctrl-T on macOS)."""
+    import signal
+
+    signal.signal(signal.SIGQUIT, _sigshow)
+    try:
+        signal.signal(signal.SIGINFO, _sigshow)  # only available on macOS
+    except AttributeError:
+        pass
+    yield
+
+
+def _get_path_func(tool: str, ext: str):
+    fmt = f"{tool}.dvc-{{now:%Y%m%d}}_{{now:%H%M%S}}.{ext}"
+
+    def func(now: Optional["datetime"] = None) -> str:
+        return fmt.format(now=now or datetime.now())
+
+    return func
+
+
 @contextmanager
 def debugtools(args: "Namespace" = None, **kwargs):
     kw = vars(args) if args else {}
@@ -121,19 +165,21 @@ def debugtools(args: "Namespace" = None, **kwargs):
             stack.enter_context(profile(kw.get("cprofile_dump")))
         if kw.get("instrument") or kw.get("instrument_open"):
             stack.enter_context(instrument(kw.get("instrument_open", False)))
+        if kw.get("show_stack", False):
+            stack.enter_context(show_stack())
         if kw.get("yappi"):
-            output = "callgrind.dvc-{0:%Y%m%d}_{0:%H%M%S}.out"
+            path_func = _get_path_func("callgrind", "out")
             stack.enter_context(
-                yappi_profile(path=lambda: output.format(datetime.now()))
-            )
-        if kw.get("viztracer") or kw.get("viztracer_depth"):
-            output = "viztracer.dvc-{0:%Y%m%d}_{0:%H%M%S}.json"
-            stack.enter_context(
-                viztracer_profile(
-                    path=lambda: output.format(datetime.now()),
-                    max_stack_depth=kw.get("viztracer_depth") or -1,
+                yappi_profile(
+                    path=path_func,
+                    separate_threads=kw.get("yappi_separate_threads"),
                 )
             )
+        if kw.get("viztracer") or kw.get("viztracer_depth"):
+            path_func = _get_path_func("viztracer", "json")
+            depth = kw.get("viztracer_depth") or -1
+            prof = viztracer_profile(path=path_func, depth=depth)
+            stack.enter_context(prof)
         yield
 
 
@@ -146,6 +192,12 @@ def add_debugging_flags(parser):
     parser.add_argument(
         "--yappi", action="store_true", default=False, help=SUPPRESS
     )
+    parser.add_argument(
+        "--yappi-separate-threads",
+        action="store_true",
+        default=False,
+        help=SUPPRESS,
+    )
     parser.add_argument(
         "--viztracer", action="store_true", default=False, help=SUPPRESS
     )
@@ -160,3 +212,10 @@ def add_debugging_flags(parser):
     parser.add_argument(
         "--instrument-open", action="store_true", default=False, help=SUPPRESS
     )
+    parser.add_argument(
+        "--show-stack",
+        "--ss",
+        action="store_true",
+        default=False,
+        help=SUPPRESS,
+    )

dvc/analytics.py

@@ -50,7 +50,7 @@ def is_enabled():
             Config(validate=False).get("core", {}).get("analytics", "true")
         )
 
-    logger.debug("Analytics is {}abled.".format("en" if enabled else "dis"))
+    logger.debug("Analytics is %sabled.", "en" if enabled else "dis")
 
     return enabled
 
@@ -102,6 +102,8 @@ def _runtime_info():
     """
     Gather information from the environment where DVC runs to fill a report.
     """
+    from iterative_telemetry import find_or_create_user_id
+
     from dvc import __version__
     from dvc.utils import is_binary
 
@@ -110,7 +112,7 @@ def _runtime_info():
         "is_binary": is_binary(),
         "scm_class": _scm_in_use(),
         "system_info": _system_info(),
-        "user_id": _find_or_create_user_id(),
+        "user_id": find_or_create_user_id(),
     }
 
 
@@ -146,45 +148,3 @@ def _system_info():
 
     # We don't collect data for any other system.
     raise NotImplementedError
-
-
-def _find_or_create_user_id():
-    """
-    The user's ID is stored on a file under the global config directory.
-
-    The file should contain a JSON with a "user_id" key:
-
-        {"user_id": "16fd2706-8baf-433b-82eb-8c7fada847da"}
-
-    IDs are generated randomly with UUID.
-    """
-    import uuid
-
-    from dvc.config import Config
-    from dvc.lock import Lock, LockError
-    from dvc.utils.fs import makedirs
-
-    config_dir = Config.get_dir("global")
-    fname = os.path.join(config_dir, "user_id")
-    lockfile = os.path.join(config_dir, "user_id.lock")
-
-    # Since the `fname` and `lockfile` are under the global config,
-    # we need to make sure such directory exist already.
-    makedirs(config_dir, exist_ok=True)
-
-    try:
-        with Lock(lockfile):
-            try:
-                with open(fname, encoding="utf-8") as fobj:
-                    user_id = json.load(fobj)["user_id"]
-
-            except (FileNotFoundError, ValueError, KeyError):
-                user_id = str(uuid.uuid4())
-
-                with open(fname, "w", encoding="utf-8") as fobj:
-                    json.dump({"user_id": user_id}, fobj)
-
-            return user_id
-
-    except LockError:
-        logger.debug(f"Failed to acquire '{lockfile}'")

dvc/annotations.py

@@ -0,0 +1,35 @@
+from dataclasses import asdict, dataclass, field, fields
+from typing import Any, ClassVar, Dict, List, Optional
+
+from funcy import compact
+
+
+@dataclass
+class Annotation:
+    PARAM_DESC: ClassVar[str] = "desc"
+    PARAM_TYPE: ClassVar[str] = "type"
+    PARAM_LABELS: ClassVar[str] = "labels"
+    PARAM_META: ClassVar[str] = "meta"
+
+    desc: Optional[str] = None
+    type: Optional[str] = None
+    labels: List[str] = field(default_factory=list)
+    meta: Dict[str, Any] = field(default_factory=dict)
+
+    def update(self, **kwargs) -> "Annotation":
+        for attr, value in kwargs.items():
+            if value and hasattr(self, attr):
+                setattr(self, attr, value)
+        return self
+
+    def to_dict(self) -> Dict[str, str]:
+        return compact(asdict(self))
+
+
+ANNOTATION_FIELDS = [field.name for field in fields(Annotation)]
+ANNOTATION_SCHEMA = {
+    Annotation.PARAM_DESC: str,
+    Annotation.PARAM_TYPE: str,
+    Annotation.PARAM_LABELS: [str],
+    Annotation.PARAM_META: object,
+}

dvc/api.py

@@ -1,123 +0,0 @@
-import os
-from contextlib import _GeneratorContextManager as GCM
-
-from funcy import reraise
-
-from dvc.exceptions import OutputNotFoundError, PathMissingError
-from dvc.repo import Repo
-
-
-def get_url(path, repo=None, rev=None, remote=None):
-    """
-    Returns the URL to the storage location of a data file or directory tracked
-    in a DVC repo. For Git repos, HEAD is used unless a rev argument is
-    supplied. The default remote is tried unless a remote argument is supplied.
-
-    Raises OutputNotFoundError if the file is not tracked by DVC.
-
-    NOTE: This function does not check for the actual existence of the file or
-    directory in the remote storage.
-    """
-    with Repo.open(repo, rev=rev, subrepos=True, uninitialized=True) as _repo:
-        with reraise(FileNotFoundError, PathMissingError(path, repo)):
-            info = _repo.repo_fs.info(path)
-
-        dvc_info = info.get("dvc_info")
-        if not dvc_info:
-            raise OutputNotFoundError(path, repo)
-
-        dvc_repo = info["repo"]
-        md5 = dvc_info["md5"]
-
-        return dvc_repo.cloud.get_url_for(remote, checksum=md5)
-
-
-def open(  # noqa, pylint: disable=redefined-builtin
-    path, repo=None, rev=None, remote=None, mode="r", encoding=None
-):
-    """
-    Open file in the supplied path tracked in a repo (both DVC projects and
-    plain Git repos are supported). For Git repos, HEAD is used unless a rev
-    argument is supplied. The default remote is tried unless a remote argument
-    is supplied. It may only be used as a context manager:
-
-        with dvc.api.open(
-                'path/to/file',
-                repo='https://example.com/url/to/repo'
-                ) as fd:
-            # ... Handle file object fd
-    """
-    args = (path,)
-    kwargs = {
-        "repo": repo,
-        "remote": remote,
-        "rev": rev,
-        "mode": mode,
-        "encoding": encoding,
-    }
-    return _OpenContextManager(_open, args, kwargs)
-
-
-class _OpenContextManager(GCM):
-    def __init__(
-        self, func, args, kwds
-    ):  # pylint: disable=super-init-not-called
-        self.gen = func(*args, **kwds)
-        self.func, self.args, self.kwds = func, args, kwds
-
-    def __getattr__(self, name):
-        raise AttributeError(
-            "dvc.api.open() should be used in a with statement."
-        )
-
-
-def _open(path, repo=None, rev=None, remote=None, mode="r", encoding=None):
-    with Repo.open(repo, rev=rev, subrepos=True, uninitialized=True) as _repo:
-        with _repo.open_by_relpath(
-            path, remote=remote, mode=mode, encoding=encoding
-        ) as fd:
-            yield fd
-
-
-def read(path, repo=None, rev=None, remote=None, mode="r", encoding=None):
-    """
-    Returns the contents of a tracked file (by DVC or Git). For Git repos, HEAD
-    is used unless a rev argument is supplied. The default remote is tried
-    unless a remote argument is supplied.
-    """
-    with open(
-        path, repo=repo, rev=rev, remote=remote, mode=mode, encoding=encoding
-    ) as fd:
-        return fd.read()
-
-
-def make_checkpoint():
-    """
-    Signal DVC to create a checkpoint experiment.
-
-    If the current process is being run from DVC, this function will block
-    until DVC has finished creating the checkpoint. Otherwise, this function
-    will return immediately.
-    """
-    import builtins
-    from time import sleep
-
-    from dvc.env import DVC_CHECKPOINT, DVC_ROOT
-    from dvc.stage.monitor import CheckpointTask
-
-    if os.getenv(DVC_CHECKPOINT) is None:
-        return
-
-    root_dir = os.getenv(DVC_ROOT, Repo.find_root())
-    signal_file = os.path.join(
-        root_dir, Repo.DVC_DIR, "tmp", CheckpointTask.SIGNAL_FILE
-    )
-
-    with builtins.open(signal_file, "w", encoding="utf-8") as fobj:
-        # NOTE: force flushing/writing empty file to disk, otherwise when
-        # run in certain contexts (pytest) file may not actually be written
-        fobj.write("")
-        fobj.flush()
-        os.fsync(fobj.fileno())
-    while os.path.exists(signal_file):
-        sleep(0.1)

dvc/api/__init__.py

@@ -0,0 +1,15 @@
+from dvc.fs.dvc import _DVCFileSystem as DVCFileSystem
+
+from .data import open  # pylint: disable=redefined-builtin
+from .data import get_url, read
+from .experiments import make_checkpoint
+from .params import params_show
+
+__all__ = [
+    "get_url",
+    "make_checkpoint",
+    "open",
+    "params_show",
+    "read",
+    "DVCFileSystem",
+]

dvc/api/data.py

@@ -0,0 +1,213 @@
+from contextlib import _GeneratorContextManager as GCM
+from typing import Optional
+
+from funcy import reraise
+
+from dvc.exceptions import OutputNotFoundError, PathMissingError
+from dvc.repo import Repo
+
+
+def get_url(path, repo=None, rev=None, remote=None):
+    """
+    Returns the URL to the storage location of a data file or directory tracked
+    in a DVC repo. For Git repos, HEAD is used unless a rev argument is
+    supplied. The default remote is tried unless a remote argument is supplied.
+
+    Raises OutputNotFoundError if the file is not tracked by DVC.
+
+    NOTE: This function does not check for the actual existence of the file or
+    directory in the remote storage.
+    """
+    with Repo.open(repo, rev=rev, subrepos=True, uninitialized=True) as _repo:
+        fs_path = _repo.dvcfs.from_os_path(path)
+        with reraise(FileNotFoundError, PathMissingError(path, repo)):
+            info = _repo.dvcfs.info(fs_path)
+
+        dvc_info = info.get("dvc_info")
+        if not dvc_info:
+            raise OutputNotFoundError(path, repo)
+
+        dvc_repo = info["repo"]
+        md5 = dvc_info["md5"]
+
+        return dvc_repo.cloud.get_url_for(remote, checksum=md5)
+
+
+class _OpenContextManager(GCM):
+    def __init__(
+        self, func, args, kwds
+    ):  # pylint: disable=super-init-not-called
+        self.gen = func(*args, **kwds)
+        self.func, self.args, self.kwds = func, args, kwds
+
+    def __getattr__(self, name):
+        raise AttributeError(
+            "dvc.api.open() should be used in a with statement."
+        )
+
+
+def open(  # noqa, pylint: disable=redefined-builtin
+    path: str,
+    repo: Optional[str] = None,
+    rev: Optional[str] = None,
+    remote: Optional[str] = None,
+    mode: str = "r",
+    encoding: Optional[str] = None,
+):
+    """
+    Opens a file tracked in a DVC project.
+
+    This function may only be used as a context manager (using the `with`
+    keyword, as shown in the examples).
+
+    This function makes a direct connection to the remote storage, so the file
+    contents can be streamed. Your code can process the data buffer as it's
+    streamed, which optimizes memory usage.
+
+    Note:
+        Use dvc.api.read() to load the complete file contents
+        in a single function call, no context manager involved.
+        Neither function utilizes disc space.
+
+    Args:
+        path (str): location and file name of the target to open,
+        relative to the root of `repo`.
+        repo (str, optional): location of the DVC project or Git Repo.
+            Defaults to the current DVC project (found by walking up from the
+            current working directory tree).
+            It can be a URL or a file system path.
+            Both HTTP and SSH protocols are supported for online Git repos
+            (e.g. [user@]server:project.git).
+        rev (str, optional): Any `Git revision`_ such as a branch or tag name,
+            a commit hash or a dvc experiment name.
+            Defaults to HEAD.
+            If `repo` is not a Git repo, this option is ignored.
+        remote (str, optional): Name of the `DVC remote`_ used to form the
+            returned URL string.
+            Defaults to the `default remote`_ of `repo`.
+            For local projects, the cache is tried before the default remote.
+        mode (str, optional): Specifies the mode in which the file is opened.
+            Defaults to "r" (read).
+            Mirrors the namesake parameter in builtin `open()`_.
+            Only reading `mode` is supported.
+        encoding(str, optional): `Codec`_ used to decode the file contents.
+            Defaults to None.
+            This should only be used in text mode.
+            Mirrors the namesake parameter in builtin `open()`_.
+
+    Returns:
+        _OpenContextManager: A context manager that generatse a corresponding
+            `file object`_.
+            The exact type of file object depends on the mode used.
+            For more details, please refer to Python's `open()`_ built-in,
+            which is used under the hood.
+
+    Raises:
+        AttributeError: If this method is not used as a context manager.
+        ValueError: If non-read `mode` is used.
+
+    Examples:
+
+        - Use data or models from a DVC repository.
+
+        Any file tracked in a DVC project (and stored remotely) can be
+        processed directly in your Python code with this API.
+        For example, an XML file tracked in a public DVC repo on GitHub can be
+        processed like this:
+
+        >>> from xml.sax import parse
+        >>> import dvc.api
+        >>> from mymodule import mySAXHandler
+
+        >>> with dvc.api.open(
+        ...     'get-started/data.xml',
+        ...     repo='https://github.com/iterative/dataset-registry'
+        ... ) as fd:
+        ...     parse(fd, mySAXHandler)
+
+        We use a SAX XML parser here because dvc.api.open() is able to stream
+        the data from remote storage.
+        The mySAXHandler object should handle the event-driven parsing of the
+        document in this case.
+        This increases the performance of the code (minimizing memory usage),
+        and is typically faster than loading the whole data into memory.
+
+        - Accessing private repos
+
+        This is just a matter of using the right repo argument, for example an
+        SSH URL (requires that the credentials are configured locally):
+
+        >>> import dvc.api
+
+        >>> with dvc.api.open(
+        ...     'features.dat',
+        ...     repo='[email protected]:path/to/repo.git'
+        ... ) as fd:
+        ...     # ... Process 'features'
+        ...     pass
+
+        - Use different versions of data
+
+        Any git revision (see `rev`) can be accessed programmatically.
+        For example, if your DVC repo has tagged releases of a CSV dataset:
+
+        >>> import csv
+        >>> import dvc.api
+        >>> with dvc.api.open(
+        ...     'clean.csv',
+        ...     rev='v1.1.0'
+        ... ) as fd:
+        ...     reader = csv.reader(fd)
+        ...     # ... Process 'clean' data from version 1.1.0
+
+    .. _Git revision:
+        https://git-scm.com/docs/revisions
+
+    .. _DVC remote:
+        https://dvc.org/doc/command-reference/remote
+
+    .. _default remote:
+        https://dvc.org/doc/command-reference/remote/default
+
+    .. _open():
+        https://docs.python.org/3/library/functions.html#open
+
+    .. _Codec:
+        https://docs.python.org/3/library/codecs.html#standard-encodings
+
+    .. _file object:
+        https://docs.python.org/3/glossary.html#term-file-object
+
+    """
+    if "r" not in mode:
+        raise ValueError("Only reading `mode` is supported.")
+
+    args = (path,)
+    kwargs = {
+        "repo": repo,
+        "remote": remote,
+        "rev": rev,
+        "mode": mode,
+        "encoding": encoding,
+    }
+    return _OpenContextManager(_open, args, kwargs)
+
+
+def _open(path, repo=None, rev=None, remote=None, mode="r", encoding=None):
+    with Repo.open(repo, rev=rev, subrepos=True, uninitialized=True) as _repo:
+        with _repo.open_by_relpath(
+            path, remote=remote, mode=mode, encoding=encoding
+        ) as fd:
+            yield fd
+
+
+def read(path, repo=None, rev=None, remote=None, mode="r", encoding=None):
+    """
+    Returns the contents of a tracked file (by DVC or Git). For Git repos, HEAD
+    is used unless a rev argument is supplied. The default remote is tried
+    unless a remote argument is supplied.
+    """
+    with open(
+        path, repo=repo, rev=rev, remote=remote, mode=mode, encoding=encoding
+    ) as fd:
+        return fd.read()

dvc/api/experiments.py

@@ -0,0 +1,33 @@
+import builtins
+import os
+from time import sleep
+
+from dvc.env import DVC_CHECKPOINT, DVC_ROOT
+from dvc.repo import Repo
+from dvc.stage.monitor import CheckpointTask
+
+
+def make_checkpoint():
+    """
+    Signal DVC to create a checkpoint experiment.
+
+    If the current process is being run from DVC, this function will block
+    until DVC has finished creating the checkpoint. Otherwise, this function
+    will return immediately.
+    """
+    if os.getenv(DVC_CHECKPOINT) is None:
+        return
+
+    root_dir = os.getenv(DVC_ROOT, Repo.find_root())
+    signal_file = os.path.join(
+        root_dir, Repo.DVC_DIR, "tmp", CheckpointTask.SIGNAL_FILE
+    )
+
+    with builtins.open(signal_file, "w", encoding="utf-8") as fobj:
+        # NOTE: force flushing/writing empty file to disk, otherwise when
+        # run in certain contexts (pytest) file may not actually be written
+        fobj.write("")
+        fobj.flush()
+        os.fsync(fobj.fileno())
+    while os.path.exists(signal_file):
+        sleep(0.1)

dvc/api/params.py

@@ -0,0 +1,278 @@
+from collections import Counter
+from typing import Dict, Iterable, Optional, Union
+
+from funcy import first
+
+from dvc.exceptions import DvcException
+from dvc.repo import Repo
+
+
+def params_show(
+    *targets: str,
+    repo: Optional[str] = None,
+    stages: Optional[Union[str, Iterable[str]]] = None,
+    rev: Optional[str] = None,
+    deps: bool = False,
+) -> Dict:
+    """Get parameters tracked in `repo`.
+
+    Without arguments, this function will retrieve all params from all tracked
+    parameter files, for the current working tree.
+
+    See the options below to restrict the parameters retrieved.
+
+    Args:
+        *targets (str, optional): Names of the parameter files to retrieve
+        params from. For example, "params.py, myparams.toml".
+        If no `targets` are provided, all parameter files tracked in `dvc.yaml`
+        will be used.
+        Note that targets don't necessarily have to be defined in `dvc.yaml`.
+        repo (str, optional): location of the DVC repository.
+            Defaults to the current project (found by walking up from the
+            current working directory tree).
+            It can be a URL or a file system path.
+            Both HTTP and SSH protocols are supported for online Git repos
+            (e.g. [user@]server:project.git).
+        stages (Union[str, Iterable[str]], optional): Name or names of the
+            stages to retrieve parameters from.
+            Defaults to `None`.
+            If `None`, all parameters from all stages will be retrieved.
+            If this method is called from a different location to the one where
+            the `dvc.yaml` is found, the relative path to the `dvc.yaml` must
+            be provided as a prefix with the syntax `{relpath}:{stage}`.
+            For example: `subdir/dvc.yaml:stage-0` or `../dvc.yaml:stage-1`.
+        rev (str, optional): Name of the `Git revision`_ to retrieve parameters
+            from.
+            Defaults to `None`.
+            An example of git revision can be a branch or tag name, a commit
+            hash or a dvc experiment name.
+            If `repo` is not a Git repo, this option is ignored.
+            If `None`, the current working tree will be used.
+        deps (bool, optional): Whether to retrieve only parameters that are
+            stage dependencies or not.
+            Defaults to `False`.
+
+    Returns:
+        Dict: See Examples below.
+
+    Raises:
+        DvcException: If no params are found in `repo`.
+
+    Examples:
+
+        - No arguments.
+
+        Working on https://github.com/iterative/example-get-started
+
+        >>> import json
+        >>> import dvc.api
+        >>> params = dvc.api.params_show()
+        >>> print(json.dumps(params, indent=4))
+        {
+            "prepare": {
+                "split": 0.2,
+                "seed": 20170428
+            },
+            "featurize": {
+                "max_features": 200,
+                "ngrams": 2
+            },
+            "train": {
+                "seed": 20170428,
+                "n_est": 50,
+                "min_split": 0.01
+            }
+        }
+
+        ---
+
+        - Filtering with `stages`.
+
+        Working on https://github.com/iterative/example-get-started
+
+        `stages` can a single string:
+
+        >>> import json
+        >>> import dvc.api
+        >>> params = dvc.api.params_show(stages="prepare")
+        >>> print(json.dumps(params, indent=4))
+        {
+            "prepare": {
+                "split": 0.2,
+                "seed": 20170428
+            }
+        }
+
+        Or an iterable of strings:
+
+        >>> import json
+        >>> import dvc.api
+        >>> params = dvc.api.params_show(stages=["prepare", "train"])
+        >>> print(json.dumps(params, indent=4))
+        {
+            "prepare": {
+                "split": 0.2,
+                "seed": 20170428
+            },
+            "train": {
+                "seed": 20170428,
+                "n_est": 50,
+                "min_split": 0.01
+            }
+        }
+
+        ---
+
+        - Using `rev`.
+
+        Working on https://github.com/iterative/example-get-started
+
+        >>> import json
+        >>> import dvc.api
+        >>> params = dvc.api.params_show(rev="tune-hyperparams")
+        >>> print(json.dumps(params, indent=4))
+        {
+            "prepare": {
+                "split": 0.2,
+                "seed": 20170428
+            },
+            "featurize": {
+                "max_features": 200,
+                "ngrams": 2
+            },
+            "train": {
+                "seed": 20170428,
+                "n_est": 100,
+                "min_split": 8
+            }
+        }
+
+        ---
+
+        - Using `targets`.
+
+        Working on `multi-params-files` folder of
+        https://github.com/iterative/pipeline-conifguration
+
+        You can pass a single target:
+
+        >>> import json
+        >>> import dvc.api
+        >>> params = dvc.api.params_show("params.yaml")
+        >>> print(json.dumps(params, indent=4))
+        {
+            "run_mode": "prod",
+            "configs": {
+                "dev": "configs/params_dev.yaml",
+                "test": "configs/params_test.yaml",
+                "prod": "configs/params_prod.yaml"
+            },
+            "evaluate": {
+                "dataset": "micro",
+                "size": 5000,
+                "metrics": ["f1", "roc-auc"],
+                "metrics_file": "reports/metrics.json",
+                "plots_cm": "reports/plot_confusion_matrix.png"
+            }
+        }
+
+
+        Or multiple targets:
+
+        >>> import json
+        >>> import dvc.api
+        >>> params = dvc.api.params_show(
+        ...     "configs/params_dev.yaml", "configs/params_prod.yaml")
+        >>> print(json.dumps(params, indent=4))
+        {
+            "configs/params_prod.yaml:run_mode": "prod",
+            "configs/params_prod.yaml:config_file": "configs/params_prod.yaml",
+            "configs/params_prod.yaml:data_load": {
+                "dataset": "large",
+                "sampling": {
+                "enable": true,
+                "size": 50000
+                }
+            },
+            "configs/params_prod.yaml:train": {
+                "epochs": 1000
+            },
+            "configs/params_dev.yaml:run_mode": "dev",
+            "configs/params_dev.yaml:config_file": "configs/params_dev.yaml",
+            "configs/params_dev.yaml:data_load": {
+                "dataset": "development",
+                "sampling": {
+                "enable": true,
+                "size": 1000
+                }
+            },
+            "configs/params_dev.yaml:train": {
+                "epochs": 10
+            }
+        }
+
+        ---
+
+        - Git URL as `repo`.
+
+        >>> import json
+        >>> import dvc.api
+        >>> params = dvc.api.params_show(
+        ...     repo="https://github.com/iterative/demo-fashion-mnist")
+        {
+            "train": {
+                "batch_size": 128,
+                "hidden_units": 64,
+                "dropout": 0.4,
+                "num_epochs": 10,
+                "lr": 0.001,
+                "conv_activation": "relu"
+            }
+        }
+
+
+    .. _Git revision:
+        https://git-scm.com/docs/revisions
+
+    """
+    if isinstance(stages, str):
+        stages = [stages]
+
+    def _onerror_raise(result: Dict, exception: Exception, *args, **kwargs):
+        raise exception
+
+    def _postprocess(params):
+        processed = {}
+        for rev, rev_data in params.items():
+            processed[rev] = {}
+
+            counts = Counter()
+            for file_data in rev_data["data"].values():
+                for k in file_data["data"]:
+                    counts[k] += 1
+
+            for file_name, file_data in rev_data["data"].items():
+                to_merge = {
+                    (k if counts[k] == 1 else f"{file_name}:{k}"): v
+                    for k, v in file_data["data"].items()
+                }
+                processed[rev] = {**processed[rev], **to_merge}
+
+        if "workspace" in processed:
+            del processed["workspace"]
+
+        if not processed:
+            raise DvcException("No params found")
+
+        return processed[first(processed)]
+
+    with Repo.open(repo) as _repo:
+        params = _repo.params.show(
+            revs=rev if rev is None else [rev],
+            targets=targets,
+            deps=deps,
+            onerror=_onerror_raise,
+            stages=stages,
+        )
+
+    return _postprocess(params)

dvc/cli/__init__.py

@@ -6,6 +6,7 @@ import sys
 # Workaround for CPython bug. See [1] and [2] for more info.
 # [1] https://github.com/aws/aws-cli/blob/1.16.277/awscli/clidriver.py#L55
 # [2] https://bugs.python.org/issue29288
+from typing import Optional
 
 "".encode("idna")
 
@@ -36,6 +37,100 @@ def parse_args(argv=None):
     return args
 
 
+def _log_unknown_exceptions() -> None:
+    from dvc.info import get_dvc_info
+    from dvc.logger import FOOTER
+
+    logger.exception("unexpected error")
+    if logger.isEnabledFor(logging.DEBUG):
+        logger.debug("Version info for developers:\n%s", get_dvc_info())
+    logger.info(FOOTER)
+    return None
+
+
+def _log_exceptions(exc: Exception) -> Optional[int]:
+    """Try to log some known exceptions, that are not DVCExceptions."""
+    from dvc.utils import error_link, format_link
+
+    if isinstance(exc, OSError):
+        import errno
+
+        if exc.errno == errno.EMFILE:
+            logger.exception(
+                "too many open files, please visit "
+                "{} to see how to handle this "
+                "problem".format(error_link("many-files")),
+                extra={"tb_only": True},
+            )
+        else:
+            _log_unknown_exceptions()
+        return None
+
+    from dvc.fs import AuthError, ConfigError, RemoteMissingDepsError
+
+    if isinstance(exc, RemoteMissingDepsError):
+        from dvc.utils.pkg import PKG
+
+        proto = exc.protocol
+        by_pkg = {
+            "pip": f"pip install 'dvc[{proto}]'",
+            "conda": f"conda install -c conda-forge dvc-{proto}",
+        }
+
+        cmd = by_pkg.get(PKG)
+        if cmd:
+            link = format_link("https://dvc.org/doc/install")
+            hint = (
+                f"To install dvc with those dependencies, run:\n"
+                "\n"
+                f"\t{cmd}\n"
+                "\n"
+                f"See {link} for more info."
+            )
+        else:
+            link = format_link("https://github.com/iterative/dvc/issues")
+            hint = f"\nPlease report this bug to {link}. Thank you!"
+
+        logger.exception(
+            f"URL '{exc.url}' is supported but requires these missing "
+            f"dependencies: {exc.missing_deps}. {hint}",
+            extra={"tb_only": True},
+        )
+        return None
+
+    if isinstance(exc, (AuthError, ConfigError)):
+        link = format_link("https://man.dvc.org/remote/modify")
+        logger.exception("configuration error")
+        logger.exception(
+            f"{exc!s}\nLearn more about configuration settings at {link}.",
+            extra={"tb_only": True},
+        )
+        return 251
+
+    from dvc_data.hashfile.cache import DiskError
+
+    if isinstance(exc, DiskError):
+        from dvc.utils import relpath
+
+        directory = relpath(exc.directory)
+        logger.exception(
+            f"Could not open pickled '{exc.type}' cache.\n"
+            f"Remove the '{directory}' directory and then retry this command."
+            f"\nSee {error_link('pickle')} for more information.",
+            extra={"tb_only": True},
+        )
+        return None
+
+    from dvc_data.hashfile.build import IgnoreInCollectedDirError
+
+    if isinstance(exc, IgnoreInCollectedDirError):
+        logger.exception("")
+        return None
+
+    _log_unknown_exceptions()
+    return None
+
+
 def main(argv=None):  # noqa: C901
     """Main entry point for dvc CLI.
 
@@ -48,7 +143,7 @@ def main(argv=None):  # noqa: C901
     from dvc._debug import debugtools
     from dvc.config import ConfigError
     from dvc.exceptions import DvcException, NotDvcRepoError
-    from dvc.logger import FOOTER, disable_other_loggers
+    from dvc.logger import disable_other_loggers, set_loggers_level
 
     # NOTE: stderr/stdout may be closed if we are running from dvc.daemon.
     # On Linux we directly call cli.main after double forking and closing
@@ -76,7 +171,7 @@ def main(argv=None):  # noqa: C901
             level = logging.TRACE
 
         if level is not None:
-            logger.setLevel(level)
+            set_loggers_level(level)
 
         logger.trace(args)
 
@@ -94,6 +189,15 @@ def main(argv=None):  # noqa: C901
     except KeyboardInterrupt:
         logger.exception("interrupted by the user")
         ret = 252
+    except BrokenPipeError:
+        import os
+
+        # Python flushes standard streams on exit; redirect remaining output
+        # to devnull to avoid another BrokenPipeError at shutdown
+        # See: https://docs.python.org/3/library/signal.html#note-on-sigpipe
+        devnull = os.open(os.devnull, os.O_WRONLY)
+        os.dup2(devnull, sys.stdout.fileno())
+        ret = 141  # 128 + 13 (SIGPIPE)
     except NotDvcRepoError:
         logger.exception("")
         ret = 253
@@ -104,27 +208,7 @@ def main(argv=None):  # noqa: C901
         ret = 254
     except Exception as exc:  # noqa, pylint: disable=broad-except
         # pylint: disable=no-member
-        import errno
-
-        if isinstance(exc, OSError) and exc.errno == errno.EMFILE:
-            from dvc.utils import error_link
-
-            logger.exception(
-                "too many open files, please visit "
-                "{} to see how to handle this "
-                "problem".format(error_link("many-files")),
-                extra={"tb_only": True},
-            )
-        else:
-            from dvc.info import get_dvc_info
-
-            logger.exception("unexpected error")
-
-            dvc_info = get_dvc_info()
-            logger.debug("Version info for developers:\n%s", dvc_info)
-
-            logger.info(FOOTER)
-        ret = 255
+        ret = _log_exceptions(exc) or 255
 
     try:
         from dvc import analytics

dvc/cli/actions.py

@@ -0,0 +1,20 @@
+from argparse import _AppendAction
+
+
+class CommaSeparatedArgs(_AppendAction):  # pylint: disable=protected-access
+    def __call__(self, parser, namespace, values, option_string=None):
+        from funcy import ldistinct
+
+        items = getattr(namespace, self.dest) or []
+        items.extend(map(str.strip, values.split(",")))
+        setattr(namespace, self.dest, ldistinct(items))
+
+
+class KeyValueArgs(_AppendAction):
+    def __call__(self, parser, namespace, values, option_string=None):
+        items = getattr(namespace, self.dest) or {}
+        for value in filter(bool, values):
+            key, _, value = value.partition("=")
+            items[key.strip()] = value
+
+        setattr(namespace, self.dest, items)

dvc/cli/command.py

@@ -13,7 +13,7 @@ class CmdBase(ABC):
 
         os.chdir(args.cd)
 
-        self.repo = Repo(uninitialized=self.UNINITIALIZED)
+        self.repo: "Repo" = Repo(uninitialized=self.UNINITIALIZED)
         self.config = self.repo.config
         self.args = args
 

dvc/cli/parser.py

@@ -2,8 +2,8 @@
 import argparse
 import logging
 import os
-import sys
 
+from dvc import __version__
 from dvc.commands import (
     add,
     cache,
@@ -14,6 +14,7 @@ from dvc.commands import (
     config,
     daemon,
     dag,
+    data,
     data_sync,
     destroy,
     diff,
@@ -28,11 +29,13 @@ from dvc.commands import (
     init,
     install,
     ls,
+    ls_url,
     machine,
     metrics,
     move,
     params,
     plots,
+    queue,
     remote,
     remove,
     repro,
@@ -50,6 +53,7 @@ logger = logging.getLogger(__name__)
 
 COMMANDS = [
     init,
+    queue,
     get,
     get_url,
     destroy,
@@ -72,6 +76,7 @@ COMMANDS = [
     install,
     root,
     ls,
+    ls_url,
     freeze,
     dag,
     daemon,
@@ -86,6 +91,7 @@ COMMANDS = [
     experiments,
     check_ignore,
     machine,
+    data,
 ]
 
 
@@ -122,17 +128,6 @@ class DvcParser(argparse.ArgumentParser):
         return args
 
 
-class VersionAction(argparse.Action):  # pragma: no cover
-    # pylint: disable=too-few-public-methods
-    """Shows DVC version and exits."""
-
-    def __call__(self, parser, namespace, values, option_string=None):
-        from dvc import __version__
-
-        print(__version__)
-        sys.exit(0)
-
-
 def get_parent_parser():
     """Create instances of a parser containing common arguments shared among
     all the commands.
@@ -180,13 +175,11 @@ def get_main_parser():
         help="Show this help message and exit.",
     )
 
-    # NOTE: On some python versions action='version' prints to stderr
-    # instead of stdout https://bugs.python.org/issue18920
     parser.add_argument(
         "-V",
         "--version",
-        action=VersionAction,
-        nargs=0,
+        action="version",
+        version=__version__,
         help="Show program's version.",
     )
 

dvc/cli/utils.py

@@ -21,10 +21,13 @@ def append_doc_link(help_message, path):
     )
 
 
-def fix_plumbing_subparsers(subparsers):
-    # metavar needs to be explicitly set in order to hide plumbing subcommands
+def hide_subparsers_from_help(subparsers):
+    # metavar needs to be explicitly set in order to hide subcommands
     # from the 'positional arguments' choices list
     # see: https://bugs.python.org/issue22848
+    # Need to set `add_help=False`, but avoid setting `help`
+    # (not even to `argparse.SUPPPRESS`).
+    # NOTE: The argument is the parent subparser, not the subcommand parser.
     cmds = [
         cmd for cmd, parser in subparsers.choices.items() if parser.add_help
     ]

dvc/commands/add.py

@@ -2,12 +2,43 @@ import argparse
 import logging
 
 from dvc.cli import completion
+from dvc.cli.actions import KeyValueArgs
 from dvc.cli.command import CmdBase
 from dvc.cli.utils import append_doc_link
 
 logger = logging.getLogger(__name__)
 
 
+def _add_annotating_args(parser: argparse.ArgumentParser) -> None:
+    parser.add_argument(
+        "--desc",
+        type=str,
+        metavar="<text>",
+        help="User description of the data.",
+    )
+    parser.add_argument(
+        "--meta",
+        metavar="key=value",
+        nargs=1,
+        action=KeyValueArgs,
+        help="Custom metadata to add to the data.",
+    )
+    parser.add_argument(
+        "--label",
+        dest="labels",
+        type=str,
+        action="append",
+        metavar="<str>",
+        help="Label for the data.",
+    )
+    parser.add_argument(
+        "--type",
+        type=str,
+        metavar="<str>",
+        help="Type of the data.",
+    )
+
+
 class CmdAdd(CmdBase):
     def run(self):
         from dvc.exceptions import (
@@ -28,6 +59,9 @@ class CmdAdd(CmdBase):
                 glob=self.args.glob,
                 desc=self.args.desc,
                 out=self.args.out,
+                type=self.args.type,
+                labels=self.args.labels,
+                meta=self.args.meta,
                 remote=self.args.remote,
                 to_remote=self.args.to_remote,
                 jobs=self.args.jobs,
@@ -109,15 +143,8 @@ def add_parser(subparsers, parent_parser):
         ),
         metavar="<number>",
     )
-    parser.add_argument(
-        "--desc",
-        type=str,
-        metavar="<text>",
-        help=(
-            "User description of the data (optional). "
-            "This doesn't affect any DVC operations."
-        ),
-    )
+
+    _add_annotating_args(parser)
     parser.add_argument(
         "targets", nargs="+", help="Input files/directories to add."
     ).complete = completion.FILE

dvc/commands/checkout.py

@@ -55,6 +55,10 @@ class CmdCheckout(CmdBase):
 
         if exc:
             raise exc
+
+        if self.args.relink:
+            msg = "Relinked successfully"
+            ui.write(msg)
         return 0
 
 
@@ -106,7 +110,7 @@ def add_parser(subparsers, parent_parser):
         nargs="*",
         help=(
             "Limit command scope to these tracked files/directories, "
-            ".dvc files, or stage names."
+            ".dvc files and stage names."
         ),
     ).complete = completion.DVC_FILE
     checkout_parser.set_defaults(func=CmdCheckout)

dvc/commands/commit.py

@@ -70,7 +70,9 @@ def add_parser(subparsers, parent_parser):
     commit_parser.add_argument(
         "targets",
         nargs="*",
-        help="stages or .dvc files to commit. Optional. "
-        "(Finds all DVC files in the workspace by default.)",
+        help=(
+            "Limit command scope to these tracked files/directories, "
+            ".dvc files and stage names."
+        ),
     ).complete = completion.DVCFILES_AND_STAGE
     commit_parser.set_defaults(func=CmdCommit)

dvc/commands/dag.py

@@ -1,11 +1,15 @@
 import argparse
+from typing import TYPE_CHECKING
 
 from dvc.cli.command import CmdBase
 from dvc.cli.utils import append_doc_link
 from dvc.ui import ui
 
+if TYPE_CHECKING:
+    from networkx import DiGraph
 
-def _show_ascii(G):
+
+def _show_ascii(G: "DiGraph"):
     from dvc.dagascii import draw
     from dvc.repo.graph import get_pipelines
 
@@ -18,13 +22,25 @@ def _show_ascii(G):
     return "\n".join(ret)
 
 
-def _show_dot(G):
+def _quote_label(node):
+    label = str(node)
+    # Node names should not contain ":" unless they are quoted with "".
+    # See: https://github.com/pydot/pydot/issues/258.
+    if label[0] != '"' and label[-1] != '"':
+        return f'"{label}"'
+    return label
+
+
+def _show_dot(G: "DiGraph"):
     import io
 
+    import networkx as nx
     from networkx.drawing.nx_pydot import write_dot
 
     dot_file = io.StringIO()
-    write_dot(G, dot_file)
+
+    nx.relabel_nodes(G, _quote_label, copy=False)
+    write_dot(G.reverse(), dot_file)
     return dot_file.getvalue()
 
 
@@ -42,7 +58,7 @@ def _show_mermaid(G, markdown: bool = False):
         for node in nodes:
             total_nodes += 1
             node_id = f"node{total_nodes}"
-            graph += f"\n\t{node_id}[{node}]"
+            graph += f'\n\t{node_id}["{node}"]'
             node_ids[node] = node_id
         edges = sorted((str(a), str(b)) for b, a in pipeline.edges)
         for a, b in edges:
@@ -70,9 +86,9 @@ def _collect_targets(repo, target, outs):
             targets.extend([str(out) for out in stage.outs])
             continue
 
-        for out in outs_trie.itervalues(
+        for out in outs_trie.itervalues(  # noqa: B301
             prefix=repo.fs.path.parts(path)
-        ):  # noqa: B301
+        ):
             targets.extend(str(out))
 
     return targets
@@ -188,7 +204,7 @@ def add_parser(subparsers, parent_parser):
     dag_parser.add_argument(
         "target",
         nargs="?",
-        help="Stage or output to show pipeline for (optional). "
+        help="Stage name or output to show pipeline for. "
         "Finds all stages in the workspace by default.",
     )
     dag_parser.set_defaults(func=CmdDAG)

dvc/commands/data.py

@@ -0,0 +1,297 @@
+import argparse
+import logging
+from operator import itemgetter
+from typing import TYPE_CHECKING, Any, Dict, Iterable, Set
+
+from funcy import chunks, compact, log_durations
+
+from dvc.cli import completion
+from dvc.cli.actions import CommaSeparatedArgs
+from dvc.cli.command import CmdBase
+from dvc.cli.utils import (
+    append_doc_link,
+    fix_subparsers,
+    hide_subparsers_from_help,
+)
+from dvc.ui import ui
+from dvc.utils import colorize
+
+if TYPE_CHECKING:
+    from dvc.repo.data import Status as DataStatus
+
+
+logger = logging.getLogger(__name__)
+
+
+class CmdDataStatus(CmdBase):
+    COLORS = {
+        "not_in_cache": "red",
+        "committed": "green",
+        "uncommitted": "yellow",
+        "untracked": "cyan",
+    }
+    LABELS = {
+        "not_in_cache": "Not in cache",
+        "committed": "DVC committed changes",
+        "uncommitted": "DVC uncommitted changes",
+        "untracked": "Untracked files",
+        "unchanged": "DVC unchanged files",
+    }
+    HINTS = {
+        "not_in_cache": ('use "dvc fetch <file>..." to download files',),
+        "committed": (
+            "git commit the corresponding dvc files to update the repo",
+        ),
+        "uncommitted": (
+            'use "dvc commit <file>..." to track changes',
+            'use "dvc checkout <file>..." to discard changes',
+        ),
+        "untracked": (
+            'use "git add <file> ..." or '
+            '"dvc add <file>..." to commit to git or to dvc',
+        ),
+        "git_dirty": (
+            "there are {}changes not tracked by dvc, "
+            'use "git status" to see',
+        ),
+    }
+
+    @staticmethod
+    def _process_status(status: "DataStatus"):
+        """Flatten stage status, and filter empty stage status contents."""
+        for stage, stage_status in status.items():
+            items = stage_status
+            if isinstance(stage_status, dict):
+                items = {
+                    file: state
+                    for state, files in stage_status.items()
+                    for file in files
+                }
+            if not items:
+                continue
+            yield stage, items
+
+    @classmethod
+    def _show_status(cls, status: "DataStatus") -> int:
+        git_info = status.pop("git")  # type: ignore[misc]
+        result = dict(cls._process_status(status))
+        if not result:
+            no_changes = "No changes"
+            if git_info.get("is_empty", False):
+                no_changes += " in an empty git repo"
+            ui.write(f"{no_changes}.")
+
+        for idx, (stage, stage_status) in enumerate(result.items()):
+            if idx:
+                ui.write()
+
+            label = cls.LABELS.get(stage, stage.capitalize() + " files")
+            header = f"{label}:"
+            color = cls.COLORS.get(stage, None)
+
+            ui.write(header)
+            if hints := cls.HINTS.get(stage):
+                for hint in hints:
+                    ui.write(f"  ({hint})")
+
+            if isinstance(stage_status, dict):
+                items = [
+                    ": ".join([state, file])
+                    for file, state in stage_status.items()
+                ]
+            else:
+                items = stage_status
+
+            tabs = "\t".expandtabs(8)
+            for chunk in chunks(1000, items):
+                out = "\n".join(tabs + item for item in chunk)
+                ui.write(colorize(out, color))
+
+        if (hints := cls.HINTS.get("git_dirty")) and git_info.get("is_dirty"):
+            for hint in hints:
+                message = hint.format("other " if result else "")
+                ui.write(f"[blue]({message})[/]", styled=True)
+        return 0
+
+    def run(self) -> int:
+        with log_durations(logger.trace, "in data_status"):  # type: ignore
+            status = self.repo.data_status(
+                granular=self.args.granular,
+                untracked_files=self.args.untracked_files,
+            )
+
+        if not self.args.unchanged:
+            status.pop("unchanged")  # type: ignore[misc]
+        if self.args.untracked_files == "no":
+            status.pop("untracked")
+        if self.args.json:
+            status.pop("git")  # type: ignore[misc]
+            ui.write_json(compact(status))
+            return 0
+        return self._show_status(status)
+
+
+class CmdDataLs(CmdBase):
+    @staticmethod
+    def _show_table(
+        d: Iterable[Dict[str, Any]],
+        filter_types: Set[str],
+        filter_labels: Set[str],
+        markdown: bool = False,
+    ) -> None:
+        from rich.style import Style
+
+        from dvc.compare import TabularData
+
+        td = TabularData(
+            columns=["Path", "Type", "Labels", "Description"], fill_value="-"
+        )
+        for entry in sorted(d, key=itemgetter("path")):
+            typ = entry.get("type", "")
+            desc = entry.get("desc", "-")
+            labels = entry.get("labels", [])
+
+            if filter_types and typ not in filter_types:
+                continue
+            if filter_labels and filter_labels.isdisjoint(labels):
+                continue
+
+            rich_label = ui.rich_text()
+            for index, label in enumerate(labels):
+                if index:
+                    rich_label.append(",")
+                style = Style(bold=label in filter_labels, color="green")
+                rich_label.append(label, style=style)
+
+            if markdown and desc:
+                desc = desc.partition("\n")[0]
+
+            path = ui.rich_text(entry["path"], style="cyan")
+            type_style = Style(bold=typ in filter_types, color="yellow")
+            typ = ui.rich_text(entry.get("type", ""), style=type_style)
+            td.append([path, typ or "-", rich_label or "-", desc])
+
+        td.render(markdown=markdown, rich_table=True)
+
+    def run(self):
+        from dvc.repo.data import ls
+
+        filter_labels = set(self.args.labels)
+        filter_types = set(self.args.type)
+        d = ls(
+            self.repo, targets=self.args.targets, recursive=self.args.recursive
+        )
+        self._show_table(
+            d,
+            filter_labels=filter_labels,
+            filter_types=filter_types,
+            markdown=self.args.markdown,
+        )
+        return 0
+
+
+def add_parser(subparsers, parent_parser):
+    data_parser = subparsers.add_parser(
+        "data",
+        parents=[parent_parser],
+        formatter_class=argparse.RawDescriptionHelpFormatter,
+    )
+    data_subparsers = data_parser.add_subparsers(
+        dest="cmd",
+        help="Use `dvc data CMD --help` to display command-specific help.",
+    )
+    fix_subparsers(data_subparsers)
+
+    DATA_STATUS_HELP = (
+        "Show changes between the last git commit, "
+        "the dvcfiles and the workspace."
+    )
+    data_status_parser = data_subparsers.add_parser(
+        "status",
+        parents=[parent_parser],
+        description=append_doc_link(DATA_STATUS_HELP, "data/status"),
+        formatter_class=argparse.RawDescriptionHelpFormatter,
+        help=DATA_STATUS_HELP,
+    )
+    data_status_parser.add_argument(
+        "--json",
+        action="store_true",
+        default=False,
+        help="Show output in JSON format.",
+    )
+    data_status_parser.add_argument(
+        "--show-json",
+        action="store_true",
+        default=False,
+        dest="json",
+        help=argparse.SUPPRESS,
+    )
+    data_status_parser.add_argument(
+        "--granular",
+        action="store_true",
+        default=False,
+        help="Show granular file-level info for DVC-tracked directories.",
+    )
+    data_status_parser.add_argument(
+        "--unchanged",
+        action="store_true",
+        default=False,
+        help="Show unmodified DVC-tracked files.",
+    )
+    data_status_parser.add_argument(
+        "--untracked-files",
+        choices=["no", "all"],
+        default="no",
+        const="all",
+        nargs="?",
+        help="Show untracked files.",
+    )
+    data_status_parser.set_defaults(func=CmdDataStatus)
+
+    DATA_LS_HELP = "List data tracked by DVC with its metadata."
+    data_ls_parser = data_subparsers.add_parser(
+        "ls",
+        aliases=["list"],
+        parents=[parent_parser],
+        description=append_doc_link(DATA_LS_HELP, "data/ls"),
+        formatter_class=argparse.RawDescriptionHelpFormatter,
+        add_help=False,
+    )
+    data_ls_parser.add_argument(
+        "--md",
+        "--show-md",
+        dest="markdown",
+        action="store_true",
+        default=False,
+        help="Show tabulated output in the Markdown format (GFM).",
+    )
+    data_ls_parser.add_argument(
+        "--type",
+        action=CommaSeparatedArgs,
+        default=[],
+        help="Comma-separated list of type to filter.",
+    )
+    data_ls_parser.add_argument(
+        "--labels",
+        action=CommaSeparatedArgs,
+        default=[],
+        help="Comma-separated list of labels to filter.",
+    )
+    data_ls_parser.add_argument(
+        "-R",
+        "--recursive",
+        action="store_true",
+        default=False,
+        help="Recursively list from the specified directories.",
+    )
+    data_ls_parser.add_argument(
+        "targets",
+        default=None,
+        nargs="*",
+        help=(
+            "Limit command scope to these tracked files/directories, "
+            ".dvc files, or stage names."
+        ),
+    ).complete = completion.DVCFILES_AND_STAGE
+    data_ls_parser.set_defaults(func=CmdDataLs)
+    hide_subparsers_from_help(data_subparsers)

dvc/commands/data_sync.py

@@ -43,7 +43,8 @@ class CmdDataPull(CmdDataBase):
             )
             self.log_summary(stats)
         except (CheckoutError, DvcException) as exc:
-            self.log_summary(getattr(exc, "stats", {}))
+            if stats := getattr(exc, "stats", {}):
+                self.log_summary(stats)
             logger.exception("failed to pull data from the cloud")
             return 1
 
@@ -119,7 +120,7 @@ def shared_parent_parser():
         nargs="*",
         help=(
             "Limit command scope to these tracked files/directories, "
-            ".dvc files, or stage names."
+            ".dvc files and stage names."
         ),
     ).complete = completion.DVC_FILE
 
@@ -194,7 +195,7 @@ def add_parser(subparsers, _parent_parser):
         "--glob",
         action="store_true",
         default=False,
-        help="Pull cache for targets matching shell-style wildcards.",
+        help=argparse.SUPPRESS,
     )
     pull_parser.set_defaults(func=CmdDataPull)
 

dvc/commands/experiments/__init__.py

@@ -2,8 +2,8 @@ import argparse
 
 from dvc.cli.utils import (
     append_doc_link,
-    fix_plumbing_subparsers,
     fix_subparsers,
+    hide_subparsers_from_help,
 )
 from dvc.commands.experiments import (
     apply,
@@ -15,6 +15,7 @@ from dvc.commands.experiments import (
     ls,
     pull,
     push,
+    queue_worker,
     remove,
     run,
     show,
@@ -30,6 +31,7 @@ SUB_COMMANDS = [
     ls,
     pull,
     push,
+    queue_worker,
     remove,
     run,
     show,
@@ -57,7 +59,7 @@ def add_parser(subparsers, parent_parser):
     fix_subparsers(experiments_subparsers)
     for cmd in SUB_COMMANDS:
         cmd.add_parser(experiments_subparsers, parent_parser)
-    fix_plumbing_subparsers(experiments_subparsers)
+    hide_subparsers_from_help(experiments_subparsers)
 
 
 def add_rev_selection_flags(

dvc/commands/experiments/branch.py

@@ -29,6 +29,6 @@ def add_parser(experiments_subparsers, parent_parser):
         "experiment", help="Experiment to be promoted."
     )
     experiments_branch_parser.add_argument(
-        "branch", help="Git branch name to use."
+        "branch", help="Name for the new Git branch"
     )
     experiments_branch_parser.set_defaults(func=CmdExperimentsBranch)

dvc/commands/experiments/diff.py

@@ -27,7 +27,7 @@ class CmdExperimentsDiff(CmdBase):
 
         if self.args.json:
             ui.write_json(diff)
-        else:
+        elif diff:
             from dvc.compare import show_diff
 
             precision = self.args.precision or DEFAULT_PRECISION

dvc/commands/experiments/gc.py

@@ -12,8 +12,8 @@ logger = logging.getLogger(__name__)
 def _raise_error_if_all_disabled(**kwargs):
     if not any(kwargs.values()):
         raise InvalidArgumentError(
-            "Either of `-w|--workspace`, `-a|--all-branches`, `-T|--all-tags` "
-            "or `--all-commits` needs to be set."
+            "Either of `-w|--workspace`, `-a|--all-branches`, `-T|--all-tags`,"
+            " `--all-commits` or `--date` needs to be set."
         )
 
 
@@ -23,6 +23,7 @@ class CmdExperimentsGC(CmdBase):
             all_branches=self.args.all_branches,
             all_tags=self.args.all_tags,
             all_commits=self.args.all_commits,
+            commit_date=self.args.commit_date,
             workspace=self.args.workspace,
         )
 
@@ -31,13 +32,21 @@ class CmdExperimentsGC(CmdBase):
         msg += "the workspace"
         if self.args.all_commits:
             msg += " and all git commits"
-        elif self.args.all_branches and self.args.all_tags:
-            msg += " and all git branches and tags"
-        elif self.args.all_branches:
-            msg += " and all git branches"
-        elif self.args.all_tags:
-            msg += " and all git tags"
+        else:
+            if self.args.all_branches and self.args.all_tags:
+                msg += " and all git branches and tags"
+            elif self.args.all_branches:
+                msg += " and all git branches"
+            elif self.args.all_tags:
+                msg += " and all git tags"
+
+            if self.args.commit_date:
+                msg += (
+                    " and all git commits before date "
+                    f"{self.args.commit_date}"
+                )
         msg += " of the current repo."
+
         if self.args.queued:
             msg += " Run queued experiments will be preserved."
         else:
@@ -54,6 +63,7 @@ class CmdExperimentsGC(CmdBase):
             all_tags=self.args.all_tags,
             all_commits=self.args.all_commits,
             workspace=self.args.workspace,
+            commit_date=self.args.commit_date,
             queued=self.args.queued,
         )
 
@@ -110,6 +120,17 @@ def add_parser(experiments_subparsers, parent_parser):
         default=False,
         help="Keep experiments derived from all Git commits.",
     )
+    experiments_gc_parser.add_argument(
+        "--date",
+        type=str,
+        dest="commit_date",
+        metavar="<YYYY-MM-DD>",
+        default=None,
+        help=(
+            "Keep experiments from the commits after (inclusive) a certain "
+            "date. Date must match the extended ISO 8601 format (YYYY-MM-DD)."
+        ),
+    )
     experiments_gc_parser.add_argument(
         "--queued",
         action="store_true",

dvc/commands/experiments/init.py

@@ -25,7 +25,6 @@ class CmdExperimentsInit(CmdBase):
     DEFAULT_METRICS = "metrics.json"
     DEFAULT_PARAMS = "params.yaml"
     PLOTS = "plots"
-    DVCLIVE = "dvclive"
     DEFAULTS = {
         "code": CODE,
         "data": DATA,
@@ -33,7 +32,6 @@ class CmdExperimentsInit(CmdBase):
         "metrics": DEFAULT_METRICS,
         "params": DEFAULT_PARAMS,
         "plots": PLOTS,
-        "live": DVCLIVE,
     }
 
     def run(self):
@@ -63,7 +61,7 @@ class CmdExperimentsInit(CmdBase):
             }
         )
 
-        initialized_stage, initialized_deps = init(
+        initialized_stage, initialized_deps, initialized_out_dirs = init(
             self.repo,
             name=self.args.name,
             type=self.args.type,
@@ -72,21 +70,32 @@ class CmdExperimentsInit(CmdBase):
             interactive=self.args.interactive,
             force=self.args.force,
         )
-        self._post_init_display(initialized_stage, initialized_deps)
+        self._post_init_display(
+            initialized_stage, initialized_deps, initialized_out_dirs
+        )
         if self.args.run:
             self.repo.experiments.run(targets=[initialized_stage.addressing])
         return 0
 
     def _post_init_display(
-        self, stage: "PipelineStage", new_deps: List["Dependency"]
+        self,
+        stage: "PipelineStage",
+        new_deps: List["Dependency"],
+        new_out_dirs: List[str],
     ) -> None:
         from dvc.utils import humanize
 
-        path_fmt = "[green]{0}[/green]".format
+        path_fmt = "[green]{}[/green]".format
         if new_deps:
             deps_paths = humanize.join(map(path_fmt, new_deps))
             ui.write(f"Creating dependencies: {deps_paths}", styled=True)
 
+        if new_out_dirs:
+            out_dirs_paths = humanize.join(map(path_fmt, new_out_dirs))
+            ui.write(
+                f"Creating output directories: {out_dirs_paths}", styled=True
+            )
+
         ui.write(
             f"Creating [b]{self.args.name}[/b] stage in [green]dvc.yaml[/]",
             styled=True,
@@ -190,12 +199,11 @@ def add_parser(experiments_subparsers, parent_parser):
     )
     experiments_init_parser.add_argument(
         "--live",
-        help="Path to log dvclive outputs for your experiments"
-        f" (default: {CmdExperimentsInit.DVCLIVE})",
+        help="Path to log dvclive outputs for your experiments",
     )
     experiments_init_parser.add_argument(
         "--type",
-        choices=["default", "dl"],
+        choices=["default", "checkpoint"],
         default="default",
         help="Select type of stage to create (default: %(default)s)",
     )

dvc/commands/experiments/ls.py

@@ -16,13 +16,15 @@ class CmdExperimentsList(CmdBase):
             num=self.args.num,
             git_remote=self.args.git_remote,
         )
+        tags = self.repo.scm.describe(exps)
+        remained = {baseline for baseline, tag in tags.items() if tag is None}
+        base = "refs/heads"
+        ref_heads = self.repo.scm.describe(remained, base=base)
+
         for baseline in exps:
-            tag = self.repo.scm.describe(baseline)
-            if not tag:
-                branch = self.repo.scm.describe(baseline, base="refs/heads")
-                if branch:
-                    tag = branch.split("/")[-1]
-            name = tag if tag else baseline[:7]
+            name = baseline[:7]
+            if tags[baseline] or ref_heads[baseline]:
+                name = tags[baseline] or ref_heads[baseline][len(base) + 1 :]
             if not name_only:
                 print(f"{name}:")
             for exp_name in exps[baseline]:
@@ -59,6 +61,6 @@ def add_parser(experiments_subparsers, parent_parser):
             "If provided, experiments from the specified Git repository "
             " will be listed instead of local ones."
         ),
-        metavar="[<git_remote>]",
+        metavar="<git_remote>",
     )
     experiments_list_parser.set_defaults(func=CmdExperimentsList)

dvc/commands/experiments/queue_worker.py

@@ -0,0 +1,25 @@
+import logging
+
+from dvc.cli.command import CmdBase
+
+logger = logging.getLogger(__name__)
+
+
+class CmdQueueWorker(CmdBase):
+    """Run the exp queue worker."""
+
+    def run(self):
+        self.repo.experiments.celery_queue.worker.start(self.args.name)
+        return 0
+
+
+def add_parser(experiments_subparsers, parent_parser):
+    QUEUE_WORKER_HELP = "Run the exp queue worker."
+    parser = experiments_subparsers.add_parser(
+        "queue-worker",
+        parents=[parent_parser],
+        description=QUEUE_WORKER_HELP,
+        add_help=False,
+    )
+    parser.add_argument("name", help="Celery worker name.")
+    parser.set_defaults(func=CmdQueueWorker)

dvc/commands/experiments/remove.py

@@ -10,23 +10,22 @@ logger = logging.getLogger(__name__)
 
 
 class CmdExperimentsRemove(CmdBase):
-    def raise_error_if_all_disabled(self):
+    def check_arguments(self):
         if not any(
             [
-                self.args.experiment,
                 self.args.all_commits,
                 self.args.rev,
                 self.args.queue,
             ]
-        ):
+        ) ^ bool(self.args.experiment):
             raise InvalidArgumentError(
                 "Either provide an `experiment` argument, or use the "
-                "`--rev` or `--all-commits` flag."
+                "`--rev` or `--all-commits` or `--queue` flag."
             )
 
     def run(self):
 
-        self.raise_error_if_all_disabled()
+        self.check_arguments()
 
         removed_list = self.repo.experiments.remove(
             exp_names=self.args.experiment,

dvc/commands/experiments/show.py

@@ -95,11 +95,8 @@ def _collect_rows(
 
         exp = results.get("data", {})
 
-        if exp.get("running"):
-            state = "Running"
-        elif exp.get("queued"):
-            state = "Queued"
-        else:
+        state = exp.get("status")
+        if state == "Success":
             state = fill_value
 
         is_baseline = rev == "baseline"
@@ -138,7 +135,7 @@ def _collect_rows(
         row_dict["Experiment"] = exp.get("name", "")
         row_dict["rev"] = name_rev
         row_dict["typ"] = typ
-        row_dict["Created"] = _format_time(
+        row_dict["Created"] = format_time(
             exp.get("timestamp"), fill_value, iso
         )
         row_dict["parent"] = parent
@@ -218,7 +215,7 @@ def _sort_exp(experiments, sort_path, sort_name, typ, reverse):
     return ret
 
 
-def _format_time(datetime_obj, fill_value=FILL_VALUE, iso=False):
+def format_time(datetime_obj, fill_value=FILL_VALUE, iso=False):
     if datetime_obj is None:
         return fill_value
 
@@ -476,6 +473,8 @@ class CmdExperimentsShow(CmdBase):
                 all_branches=self.args.all_branches,
                 all_tags=self.args.all_tags,
                 all_commits=self.args.all_commits,
+                hide_queued=self.args.hide_queued,
+                hide_failed=self.args.hide_failed,
                 revs=self.args.rev,
                 num=self.args.num,
                 sha_only=self.args.sha,
@@ -594,6 +593,18 @@ def add_parser(experiments_subparsers, parent_parser):
         default=False,
         help="Always show git commit SHAs instead of branch/tag names.",
     )
+    experiments_show_parser.add_argument(
+        "--hide-failed",
+        action="store_true",
+        default=False,
+        help="Hide failed experiments in the table.",
+    )
+    experiments_show_parser.add_argument(
+        "--hide-queued",
+        action="store_true",
+        default=False,
+        help="Hide queued experiments in the table.",
+    )
     experiments_show_parser.add_argument(
         "--json",
         "--show-json",

dvc/commands/freeze.py

@@ -16,7 +16,7 @@ class CmdFreezeBase(CmdBase):
             try:
                 func(target)
             except DvcException:
-                logger.exception(f"failed to {name} '{target}'")
+                logger.exception("failed to %s '%s'", name, target)
                 ret = 1
         return ret
 

dvc/commands/gc.py

@@ -11,29 +11,46 @@ logger = logging.getLogger(__name__)
 
 class CmdGC(CmdBase):
     def run(self):
-        from dvc.repo.gc import _raise_error_if_all_disabled
+        from dvc.repo.gc import _validate_args
 
-        _raise_error_if_all_disabled(
+        _validate_args(
             all_branches=self.args.all_branches,
             all_tags=self.args.all_tags,
             all_commits=self.args.all_commits,
+            all_experiments=self.args.all_experiments,
+            commit_date=self.args.commit_date,
             workspace=self.args.workspace,
+            rev=self.args.rev,
+            num=self.args.num,
         )
 
+        if self.args.rev:
+            self.args.num = self.args.num or 1
+
         msg = "This will remove all cache except items used in "
 
         msg += "the workspace"
         if self.args.all_commits:
             msg += " and all git commits"
-        elif self.args.all_branches and self.args.all_tags:
-            msg += " and all git branches and tags"
-        elif self.args.all_branches:
-            msg += " and all git branches"
-        elif self.args.all_tags:
-            msg += " and all git tags"
-        elif self.args.all_experiments:
-            msg += " and all experiments"
+        else:
+            if self.args.all_branches and self.args.all_tags:
+                msg += " and all git branches and tags"
+            elif self.args.all_branches:
+                msg += " and all git branches"
+            elif self.args.all_tags:
+                msg += " and all git tags"
+            if self.args.commit_date:
+                msg += (
+                    " and all git commits before date "
+                    f"{self.args.commit_date}"
+                )
+            if self.args.rev:
+                msg += (
+                    f" and last {self.args.num} commits from {self.args.rev}"
+                )
 
+        if self.args.all_experiments:
+            msg += " and all experiments"
         if self.args.repos:
             msg += " of the current and the following repos:"
 
@@ -53,12 +70,15 @@ class CmdGC(CmdBase):
             all_tags=self.args.all_tags,
             all_commits=self.args.all_commits,
             all_experiments=self.args.all_experiments,
+            commit_date=self.args.commit_date,
             cloud=self.args.cloud,
             remote=self.args.remote,
             force=self.args.force,
             jobs=self.args.jobs,
             repos=self.args.repos,
             workspace=self.args.workspace,
+            rev=self.args.rev,
+            num=self.args.num,
         )
         return 0
 
@@ -83,6 +103,26 @@ def add_parser(subparsers, parent_parser):
         default=False,
         help="Keep data files used in the current workspace.",
     )
+    gc_parser.add_argument(
+        "--rev",
+        type=str,
+        default=None,
+        help="Keep data files used in the specified <commit>.",
+        metavar="<commit>",
+    )
+    gc_parser.add_argument(
+        "-n",
+        "--num",
+        type=int,
+        dest="num",
+        metavar="<num>",
+        help=(
+            "Keep data files used in the last `num` commits "
+            "starting from the `--rev` <commit>. "
+            "Only used if `--rev` is also provided. "
+            "Defaults to `1`."
+        ),
+    )
     gc_parser.add_argument(
         "-a",
         "--all-branches",
@@ -104,6 +144,18 @@ def add_parser(subparsers, parent_parser):
         default=False,
         help="Keep data files for all Git commits.",
     )
+    gc_parser.add_argument(
+        "--date",
+        type=str,
+        dest="commit_date",
+        metavar="<YYYY-MM-DD>",
+        default=None,
+        help=(
+            "Keep cached data referenced in the commits after ( inclusive )"
+            " a certain time. Date must match the extended ISO 8601 format "
+            "(YYYY-MM-DD)."
+        ),
+    )
     gc_parser.add_argument(
         "--all-experiments",
         action="store_true",

dvc/commands/get_url.py

@@ -18,7 +18,7 @@ class CmdGetUrl(CmdBaseNoRepo):
             Repo.get_url(self.args.url, out=self.args.out, jobs=self.args.jobs)
             return 0
         except DvcException:
-            logger.exception(f"failed to get '{self.args.url}'")
+            logger.exception("failed to get '%s'", self.args.url)
             return 1
 
 

dvc/commands/git_hook.py

@@ -74,7 +74,7 @@ class CmdMergeDriver(CmdHookBase):
             our = Dvcfile(dvc, self.args.our, verify=False)
             their = Dvcfile(dvc, self.args.their, verify=False)
 
-            our.merge(ancestor, their)
+            our.merge(ancestor, their, allowed=["add", "remove", "change"])
 
             return 0
         finally:

dvc/commands/imp.py

@@ -19,7 +19,11 @@ class CmdImport(CmdBase):
                 fname=self.args.file,
                 rev=self.args.rev,
                 no_exec=self.args.no_exec,
+                no_download=self.args.no_download,
                 desc=self.args.desc,
+                type=self.args.type,
+                labels=self.args.labels,
+                meta=self.args.meta,
                 jobs=self.args.jobs,
             )
         except DvcException:
@@ -33,6 +37,8 @@ class CmdImport(CmdBase):
 
 
 def add_parser(subparsers, parent_parser):
+    from .add import _add_annotating_args
+
     IMPORT_HELP = (
         "Download file or directory tracked by DVC or by Git "
         "into the workspace, and track it."
@@ -69,20 +75,19 @@ def add_parser(subparsers, parent_parser):
         help="Specify name of the .dvc file this command will generate.",
         metavar="<filename>",
     )
-    import_parser.add_argument(
+    no_download_exec_group = import_parser.add_mutually_exclusive_group()
+    no_download_exec_group.add_argument(
         "--no-exec",
         action="store_true",
         default=False,
-        help="Only create .dvc file without actually downloading it.",
+        help="Only create .dvc file without actually importing target data.",
     )
-    import_parser.add_argument(
-        "--desc",
-        type=str,
-        metavar="<text>",
-        help=(
-            "User description of the data (optional). "
-            "This doesn't affect any DVC operations."
-        ),
+    no_download_exec_group.add_argument(
+        "--no-download",
+        action="store_true",
+        default=False,
+        help="Create .dvc file including target data hash value(s)"
+        " but do not actually download the file(s).",
     )
     import_parser.add_argument(
         "-j",
@@ -94,4 +99,6 @@ def add_parser(subparsers, parent_parser):
         ),
         metavar="<number>",
     )
+
+    _add_annotating_args(import_parser)
     import_parser.set_defaults(func=CmdImport)

dvc/commands/imp_url.py

@@ -17,10 +17,15 @@ class CmdImportUrl(CmdBase):
                 out=self.args.out,
                 fname=self.args.file,
                 no_exec=self.args.no_exec,
+                no_download=self.args.no_download,
                 remote=self.args.remote,
                 to_remote=self.args.to_remote,
                 desc=self.args.desc,
+                type=self.args.type,
+                labels=self.args.labels,
+                meta=self.args.meta,
                 jobs=self.args.jobs,
+                version_aware=self.args.version_aware,
             )
         except DvcException:
             logger.exception(
@@ -34,6 +39,8 @@ class CmdImportUrl(CmdBase):
 
 
 def add_parser(subparsers, parent_parser):
+    from .add import _add_annotating_args
+
     IMPORT_HELP = (
         "Download or copy file from URL and take it under DVC control."
     )
@@ -66,11 +73,19 @@ def add_parser(subparsers, parent_parser):
         help="Specify name of the .dvc file this command will generate.",
         metavar="<filename>",
     ).complete = completion.DIR
-    import_parser.add_argument(
+    no_download_exec_group = import_parser.add_mutually_exclusive_group()
+    no_download_exec_group.add_argument(
         "--no-exec",
         action="store_true",
         default=False,
-        help="Only create .dvc file without actually downloading it.",
+        help="Only create .dvc file without actually importing target data.",
+    )
+    no_download_exec_group.add_argument(
+        "--no-download",
+        action="store_true",
+        default=False,
+        help="Create .dvc file including target data hash value(s)"
+        " but do not actually download the file(s).",
     )
     import_parser.add_argument(
         "--to-remote",
@@ -95,12 +110,13 @@ def add_parser(subparsers, parent_parser):
         metavar="<number>",
     )
     import_parser.add_argument(
-        "--desc",
-        type=str,
-        metavar="<text>",
+        "--version-aware",
+        action="store_true",
+        default=False,
         help=(
-            "User description of the data (optional). "
-            "This doesn't affect any DVC operations."
+            "Import using cloud versioning. Implied if the URL contains a "
+            "version ID."
         ),
     )
+    _add_annotating_args(import_parser)
     import_parser.set_defaults(func=CmdImportUrl)

dvc/commands/ls/__init__.py

@@ -42,7 +42,7 @@ class CmdList(CmdBaseNoRepo):
                 ui.write("\n".join(entries))
             return 0
         except DvcException:
-            logger.exception(f"failed to list '{self.args.url}'")
+            logger.exception("failed to list '%s'", self.args.url)
             return 1
 
 

dvc/commands/ls_url.py

@@ -0,0 +1,43 @@
+import argparse
+import logging
+
+from dvc.cli.command import CmdBaseNoRepo
+from dvc.cli.utils import append_doc_link
+from dvc.ui import ui
+
+from .ls import _prettify
+
+logger = logging.getLogger(__name__)
+
+
+class CmdListUrl(CmdBaseNoRepo):
+    def run(self):
+        from dvc.repo import Repo
+
+        entries = Repo.ls_url(self.args.url, recursive=self.args.recursive)
+        if entries:
+            entries = _prettify(entries, with_color=True)
+            ui.write("\n".join(entries))
+        return 0
+
+
+def add_parser(subparsers, parent_parser):
+    LS_HELP = "List directory contents from URL."
+    lsurl_parser = subparsers.add_parser(
+        "list-url",
+        aliases=["ls-url"],
+        parents=[parent_parser],
+        description=append_doc_link(LS_HELP, "list-url"),
+        help=LS_HELP,
+        formatter_class=argparse.RawDescriptionHelpFormatter,
+    )
+    lsurl_parser.add_argument(
+        "url", help="See `dvc import-url -h` for full list of supported URLs."
+    )
+    lsurl_parser.add_argument(
+        "-R",
+        "--recursive",
+        action="store_true",
+        help="Recursively list files.",
+    )
+    lsurl_parser.set_defaults(func=CmdListUrl)

dvc/commands/plots.py

@@ -24,6 +24,49 @@ def _show_json(renderers, split=False):
     ui.write_json(result)
 
 
+def _adjust_vega_renderers(renderers):
+    from dvc.render import REVISION_FIELD, VERSION_FIELD
+    from dvc_render import VegaRenderer
+
+    for r in renderers:
+        if isinstance(r, VegaRenderer):
+            if _data_versions_count(r) > 1:
+                summary = _summarize_version_infos(r)
+                for dp in r.datapoints:
+                    vi = dp.pop(VERSION_FIELD, {})
+                    keys = list(vi.keys())
+                    for key in keys:
+                        if not (len(summary.get(key, set())) > 1):
+                            vi.pop(key)
+                    if vi:
+                        dp["rev"] = "::".join(vi.values())
+            else:
+                for dp in r.datapoints:
+                    dp[REVISION_FIELD] = dp[VERSION_FIELD]["revision"]
+                    dp.pop(VERSION_FIELD, {})
+
+
+def _summarize_version_infos(renderer):
+    from collections import defaultdict
+
+    from dvc.render import VERSION_FIELD
+
+    result = defaultdict(set)
+
+    for dp in renderer.datapoints:
+        for key, value in dp.get(VERSION_FIELD, {}).items():
+            result[key].add(value)
+    return dict(result)
+
+
+def _data_versions_count(renderer):
+    from itertools import product
+
+    summary = _summarize_version_infos(renderer)
+    x = product(summary.get("filename", {None}), summary.get("field", {None}))
+    return len(set(x))
+
+
 class CmdPlots(CmdBase):
     def _func(self, *args, **kwargs):
         raise NotImplementedError
@@ -35,13 +78,30 @@ class CmdPlots(CmdBase):
         props = {p: getattr(self.args, p) for p in PLOT_PROPS}
         return {k: v for k, v in props.items() if v is not None}
 
+    def _config_files(self):
+        config_files = None
+        if self.args.from_config:
+            config_files = {self.args.from_config}
+        return config_files
+
+    def _html_template_path(self):
+        html_template_path = self.args.html_template
+        if not html_template_path:
+            html_template_path = self.repo.config.get("plots", {}).get(
+                "html_template", None
+            )
+            if html_template_path and not os.path.isabs(html_template_path):
+                html_template_path = os.path.join(
+                    self.repo.dvc_dir, html_template_path
+                )
+        return html_template_path
+
     def run(self):
         from pathlib import Path
 
+        from dvc.render.match import match_defs_renderers
         from dvc_render import render_html
 
-        from dvc.render.match import match_renderers
-
         if self.args.show_vega:
             if not self.args.targets:
                 logger.error("please specify a target for `--show-vega`")
@@ -59,9 +119,10 @@ class CmdPlots(CmdBase):
                 return 1
 
         try:
-
             plots_data = self._func(
-                targets=self.args.targets, props=self._props()
+                targets=self.args.targets,
+                props=self._props(),
+                config_files=self._config_files(),
             )
 
             if not plots_data:
@@ -70,54 +131,49 @@ class CmdPlots(CmdBase):
                     "visualization file will not be created."
                 )
 
-            out: str = self.args.out or "dvc_plots"
+            out: str = self.args.out or self.repo.config.get("plots", {}).get(
+                "out_dir", "dvc_plots"
+            )
 
             renderers_out = (
                 out if self.args.json else os.path.join(out, "static")
             )
-            renderers = match_renderers(
-                plots_data=plots_data, out=renderers_out
-            )
 
+            renderers = match_defs_renderers(
+                data=plots_data,
+                out=renderers_out,
+                templates_dir=self.repo.plots.templates_dir,
+            )
+            if self.args.json:
+                _show_json(renderers, self.args.split)
+                return 0
+
+            _adjust_vega_renderers(renderers)
             if self.args.show_vega:
                 renderer = first(filter(lambda r: r.TYPE == "vega", renderers))
                 if renderer:
                     ui.write_json(json.loads(renderer.get_filled_template()))
                 return 0
-            if self.args.json:
-                _show_json(renderers, self.args.split)
-                return 0
-
-            html_template_path = self.args.html_template
-            if not html_template_path:
-                html_template_path = self.repo.config.get("plots", {}).get(
-                    "html_template", None
-                )
-                if html_template_path and not os.path.isabs(
-                    html_template_path
-                ):
-                    html_template_path = os.path.join(
-                        self.repo.dvc_dir, html_template_path
-                    )
 
             output_file: Path = (Path.cwd() / out).resolve() / "index.html"
 
-            render_html(
-                renderers=renderers,
-                output_file=output_file,
-                template_path=html_template_path,
-            )
+            if renderers:
+                render_html(
+                    renderers=renderers,
+                    output_file=output_file,
+                    template_path=self._html_template_path(),
+                )
 
-            ui.write(output_file.as_uri())
-            auto_open = self.repo.config["plots"].get("auto_open", False)
-            if self.args.open or auto_open:
-                if not auto_open:
-                    ui.write(
-                        "To enable auto opening, you can run:\n"
-                        "\n"
-                        "\tdvc config plots.auto_open true"
-                    )
-                return ui.open_browser(output_file)
+                ui.write(output_file.as_uri())
+                auto_open = self.repo.config["plots"].get("auto_open", False)
+                if self.args.open or auto_open:
+                    if not auto_open:
+                        ui.write(
+                            "To enable auto opening, you can run:\n"
+                            "\n"
+                            "\tdvc config plots.auto_open true"
+                        )
+                    return ui.open_browser(output_file)
 
             return 0
 
@@ -154,29 +210,22 @@ class CmdPlotsModify(CmdPlots):
 
 
 class CmdPlotsTemplates(CmdBase):
-    TEMPLATES_CHOICES = [
-        "simple",
-        "linear",
-        "confusion",
-        "confusion_normalized",
-        "scatter",
-        "smooth",
-    ]
-
     def run(self):
-        from dvc_render.vega_templates import dump_templates
+        from dvc.exceptions import InvalidArgumentError
+        from dvc_render.vega_templates import TEMPLATES
 
         try:
-            out = (
-                os.path.join(os.getcwd(), self.args.out)
-                if self.args.out
-                else self.repo.plots.templates_dir
-            )
+            target = self.args.template
+            if target:
+                for template in TEMPLATES:
+                    if target == template.DEFAULT_NAME:
+                        ui.write_json(template.DEFAULT_CONTENT)
+                        return 0
+                raise InvalidArgumentError(f"Unexpected template: {target}.")
 
-            targets = [self.args.target] if self.args.target else None
-            dump_templates(output=out, targets=targets)
-            templates_path = os.path.relpath(out, os.getcwd())
-            ui.write(f"Templates have been written into '{templates_path}'.")
+            else:
+                for template in TEMPLATES:
+                    ui.write(template.DEFAULT_NAME)
 
             return 0
         except DvcException:
@@ -185,10 +234,7 @@ class CmdPlotsTemplates(CmdBase):
 
 
 def add_parser(subparsers, parent_parser):
-    PLOTS_HELP = (
-        "Commands to visualize and compare plot metrics in structured files "
-        "(JSON, YAML, CSV, TSV)."
-    )
+    PLOTS_HELP = "Commands to visualize and compare plot data."
 
     plots_parser = subparsers.add_parser(
         "plots",
@@ -204,7 +250,10 @@ def add_parser(subparsers, parent_parser):
 
     fix_subparsers(plots_subparsers)
 
-    SHOW_HELP = "Generate plots from metrics files."
+    SHOW_HELP = (
+        "Generate plots from target files or from `plots`"
+        " definitions in `dvc.yaml`."
+    )
     plots_show_parser = plots_subparsers.add_parser(
         "show",
         parents=[parent_parser],
@@ -215,9 +264,10 @@ def add_parser(subparsers, parent_parser):
     plots_show_parser.add_argument(
         "targets",
         nargs="*",
-        help="Files to visualize (supports any file, "
-        "even when not found as `plots` in `dvc.yaml`). "
-        "Shows all plots by default.",
+        help=(
+            "Plots files or plot IDs from `dvc.yaml` to visualize. "
+            "Shows all plots by default."
+        ),
     ).complete = completion.FILE
     _add_props_arguments(plots_show_parser)
     _add_output_argument(plots_show_parser)
@@ -225,8 +275,8 @@ def add_parser(subparsers, parent_parser):
     plots_show_parser.set_defaults(func=CmdPlotsShow)
 
     PLOTS_DIFF_HELP = (
-        "Show multiple versions of plot metrics "
-        "by plotting them in a single image."
+        "Show multiple versions of a plot by overlaying them "
+        "in a single image."
     )
     plots_diff_parser = plots_subparsers.add_parser(
         "diff",
@@ -239,8 +289,8 @@ def add_parser(subparsers, parent_parser):
         "--targets",
         nargs="*",
         help=(
-            "Specific plots file(s) to visualize "
-            "(even if not found as `plots` in `dvc.yaml`). "
+            "Specific plots to visualize. "
+            "Accepts any file path or plot name from `dvc.yaml` file. "
             "Shows all tracked plots by default."
         ),
         metavar="<paths>",
@@ -262,7 +312,7 @@ def add_parser(subparsers, parent_parser):
 
     PLOTS_MODIFY_HELP = (
         "Modify display properties of data-series plots "
-        "(has no effect on image-type plots)."
+        "defined in stages (has no effect on image plots)."
     )
     plots_modify_parser = plots_subparsers.add_parser(
         "modify",
@@ -272,7 +322,8 @@ def add_parser(subparsers, parent_parser):
         formatter_class=argparse.RawDescriptionHelpFormatter,
     )
     plots_modify_parser.add_argument(
-        "target", help="Metric file to set properties to"
+        "target",
+        help="Plots file to set properties for (defined at the stage level).",
     ).complete = completion.FILE
     _add_props_arguments(plots_modify_parser)
     plots_modify_parser.add_argument(
@@ -284,8 +335,7 @@ def add_parser(subparsers, parent_parser):
     plots_modify_parser.set_defaults(func=CmdPlotsModify)
 
     TEMPLATES_HELP = (
-        "Write built-in plots templates to a directory "
-        "(.dvc/plots by default)."
+        "List built-in plots templates or show JSON specification for one."
     )
     plots_templates_parser = plots_subparsers.add_parser(
         "templates",
@@ -295,13 +345,14 @@ def add_parser(subparsers, parent_parser):
         formatter_class=argparse.RawDescriptionHelpFormatter,
     )
     plots_templates_parser.add_argument(
-        "target",
+        "template",
         default=None,
         nargs="?",
-        choices=CmdPlotsTemplates.TEMPLATES_CHOICES,
-        help="Template to write. Writes all templates by default.",
+        help=(
+            "Template for which to show JSON specification. "
+            "List all template names by default."
+        ),
     )
-    _add_output_argument(plots_templates_parser, typ="templates")
     plots_templates_parser.set_defaults(func=CmdPlotsTemplates)
 
 
@@ -382,3 +433,9 @@ def _add_ui_arguments(parser):
         help="Custom HTML template for VEGA visualization.",
         metavar="<path>",
     )
+    parser.add_argument(
+        "--from-config",
+        default=None,
+        metavar="<path>",
+        help=argparse.SUPPRESS,
+    )

dvc/commands/queue/__init__.py

@@ -0,0 +1,34 @@
+import argparse
+
+from dvc.cli.utils import append_doc_link, fix_subparsers
+from dvc.commands.queue import kill, logs, remove, start, status, stop
+
+SUB_COMMANDS = [
+    start,
+    stop,
+    status,
+    logs,
+    remove,
+    kill,
+]
+
+
+def add_parser(subparsers, parent_parser):
+    QUEUE_HELP = "Commands to manage experiments queue."
+
+    queue_parser = subparsers.add_parser(
+        "queue",
+        parents=[parent_parser],
+        description=append_doc_link(QUEUE_HELP, "queue"),
+        formatter_class=argparse.RawDescriptionHelpFormatter,
+        help=QUEUE_HELP,
+    )
+
+    queue_subparsers = queue_parser.add_subparsers(
+        dest="cmd",
+        help="Use `dvc queue CMD --help` to display " "command-specific help.",
+    )
+
+    fix_subparsers(queue_subparsers)
+    for cmd in SUB_COMMANDS:
+        cmd.add_parser(queue_subparsers, parent_parser)

dvc/commands/queue/kill.py

@@ -0,0 +1,34 @@
+import argparse
+import logging
+
+from dvc.cli.command import CmdBase
+from dvc.cli.utils import append_doc_link
+
+logger = logging.getLogger(__name__)
+
+
+class CmdQueueKill(CmdBase):
+    """Kill exp task in queue."""
+
+    def run(self):
+        self.repo.experiments.celery_queue.kill(revs=self.args.task)
+
+        return 0
+
+
+def add_parser(queue_subparsers, parent_parser):
+    QUEUE_KILL_HELP = "Kill actively running experiment queue tasks."
+    queue_kill_parser = queue_subparsers.add_parser(
+        "kill",
+        parents=[parent_parser],
+        description=append_doc_link(QUEUE_KILL_HELP, "queue/kill"),
+        help=QUEUE_KILL_HELP,
+        formatter_class=argparse.RawDescriptionHelpFormatter,
+    )
+    queue_kill_parser.add_argument(
+        "task",
+        nargs="*",
+        help="Tasks in queue to kill.",
+        metavar="<task>",
+    )
+    queue_kill_parser.set_defaults(func=CmdQueueKill)

dvc/commands/queue/logs.py

@@ -0,0 +1,56 @@
+import argparse
+import logging
+
+from dvc.cli.command import CmdBase
+from dvc.cli.utils import append_doc_link
+
+logger = logging.getLogger(__name__)
+
+
+class CmdQueueLogs(CmdBase):
+    """Show output logs for a queued experiment."""
+
+    def run(self):
+        self.repo.experiments.celery_queue.logs(
+            rev=self.args.task,
+            encoding=self.args.encoding,
+            follow=self.args.follow,
+        )
+
+        return 0
+
+
+def add_parser(queue_subparsers, parent_parser):
+    QUEUE_LOGS_HELP = (
+        "Show output logs for running and completed experiment queue tasks."
+    )
+    queue_logs_parser = queue_subparsers.add_parser(
+        "logs",
+        parents=[parent_parser],
+        description=append_doc_link(QUEUE_LOGS_HELP, "queue/logs"),
+        help=QUEUE_LOGS_HELP,
+        formatter_class=argparse.RawDescriptionHelpFormatter,
+    )
+    queue_logs_parser.add_argument(
+        "-e",
+        "--encoding",
+        help=(
+            "Text encoding for log output. Defaults to system locale encoding."
+        ),
+        metavar="<encoding>",
+    )
+    queue_logs_parser.add_argument(
+        "-f",
+        "--follow",
+        help=(
+            "Attach to task and follow additional live output. Only "
+            "applicable if the task is still running."
+        ),
+        action="store_true",
+    )
+    queue_logs_parser.add_argument(
+        "task",
+        help="Task to show.",
+        metavar="<task>",
+    )
+    queue_logs_parser.set_defaults(func=CmdQueueLogs)

dvc/commands/queue/remove.py

@@ -0,0 +1,95 @@
+import argparse
+import logging
+
+from dvc.cli.command import CmdBase
+from dvc.cli.utils import append_doc_link
+from dvc.exceptions import InvalidArgumentError
+from dvc.ui import ui
+
+logger = logging.getLogger(__name__)
+
+
+class CmdQueueRemove(CmdBase):
+    """Remove exp in queue."""
+
+    def check_arguments(self):
+        clear_flag = any(
+            [
+                self.args.all,
+                self.args.queued,
+                self.args.failed,
+                self.args.success,
+            ]
+        )
+        if not (clear_flag ^ bool(self.args.task)):
+            raise InvalidArgumentError(
+                "Either provide an `tasks` argument, or use the "
+                "`--all`, `--queued`, `--failed`, `--success` flag."
+            )
+
+    def run(self):
+
+        self.check_arguments()
+
+        if self.args.all:
+            self.args.queued = True
+            self.args.failed = True
+            self.args.success = True
+
+        if self.args.queued or self.args.failed or self.args.success:
+            removed_list = self.repo.experiments.celery_queue.clear(
+                success=self.args.success,
+                queued=self.args.queued,
+                failed=self.args.failed,
+            )
+        else:
+            removed_list = self.repo.experiments.celery_queue.remove(
+                revs=self.args.task,
+            )
+
+        if removed_list:
+            removed = ", ".join(removed_list)
+            ui.write(f"Removed tasks in queue: {removed}")
+        else:
+            ui.write(f"No tasks found named {self.args.task}")
+
+        return 0
+
+
+def add_parser(queue_subparsers, parent_parser):
+
+    QUEUE_REMOVE_HELP = "Remove queued and completed tasks from the queue."
+    queue_remove_parser = queue_subparsers.add_parser(
+        "remove",
+        parents=[parent_parser],
+        description=append_doc_link(QUEUE_REMOVE_HELP, "queue/remove"),
+        help=QUEUE_REMOVE_HELP,
+        formatter_class=argparse.RawDescriptionHelpFormatter,
+    )
+    queue_remove_parser.add_argument(
+        "--all",
+        action="store_true",
+        help="Remove all queued and completed tasks from the queue.",
+    )
+    queue_remove_parser.add_argument(
+        "--queued",
+        action="store_true",
+        help="Remove all queued tasks from the queue.",
+    )
+    queue_remove_parser.add_argument(
+        "--success",
+        action="store_true",
+        help="Remove all successful tasks from the queue.",
+    )
+    queue_remove_parser.add_argument(
+        "--failed",
+        action="store_true",
+        help="Remove all failed tasks from the queue.",
+    )
+    queue_remove_parser.add_argument(
+        "task",
+        nargs="*",
+        help="Tasks to remove.",
+        metavar="<task>",
+    )
+    queue_remove_parser.set_defaults(func=CmdQueueRemove)

dvc/commands/queue/start.py

@@ -0,0 +1,48 @@
+import argparse
+import logging
+
+from dvc.cli.command import CmdBase
+from dvc.cli.utils import append_doc_link
+from dvc.ui import ui
+
+logger = logging.getLogger(__name__)
+
+
+class CmdQueueStart(CmdBase):
+    """Start exp queue workers."""
+
+    def run(self):
+        started = self.repo.experiments.celery_queue.start_workers(
+            self.args.jobs
+        )
+
+        suffix = "s" if started > 1 else ""
+        ui.write(
+            f"Started '{started}' new experiments task queue worker{suffix}."
+        )
+
+        return 0
+
+
+def add_parser(queue_subparsers, parent_parser):
+
+    QUEUE_START_HELP = "Start the experiments task queue worker."
+    queue_start_parser = queue_subparsers.add_parser(
+        "start",
+        parents=[parent_parser],
+        description=append_doc_link(QUEUE_START_HELP, "queue/start"),
+        help=QUEUE_START_HELP,
+        formatter_class=argparse.RawDescriptionHelpFormatter,
+    )
+    queue_start_parser.add_argument(
+        "-j",
+        "--jobs",
+        type=int,
+        default=1,
+        help=(
+            "Maximum number of concurrent queue workers to start. "
+            "Defaults to 1."
+        ),
+        metavar="<number>",
+    )
+    queue_start_parser.set_defaults(func=CmdQueueStart)

dvc/commands/queue/status.py

@@ -0,0 +1,62 @@
+import argparse
+import logging
+from typing import List, Mapping, Optional
+
+from dvc.cli.command import CmdBase
+from dvc.cli.utils import append_doc_link
+from dvc.compare import TabularData
+from dvc.ui import ui
+
+from ..experiments.show import format_time
+
+logger = logging.getLogger(__name__)
+
+
+class CmdQueueStatus(CmdBase):
+    """Show queue task and worker status."""
+
+    def run(self):
+        result: List[
+            Mapping[str, Optional[str]]
+        ] = self.repo.experiments.celery_queue.status()
+        if result:
+            all_headers = ["Task", "Name", "Created", "Status"]
+            td = TabularData(all_headers)
+            for exp in result:
+                created = format_time(exp.get("timestamp"))
+                td.append(
+                    [
+                        exp["rev"][:7],
+                        exp.get("name", ""),
+                        created,
+                        exp["status"],
+                    ]
+                )
+            td.render()
+        else:
+            ui.write("No experiment tasks in the queue.")
+        ui.write()
+
+        worker_status = self.repo.experiments.celery_queue.worker_status()
+        active_count = len(
+            [name for name, task in worker_status.items() if task]
+        )
+        idle_count = len(worker_status) - active_count
+
+        ui.write(f"Worker status: {active_count} active, {idle_count} idle")
+
+        return 0
+
+
+def add_parser(queue_subparsers, parent_parser):
+    QUEUE_STATUS_HELP = (
+        "Show the status of experiments queue tasks and workers."
+    )
+    queue_status_parser = queue_subparsers.add_parser(
+        "status",
+        parents=[parent_parser],
+        description=append_doc_link(QUEUE_STATUS_HELP, "queue/status"),
+        help=QUEUE_STATUS_HELP,
+        formatter_class=argparse.RawDescriptionHelpFormatter,
+    )
+    queue_status_parser.set_defaults(func=CmdQueueStatus)

dvc/commands/queue/stop.py

@@ -0,0 +1,43 @@
+import argparse
+import logging
+
+from dvc.cli.command import CmdBase
+from dvc.cli.utils import append_doc_link
+from dvc.ui import ui
+
+logger = logging.getLogger(__name__)
+
+
+class CmdQueueStop(CmdBase):
+    """Stop exp queue workers."""
+
+    def run(self):
+        self.repo.experiments.celery_queue.shutdown(kill=self.args.kill)
+
+        if self.args.kill:
+            ui.write(
+                "All running tasks in the queue have been killed."
+                "Queue workers are stopping."
+            )
+        else:
+            ui.write("Queue workers will stop after running tasks finish.")
+
+        return 0
+
+
+def add_parser(queue_subparsers, parent_parser):
+
+    QUEUE_STOP_HELP = "Stop all experiments task queue workers."
+    queue_stop_parser = queue_subparsers.add_parser(
+        "stop",
+        parents=[parent_parser],
+        description=append_doc_link(QUEUE_STOP_HELP, "queue/stop"),
+        help=QUEUE_STOP_HELP,
+        formatter_class=argparse.RawDescriptionHelpFormatter,
+    )
+    queue_stop_parser.add_argument(
+        "--kill",
+        action="store_true",
+        help="Kill all running tasks before stopping the queue workers.",
+    )
+    queue_stop_parser.set_defaults(func=CmdQueueStop)

dvc/commands/repro.py

@@ -16,9 +16,7 @@ class CmdRepro(CmdBase):
         if len(stages) == 0:
             ui.write(CmdDataStatus.UP_TO_DATE_MSG)
         else:
-            ui.write(
-                "Use `dvc push` to send your updates to " "remote storage."
-            )
+            ui.write("Use `dvc push` to send your updates to remote storage.")
 
         if self.args.metrics:
             from dvc.compare import show_metrics
@@ -57,7 +55,16 @@ def add_arguments(repro_parser):
     repro_parser.add_argument(
         "targets",
         nargs="*",
-        help="Stages to reproduce. 'dvc.yaml' by default.",
+        help=(
+            """\
+Stages to reproduce. 'dvc.yaml' by default.
+The targets can be path to a dvc.yaml file or `.dvc` file,
+or a stage name from dvc.yaml file from
+current working directory. To run a stage from dvc.yaml
+from other directories, the target must be a path followed by colon `:`
+and then the stage name name.
+"""
+        ),
     ).complete = completion.DVCFILES_AND_STAGE
     repro_parser.add_argument(
         "-f",
@@ -66,6 +73,13 @@ def add_arguments(repro_parser):
         default=False,
         help="Reproduce even if dependencies were not changed.",
     )
+    repro_parser.add_argument(
+        "-i",
+        "--interactive",
+        action="store_true",
+        default=False,
+        help="Ask for confirmation before reproducing each stage.",
+    )
     repro_parser.add_argument(
         "-s",
         "--single-item",
@@ -74,27 +88,6 @@ def add_arguments(repro_parser):
         help="Reproduce only single data item without recursive dependencies "
         "check.",
     )
-    repro_parser.add_argument(
-        "-m",
-        "--metrics",
-        action="store_true",
-        default=False,
-        help="Show metrics after reproduction.",
-    )
-    repro_parser.add_argument(
-        "--dry",
-        action="store_true",
-        default=False,
-        help="Only print the commands that would be executed without "
-        "actually executing.",
-    )
-    repro_parser.add_argument(
-        "-i",
-        "--interactive",
-        action="store_true",
-        default=False,
-        help="Ask for confirmation before reproducing each stage.",
-    )
     repro_parser.add_argument(
         "-p",
         "--pipeline",
@@ -118,11 +111,11 @@ def add_arguments(repro_parser):
         help="Reproduce all stages in the specified directory.",
     )
     repro_parser.add_argument(
-        "--force-downstream",
+        "-m",
+        "--metrics",
         action="store_true",
         default=False,
-        help="Reproduce all descendants of a changed stage even if their "
-        "direct dependencies didn't change.",
+        help="Show metrics after reproduction.",
     )
     repro_parser.add_argument(
         "--downstream",
@@ -130,6 +123,13 @@ def add_arguments(repro_parser):
         default=False,
         help="Start from the specified stages when reproducing pipelines.",
     )
+    repro_parser.add_argument(
+        "--force-downstream",
+        action="store_true",
+        default=False,
+        help="Reproduce all descendants of a changed stage even if their "
+        "direct dependencies didn't change.",
+    )
     repro_parser.add_argument(
         "--pull",
         action="store_true",
@@ -139,6 +139,13 @@ def add_arguments(repro_parser):
             "from the run-cache."
         ),
     )
+    repro_parser.add_argument(
+        "--dry",
+        action="store_true",
+        default=False,
+        help="Only print the commands that would be executed without "
+        "actually executing.",
+    )
 
 
 def add_parser(subparsers, parent_parser):
@@ -156,13 +163,10 @@ def add_parser(subparsers, parent_parser):
     add_arguments(repro_parser)
     # repro only args
     repro_parser.add_argument(
-        "--no-run-cache",
+        "--glob",
         action="store_true",
         default=False,
-        help=(
-            "Execute stage commands even if they have already been run with "
-            "the same command/dependencies/outputs/etc before."
-        ),
+        help="Allows targets containing shell-style wildcards.",
     )
     repro_parser.add_argument(
         "--no-commit",
@@ -171,9 +175,12 @@ def add_parser(subparsers, parent_parser):
         help="Don't put files/directories into cache.",
     )
     repro_parser.add_argument(
-        "--glob",
+        "--no-run-cache",
         action="store_true",
         default=False,
-        help="Allows targets containing shell-style wildcards.",
+        help=(
+            "Execute stage commands even if they have already been run with "
+            "the same command/dependencies/outputs/etc before."
+        ),
     )
     repro_parser.set_defaults(func=CmdRepro)

dvc/commands/run.py

@@ -69,12 +69,25 @@ def add_parser(subparsers, parent_parser):
     run_parser.add_argument(
         "--file", metavar="<filename>", help=argparse.SUPPRESS
     )
+    run_parser.add_argument(
+        "--single-stage",
+        action="store_true",
+        default=False,
+        help=argparse.SUPPRESS,
+    )
+    _add_common_args(run_parser)
     run_parser.add_argument(
         "--no-exec",
         action="store_true",
         default=False,
         help="Only create dvc.yaml without actually running it.",
     )
+    run_parser.add_argument(
+        "--no-commit",
+        action="store_true",
+        default=False,
+        help="Don't put files/directories into cache.",
+    )
     run_parser.add_argument(
         "--no-run-cache",
         action="store_true",
@@ -84,17 +97,4 @@ def add_parser(subparsers, parent_parser):
             "with the same command/dependencies/outputs/etc before."
         ),
     )
-    run_parser.add_argument(
-        "--no-commit",
-        action="store_true",
-        default=False,
-        help="Don't put files/directories into cache.",
-    )
-    run_parser.add_argument(
-        "--single-stage",
-        action="store_true",
-        default=False,
-        help=argparse.SUPPRESS,
-    )
-    _add_common_args(run_parser)
     run_parser.set_defaults(func=CmdRun)

dvc/commands/stage.py

@@ -76,7 +76,7 @@ class CmdStageList(CmdBase):
         # removing duplicates while maintaining order
         collected = chain.from_iterable(
             self.repo.stage.collect(
-                target=target, recursive=self.args.recursive, accept_group=True
+                target=target, recursive=self.args.recursive
             )
             for target in self.args.targets
         )
@@ -137,6 +137,13 @@ class CmdStageAdd(CmdBase):
 
 
 def _add_common_args(parser):
+    parser.add_argument(
+        "-f",
+        "--force",
+        action="store_true",
+        default=False,
+        help="Overwrite existing stage",
+    )
     parser.add_argument(
         "-d",
         "--deps",
@@ -145,6 +152,14 @@ def _add_common_args(parser):
         help="Declare dependencies for reproducible cmd.",
         metavar="<path>",
     ).complete = completion.FILE
+    parser.add_argument(
+        "-p",
+        "--params",
+        action="append",
+        default=[],
+        help="Declare parameter to use as additional dependency.",
+        metavar="[<filename>:]<params_list>",
+    ).complete = completion.FILE
     parser.add_argument(
         "-o",
         "--outs",
@@ -163,13 +178,36 @@ def _add_common_args(parser):
         metavar="<filename>",
     ).complete = completion.FILE
     parser.add_argument(
-        "-p",
-        "--params",
+        "-c",
+        "--checkpoints",
         action="append",
         default=[],
-        help="Declare parameter to use as additional dependency.",
-        metavar="[<filename>:]<params_list>",
+        help="Declare checkpoint output file or directory for 'dvc exp run'. "
+        "Not compatible with 'dvc repro'.",
+        metavar="<filename>",
     ).complete = completion.FILE
+    parser.add_argument(
+        "--external",
+        action="store_true",
+        default=False,
+        help="Allow outputs that are outside of the DVC repository.",
+    )
+    parser.add_argument(
+        "--outs-persist",
+        action="append",
+        default=[],
+        help="Declare output file or directory that will not be "
+        "removed upon repro.",
+        metavar="<filename>",
+    )
+    parser.add_argument(
+        "--outs-persist-no-cache",
+        action="append",
+        default=[],
+        help="Declare output file or directory that will not be "
+        "removed upon repro (do not put into DVC cache).",
+        metavar="<filename>",
+    )
     parser.add_argument(
         "-m",
         "--metrics",
@@ -200,76 +238,18 @@ def _add_common_args(parser):
         help="Declare output plot file (do not put into DVC cache).",
         metavar="<path>",
     )
-    parser.add_argument(
-        "--live", help="Declare output as dvclive.", metavar="<path>"
-    )
-    parser.add_argument(
-        "--live-no-cache",
-        help="Declare output as dvclive (do not put into DVC cache).",
-        metavar="<path>",
-    )
-    parser.add_argument(
-        "--live-no-summary",
-        action="store_true",
-        default=False,
-        help=argparse.SUPPRESS,
-    )
-    parser.add_argument(
-        "--live-no-html",
-        action="store_true",
-        default=False,
-        help="Signal dvclive logger to not produce training report.",
-    )
     parser.add_argument(
         "-w",
         "--wdir",
         help="Directory within your repo to run your command in.",
         metavar="<path>",
     )
-    parser.add_argument(
-        "-f",
-        "--force",
-        action="store_true",
-        default=False,
-        help="Overwrite existing stage",
-    )
-    parser.add_argument(
-        "--outs-persist",
-        action="append",
-        default=[],
-        help="Declare output file or directory that will not be "
-        "removed upon repro.",
-        metavar="<filename>",
-    )
-    parser.add_argument(
-        "--outs-persist-no-cache",
-        action="append",
-        default=[],
-        help="Declare output file or directory that will not be "
-        "removed upon repro (do not put into DVC cache).",
-        metavar="<filename>",
-    )
-    parser.add_argument(
-        "-c",
-        "--checkpoints",
-        action="append",
-        default=[],
-        help="Declare checkpoint output file or directory for 'dvc exp run'. "
-        "Not compatible with 'dvc repro'.",
-        metavar="<filename>",
-    ).complete = completion.FILE
     parser.add_argument(
         "--always-changed",
         action="store_true",
         default=False,
         help="Always consider this DVC-file as changed.",
     )
-    parser.add_argument(
-        "--external",
-        action="store_true",
-        default=False,
-        help="Allow outputs that are outside of the DVC repository.",
-    )
     parser.add_argument(
         "--desc",
         type=str,

dvc/commands/update.py

@@ -18,6 +18,7 @@ class CmdUpdate(CmdBase):
                 rev=self.args.rev,
                 recursive=self.args.recursive,
                 to_remote=self.args.to_remote,
+                no_download=self.args.no_download,
                 remote=self.args.remote,
                 jobs=self.args.jobs,
             )
@@ -28,7 +29,10 @@ class CmdUpdate(CmdBase):
 
 
 def add_parser(subparsers, parent_parser):
-    UPDATE_HELP = "Update data artifacts imported from other DVC repositories."
+    UPDATE_HELP = (
+        "Update data artifact imported (via dvc import or dvc import-url) "
+        "from an external DVC repository or URL."
+    )
     update_parser = subparsers.add_parser(
         "update",
         parents=[parent_parser],
@@ -52,6 +56,13 @@ def add_parser(subparsers, parent_parser):
         default=False,
         help="Update all stages in the specified directory.",
     )
+    update_parser.add_argument(
+        "--no-download",
+        action="store_true",
+        default=False,
+        help="Update .dvc file git revision/hash value(s)"
+        " but do not download the file(s).",
+    )
     update_parser.add_argument(
         "--to-remote",
         action="store_true",

dvc/config.py

@@ -86,9 +86,11 @@ class Config(dict):
     def __init__(
         self, dvc_dir=None, validate=True, fs=None, config=None
     ):  # pylint: disable=super-init-not-called
-        from dvc.fs.local import LocalFileSystem
+        from dvc.fs import LocalFileSystem
 
         self.dvc_dir = dvc_dir
+        self.wfs = LocalFileSystem()
+        self.fs = fs or self.wfs
 
         if not dvc_dir:
             try:
@@ -98,10 +100,7 @@ class Config(dict):
             except NotDvcRepoError:
                 self.dvc_dir = None
         else:
-            self.dvc_dir = os.path.abspath(os.path.realpath(dvc_dir))
-
-        self.wfs = LocalFileSystem(url=self.dvc_dir)
-        self.fs = fs or self.wfs
+            self.dvc_dir = self.fs.path.abspath(self.fs.path.realpath(dvc_dir))
 
         self.load(validate=validate, config=config)
 
@@ -124,8 +123,8 @@ class Config(dict):
         }
 
         if self.dvc_dir is not None:
-            files["repo"] = os.path.join(self.dvc_dir, self.CONFIG)
-            files["local"] = os.path.join(self.dvc_dir, self.CONFIG_LOCAL)
+            files["repo"] = self.fs.path.join(self.dvc_dir, self.CONFIG)
+            files["local"] = self.fs.path.join(self.dvc_dir, self.CONFIG_LOCAL)
 
         return files
 
@@ -177,7 +176,10 @@ class Config(dict):
 
         if fs.exists(filename):
             with fs.open(filename) as fobj:
-                conf_obj = ConfigObj(fobj)
+                try:
+                    conf_obj = ConfigObj(fobj)
+                except UnicodeDecodeError as exc:
+                    raise ConfigError(str(exc)) from exc
         else:
             conf_obj = ConfigObj()
         return _parse_named(_lower_keys(conf_obj.dict()))
@@ -188,7 +190,7 @@ class Config(dict):
         filename = self.files[level]
         fs = self._get_fs(level)
 
-        logger.debug(f"Writing '{filename}'.")
+        logger.debug("Writing '%s'.", filename)
 
         fs.makedirs(os.path.dirname(filename))
 
@@ -228,7 +230,7 @@ class Config(dict):
 
     @staticmethod
     def _to_relpath(conf_dir, path):
-        from dvc.fs.local import localfs
+        from dvc.fs import localfs
         from dvc.utils import relpath
 
         from .config_schema import RelPath

dvc/config_schema.py

@@ -78,7 +78,9 @@ REMOTE_COMMON = {
     "url": str,
     "checksum_jobs": All(Coerce(int), Range(1)),
     "jobs": All(Coerce(int), Range(1)),
+    Optional("worktree"): Bool,
     Optional("no_traverse"): Bool,  # obsoleted
+    Optional("version_aware"): Bool,
 }
 LOCAL_COMMON = {
     "type": supported_cache_type,
@@ -152,6 +154,8 @@ SCHEMA = {
                     "ssl_verify": Any(Bool, str),
                     "sse": str,
                     "sse_kms_key_id": str,
+                    "sse_customer_algorithm": str,
+                    "sse_customer_key": str,
                     "acl": str,
                     "grant_read": str,
                     "grant_read_acp": str,
@@ -166,6 +170,7 @@ SCHEMA = {
                 "gs": {
                     "projectname": str,
                     "credentialpath": str,
+                    "endpointurl": str,
                     Optional("verify", default=False): Bool,
                     **REMOTE_COMMON,
                 },
@@ -175,6 +180,8 @@ SCHEMA = {
                     "user": str,
                     "password": str,
                     "ask_password": Bool,
+                    "passphrase": str,
+                    "ask_passphrase": Bool,
                     "keyfile": str,
                     "timeout": Coerce(int),
                     "gss_auth": Bool,
@@ -217,6 +224,7 @@ SCHEMA = {
                     **REMOTE_COMMON,
                 },
                 "gdrive": {
+                    "profile": str,
                     "gdrive_use_service_account": Bool,
                     "gdrive_client_id": str,
                     "gdrive_client_secret": str,
@@ -224,6 +232,7 @@ SCHEMA = {
                     "gdrive_service_account_user_email": str,
                     "gdrive_service_account_json_file_path": str,
                     Optional("gdrive_trash_only", default=False): Bool,
+                    Optional("gdrive_acknowledge_abuse", default=False): Bool,
                     Optional("verify", default=True): Bool,
                     **REMOTE_COMMON,
                 },
@@ -269,6 +278,7 @@ SCHEMA = {
     "plots": {
         "html_template": str,
         Optional("auto_open", default=False): Bool,
+        "out_dir": str,
     },
     "exp": {
         "code": str,
@@ -279,4 +289,13 @@ SCHEMA = {
         "plots": str,
         "live": str,
     },
+    "parsing": {
+        "bool": All(Lower, Choices("store_true", "boolean_optional")),
+        "list": All(Lower, Choices("nargs", "append")),
+    },
+    "hydra": {
+        Optional("enabled", default=False): Bool,
+        "config_dir": str,
+        "config_name": str,
+    },
 }

dvc/daemon.py

@@ -6,6 +6,7 @@ import os
 import platform
 import sys
 from subprocess import Popen
+from typing import List
 
 from dvc.env import DVC_DAEMON
 from dvc.utils import fix_env, is_binary
@@ -81,7 +82,7 @@ def _spawn_posix(cmd, env):
 
 
 def _spawn(cmd, env):
-    logger.debug(f"Trying to spawn '{cmd}'")
+    logger.debug("Trying to spawn '%s'", cmd)
 
     if os.name == "nt":
         _spawn_windows(cmd, env)
@@ -90,7 +91,7 @@ def _spawn(cmd, env):
     else:
         raise NotImplementedError
 
-    logger.debug(f"Spawned '{cmd}'")
+    logger.debug("Spawned '%s'", cmd)
 
 
 def daemon(args):
@@ -99,12 +100,14 @@ def daemon(args):
     Args:
         args (list): list of arguments to append to `dvc daemon` command.
     """
+    daemonize(["daemon", "-q"] + args)
+
+
+def daemonize(cmd: List[str]):
     if os.environ.get(DVC_DAEMON):
         logger.debug("skipping launching a new daemon.")
         return
 
-    cmd = ["daemon", "-q"] + args
-
     env = fix_env()
     if not is_binary():
         file_path = os.path.abspath(inspect.stack()[0][1])

dvc/data/__init__.py

@@ -1,33 +0,0 @@
-import logging
-from typing import TYPE_CHECKING, Iterator, Union
-
-from .tree import Tree
-
-if TYPE_CHECKING:
-    from dvc.hash_info import HashInfo
-    from dvc.objects.db import ObjectDB
-    from dvc.objects.file import HashFile
-
-logger = logging.getLogger(__name__)
-
-
-def check(odb: "ObjectDB", obj: "HashFile", **kwargs):
-    if isinstance(obj, Tree):
-        for _, _, oid in obj:
-            odb.check(oid, **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

dvc/data/_progress.py

@@ -1,33 +0,0 @@
-from dvc.progress import Tqdm
-
-
-class QueryingProgress(Tqdm):
-    def __init__(self, iterable=None, total=None, name=None, phase="Querying"):
-        msg_part = "cache in " + f"'{name}'" if name else "remote cache"
-        msg_fmt = "{phase} " + msg_part
-
-        self._estimating_msg = msg_fmt.format(phase="Estimating size of")
-        self._listing_msg = msg_fmt.format(phase="Querying")
-        self.desc = desc = msg_fmt.format(phase=phase)
-        super().__init__(
-            iterable=iterable,
-            desc=desc,
-            total=total,
-            unit="files",
-            unit_scale=False,
-            bar_format=self.BAR_FMT_DEFAULT,
-        )
-
-    def callback(self, phase, *args):
-        total = args[0] if args else self.total
-        completed = args[1] if len(args) > 1 else self.n
-        desc = self.desc
-        if phase == "estimating":
-            desc = self._estimating_msg
-        elif phase == "querying":
-            desc = self._listing_msg
-
-        if desc != self.desc:
-            # let Tqdm take care of when to update
-            self.set_description_str(desc, refresh=False)
-        self.update_to(completed, total)

dvc/data/checkout.py

@@ -1,265 +0,0 @@
-import logging
-from itertools import chain
-
-from dvc import prompt
-from dvc.exceptions import CacheLinkError, CheckoutError, ConfirmRemoveError
-from dvc.fs.utils import test_links, transfer
-from dvc.ignore import DvcIgnoreFilter
-from dvc.types import Optional
-
-from .diff import ROOT
-from .diff import diff as odiff
-from .slow_link_detection import slow_link_guard  # type: ignore[attr-defined]
-from .stage import stage
-
-logger = logging.getLogger(__name__)
-
-
-def _remove(fs_path, fs, in_cache, force=False):
-    if not fs.exists(fs_path):
-        return
-
-    if force:
-        fs.remove(fs_path)
-        return
-
-    if not in_cache:
-        msg = (
-            f"file/directory '{fs_path}' is going to be removed. "
-            "Are you sure you want to proceed?"
-        )
-
-        if not prompt.confirm(msg):
-            raise ConfirmRemoveError(fs_path)
-
-    fs.remove(fs_path)
-
-
-def _relink(link, cache, cache_info, fs, path, in_cache, force):
-    _remove(path, fs, in_cache, force=force)
-    link(cache, cache_info, fs, path)
-    # NOTE: Depending on a file system (e.g. on NTFS), `_remove` might reset
-    # read-only permissions in order to delete a hardlink to protected object,
-    # which will also reset it for the object itself, making it unprotected,
-    # so we need to protect it back.
-    cache.protect(cache_info)
-
-
-def _checkout_file(
-    link,
-    fs_path,
-    fs,
-    change,
-    cache,
-    force,
-    progress_callback=None,
-    relink=False,
-    state=None,
-):
-    """The file is changed we need to checkout a new copy"""
-    modified = False
-
-    cache_fs_path = cache.hash_to_path(change.new.oid.value)
-    if change.old.oid:
-        if relink:
-            if fs.iscopy(fs_path) and cache.cache_types[0] == "copy":
-                cache.unprotect(fs_path)
-            else:
-                _relink(
-                    link,
-                    cache,
-                    cache_fs_path,
-                    fs,
-                    fs_path,
-                    change.old.in_cache,
-                    force=force,
-                )
-        else:
-            modified = True
-            _relink(
-                link,
-                cache,
-                cache_fs_path,
-                fs,
-                fs_path,
-                change.old.in_cache,
-                force=force,
-            )
-    else:
-        link(cache, cache_fs_path, fs, fs_path)
-        modified = True
-
-    if state:
-        state.save(fs_path, fs, change.new.oid)
-
-    if progress_callback:
-        progress_callback(fs_path)
-
-    return modified
-
-
-def _diff(
-    fs_path,
-    fs,
-    obj,
-    cache,
-    relink=False,
-    dvcignore: Optional[DvcIgnoreFilter] = None,
-):
-    old = None
-    try:
-        _, _, old = stage(
-            cache,
-            fs_path,
-            fs,
-            obj.hash_info.name if obj else cache.fs.PARAM_CHECKSUM,
-            dry_run=True,
-            dvcignore=dvcignore,
-        )
-    except FileNotFoundError:
-        pass
-
-    diff = odiff(old, obj, cache)
-
-    if relink:
-        diff.modified.extend(diff.unchanged)
-
-    return diff
-
-
-class Link:
-    def __init__(self, links):
-        self._links = links
-
-    @slow_link_guard
-    def __call__(self, cache, from_path, to_fs, to_path):
-        if to_fs.exists(to_path):
-            to_fs.remove(to_path)  # broken symlink
-
-        cache.makedirs(cache.fs.path.parent(to_path))
-        try:
-            transfer(cache.fs, from_path, to_fs, to_path, links=self._links)
-        except FileNotFoundError as exc:
-            raise CheckoutError([to_path]) from exc
-        except OSError as exc:
-            raise CacheLinkError([to_path]) from exc
-
-
-def _checkout(
-    diff,
-    fs_path,
-    fs,
-    cache,
-    force=False,
-    progress_callback=None,
-    relink=False,
-    state=None,
-):
-    if not diff:
-        return
-
-    links = test_links(cache.cache_types, cache.fs, cache.fs_path, fs, fs_path)
-    if not links:
-        raise CacheLinkError([fs_path])
-    link = Link(links)
-
-    for change in diff.deleted:
-        entry_path = (
-            fs.path.join(fs_path, *change.old.key)
-            if change.old.key != ROOT
-            else fs_path
-        )
-        _remove(entry_path, fs, change.old.in_cache, force=force)
-
-    failed = []
-    for change in chain(diff.added, diff.modified):
-        entry_path = (
-            fs.path.join(fs_path, *change.new.key)
-            if change.new.key != ROOT
-            else fs_path
-        )
-        if change.new.oid.isdir:
-            fs.makedirs(entry_path)
-            continue
-
-        try:
-            _checkout_file(
-                link,
-                entry_path,
-                fs,
-                change,
-                cache,
-                force,
-                progress_callback,
-                relink,
-                state=state,
-            )
-        except CheckoutError as exc:
-            failed.extend(exc.target_infos)
-
-    if failed:
-        raise CheckoutError(failed)
-
-
-def checkout(
-    fs_path,
-    fs,
-    obj,
-    cache,
-    force=False,
-    progress_callback=None,
-    relink=False,
-    quiet=False,
-    dvcignore: Optional[DvcIgnoreFilter] = None,
-    state=None,
-):
-    # if scheme(fs_path) not in ["local", cache.fs.scheme]:
-    #    raise NotImplementedError
-
-    diff = _diff(
-        fs_path,
-        fs,
-        obj,
-        cache,
-        relink=relink,
-        dvcignore=dvcignore,
-    )
-
-    failed = []
-    if not obj:
-        if not quiet:
-            logger.warning(
-                "No file hash info found for '%s'. It won't be created.",
-                fs_path,
-            )
-        failed.append(fs_path)
-    elif not diff:
-        logger.trace("Data '%s' didn't change.", fs_path)  # type: ignore
-
-    try:
-        _checkout(
-            diff,
-            fs_path,
-            fs,
-            cache,
-            force=force,
-            progress_callback=progress_callback,
-            relink=relink,
-            state=state,
-        )
-    except CheckoutError as exc:
-        failed.extend(exc.target_infos)
-
-    if diff and state:
-        state.save_link(fs_path, fs)
-        if not failed:
-            state.save(fs_path, fs, obj.hash_info)
-
-    if failed or not diff:
-        if progress_callback and obj:
-            progress_callback(fs_path, len(obj))
-        if failed:
-            raise CheckoutError(failed)
-        return
-
-    return bool(diff) and not relink

dvc/data/db/index.py

@@ -1,149 +0,0 @@
-import logging
-import os
-from abc import ABC, abstractmethod
-from typing import TYPE_CHECKING, Iterable, Set
-
-from dvc.objects.errors import ObjectDBError
-from dvc.utils.decorators import with_diskcache
-
-if TYPE_CHECKING:
-    from dvc.types import StrPath
-
-logger = logging.getLogger(__name__)
-
-
-class ObjectDBIndexBase(ABC):
-    @abstractmethod
-    def __init__(
-        self,
-        tmp_dir: "StrPath",
-        name: str,
-    ):
-        pass
-
-    @abstractmethod
-    def __iter__(self):
-        pass
-
-    @abstractmethod
-    def __contains__(self, hash_):
-        pass
-
-    def hashes(self):
-        return iter(self)
-
-    @abstractmethod
-    def dir_hashes(self):
-        pass
-
-    @abstractmethod
-    def clear(self):
-        pass
-
-    @abstractmethod
-    def update(self, dir_hashes: Iterable[str], file_hashes: Iterable[str]):
-        pass
-
-    @abstractmethod
-    def intersection(self, hashes: Set[str]):
-        pass
-
-
-class ObjectDBIndexNoop(ObjectDBIndexBase):
-    """No-op class for ODBs which are not indexed."""
-
-    def __init__(
-        self,
-        tmp_dir: "StrPath",
-        name: str,
-    ):  # pylint: disable=super-init-not-called
-        pass
-
-    def __iter__(self):
-        return iter([])
-
-    def __contains__(self, hash_):
-        return False
-
-    def dir_hashes(self):
-        return []
-
-    def clear(self):
-        pass
-
-    def update(self, dir_hashes: Iterable[str], file_hashes: Iterable[str]):
-        pass
-
-    def intersection(self, hashes: Set[str]):
-        return []
-
-
-class ObjectDBIndex(ObjectDBIndexBase):
-    """Class for indexing hashes in an ODB."""
-
-    INDEX_SUFFIX = ".idx"
-    INDEX_DIR = "index"
-
-    def __init__(
-        self,
-        tmp_dir: "StrPath",
-        name: str,
-    ):  # pylint: disable=super-init-not-called
-        from diskcache import Cache, Index
-
-        from dvc.fs.local import LocalFileSystem
-        from dvc.utils.fs import makedirs
-
-        self.index_dir = os.path.join(tmp_dir, self.INDEX_DIR, name)
-        makedirs(self.index_dir, exist_ok=True)
-        self.fs = LocalFileSystem()
-        self.index = Index.fromcache(
-            Cache(
-                self.index_dir,
-                disk_pickle_protocol=4,
-                eviction_policy="none",
-            )
-        )
-
-    @with_diskcache(name="index")
-    def __iter__(self):
-        return iter(self.index)
-
-    @with_diskcache(name="index")
-    def __contains__(self, hash_):
-        return hash_ in self.index
-
-    @with_diskcache(name="index")
-    def dir_hashes(self):
-        """Iterate over .dir hashes stored in the index."""
-        yield from (hash_ for hash_, is_dir in self.index.items() if is_dir)
-
-    @with_diskcache(name="index")
-    def clear(self):
-        """Clear this index (to force re-indexing later)."""
-        from diskcache import Timeout
-
-        try:
-            self.index.clear()
-        except Timeout as exc:
-            raise ObjectDBError("Failed to clear ODB index") from exc
-
-    @with_diskcache(name="index")
-    def update(self, dir_hashes: Iterable[str], file_hashes: Iterable[str]):
-        """Update this index, adding the specified hashes."""
-        from diskcache import Timeout
-
-        try:
-            with self.index.transact():
-                for hash_ in dir_hashes:
-                    self.index[hash_] = True
-            with self.index.transact():
-                for hash_ in file_hashes:
-                    self.index[hash_] = False
-        except Timeout as exc:
-            raise ObjectDBError("Failed to update ODB index") from exc
-
-    @with_diskcache(name="index")
-    def intersection(self, hashes: Set[str]):
-        """Iterate over values from `hashes` which exist in the index."""
-        yield from hashes.intersection(self.index.keys())

dvc/data/db/local.py

@@ -1,166 +0,0 @@
-import logging
-import os
-import stat
-from functools import partial
-
-from funcy import cached_property
-from shortuuid import uuid
-
-from dvc.hash_info import HashInfo
-from dvc.objects.db import ObjectDB, noop, wrap_iter
-from dvc.objects.errors import ObjectFormatError
-from dvc.utils import relpath
-from dvc.utils.fs import copyfile, remove, umask, walk_files
-
-logger = logging.getLogger(__name__)
-
-
-class LocalObjectDB(ObjectDB):
-    DEFAULT_CACHE_TYPES = ["reflink", "copy"]
-    CACHE_MODE = 0o444
-    UNPACKED_DIR_SUFFIX = ".unpacked"
-
-    def __init__(self, fs, fs_path, **config):
-        super().__init__(fs, fs_path, **config)
-        self.cache_dir = fs_path
-
-        shared = config.get("shared")
-        if shared:
-            self._file_mode = 0o664
-            self._dir_mode = 0o2775
-        else:
-            self._file_mode = 0o666 & ~umask
-            self._dir_mode = 0o777 & ~umask
-
-    @property
-    def cache_dir(self):
-        return self.fs_path if self.fs_path else None
-
-    @cache_dir.setter
-    def cache_dir(self, value):
-        self.fs_path = value
-
-    @cached_property
-    def cache_path(self):
-        return os.path.abspath(self.cache_dir)
-
-    def move(self, from_info, to_info):
-        super().move(from_info, to_info)
-        os.chmod(to_info, self._file_mode)
-
-    def makedirs(self, fs_path):
-        from dvc.utils.fs import makedirs
-
-        makedirs(fs_path, exist_ok=True, mode=self._dir_mode)
-
-    def hash_to_path(self, hash_):
-        # NOTE: `self.cache_path` is already normalized so we can simply use
-        # `os.sep` instead of `os.path.join`. This results in this helper
-        # being ~5.5 times faster.
-        return f"{self.cache_path}{os.sep}{hash_[0:2]}{os.sep}{hash_[2:]}"
-
-    def hashes_exist(
-        self, hashes, jobs=None, progress=noop
-    ):  # pylint: disable=unused-argument
-        ret = []
-        progress = partial(progress, "querying", len(hashes))
-
-        for hash_ in wrap_iter(hashes, progress):
-            hash_info = HashInfo(self.fs.PARAM_CHECKSUM, hash_)
-            try:
-                self.check(hash_info)
-                ret.append(hash_)
-            except (FileNotFoundError, ObjectFormatError):
-                pass
-
-        return ret
-
-    def _list_paths(self, prefix=None):
-        assert self.fs_path is not None
-        if prefix:
-            fs_path = self.fs.path.join(self.fs_path, prefix[:2])
-            if not self.fs.exists(fs_path):
-                return
-        else:
-            fs_path = self.fs_path
-
-        # NOTE: use utils.fs walk_files since fs.walk_files will not follow
-        # symlinks
-        yield from walk_files(fs_path)
-
-    def _remove_unpacked_dir(self, hash_):
-        hash_fs_path = self.hash_to_path(hash_)
-        fs_path = self.fs.path.with_name(
-            hash_fs_path,
-            self.fs.path.name(hash_fs_path) + self.UNPACKED_DIR_SUFFIX,
-        )
-        self.fs.remove(fs_path)
-
-    def _unprotect_file(self, path):
-        if self.fs.is_symlink(path) or self.fs.is_hardlink(path):
-            logger.debug("Unprotecting '%s'", path)
-            tmp = os.path.join(os.path.dirname(path), "." + uuid())
-
-            # The operations order is important here - if some application
-            # would access the file during the process of copyfile then it
-            # would get only the part of file. So, at first, the file should be
-            # copied with the temporary name, and then original file should be
-            # replaced by new.
-            copyfile(path, tmp, name=f"Unprotecting '{relpath(path)}'")
-            remove(path)
-            os.rename(tmp, path)
-
-        else:
-            logger.debug(
-                "Skipping copying for '%s', since it is not "
-                "a symlink or a hardlink.",
-                path,
-            )
-
-        os.chmod(path, self._file_mode)
-
-    def _unprotect_dir(self, path):
-        for fname in self.fs.find(path):
-            self._unprotect_file(fname)
-
-    def unprotect(self, fs_path):
-        if not os.path.exists(fs_path):
-            from dvc.exceptions import DvcException
-
-            raise DvcException(
-                f"can't unprotect non-existing data '{fs_path}'"
-            )
-
-        if os.path.isdir(fs_path):
-            self._unprotect_dir(fs_path)
-        else:
-            self._unprotect_file(fs_path)
-
-    def protect(self, fs_path):
-        try:
-            os.chmod(fs_path, self.CACHE_MODE)
-        except OSError:
-            # NOTE: not being able to protect cache file is not fatal, it
-            # might happen on funky filesystems (e.g. Samba, see #5255),
-            # read-only filesystems or in a shared cache scenario.
-            logger.trace("failed to protect '%s'", fs_path, exc_info=True)
-
-    def is_protected(self, fs_path):
-        try:
-            mode = os.stat(fs_path).st_mode
-        except FileNotFoundError:
-            return False
-
-        return stat.S_IMODE(mode) == self.CACHE_MODE
-
-    def set_exec(self, fs_path):
-        mode = os.stat(fs_path).st_mode | stat.S_IEXEC
-        try:
-            os.chmod(fs_path, mode)
-        except OSError:
-            logger.trace(
-                "failed to chmod '%s' '%s'",
-                oct(mode),
-                fs_path,
-                exc_info=True,
-            )

dvc/data/db/reference.py

@@ -1,84 +0,0 @@
-import logging
-import os
-from typing import TYPE_CHECKING, Dict
-
-from dvc.objects.db import ObjectDB
-from dvc.objects.errors import ObjectFormatError
-from dvc.scheme import Schemes
-
-from ..reference import ReferenceHashFile
-
-if TYPE_CHECKING:
-    from dvc.fs.base import AnyFSPath, FileSystem
-    from dvc.hash_info import HashInfo
-
-logger = logging.getLogger(__name__)
-
-
-class ReferenceObjectDB(ObjectDB):
-    """Reference ODB.
-
-    File objects are stored as ReferenceHashFiles which reference paths outside
-    of the staging ODB fs. Tree objects are stored natively.
-    """
-
-    def __init__(self, *args, **kwargs):
-        super().__init__(*args, **kwargs)
-        self._fs_cache: Dict[tuple, "FileSystem"] = {}
-        self._obj_cache: Dict["HashInfo", "ReferenceHashFile"] = {}
-
-    def get(self, hash_info: "HashInfo"):
-        if hash_info.isdir:
-            return super().get(hash_info)
-        try:
-            return self._obj_cache[hash_info]
-        except KeyError:
-            pass
-        fs_path = self.hash_to_path(hash_info.value)
-        try:
-            ref_file = ReferenceHashFile.from_bytes(
-                self.fs.cat_file(fs_path),
-                fs_cache=self._fs_cache,
-            )
-        except OSError:
-            raise FileNotFoundError
-        try:
-            ref_file.check(self, check_hash=False)
-        except ObjectFormatError:
-            self.fs.remove(fs_path)
-            raise
-        self._obj_cache[hash_info] = ref_file
-        return ref_file
-
-    def _add_file(
-        self,
-        from_fs: "FileSystem",
-        from_info: "AnyFSPath",
-        to_info: "AnyFSPath",
-        hash_info: "HashInfo",
-        hardlink: bool = False,
-    ):
-        self.makedirs(self.fs.path.parent(to_info))
-        if hash_info.isdir:
-            return super()._add_file(
-                from_fs,
-                from_info,
-                to_info,
-                hash_info,
-                hardlink=hardlink,
-            )
-        ref_file = ReferenceHashFile(from_info, from_fs, hash_info)
-        self._obj_cache[hash_info] = ref_file
-        try:
-            self.fs.pipe_file(to_info, ref_file.to_bytes())
-        except OSError as exc:
-            if isinstance(exc, FileExistsError) or (
-                os.name == "nt"
-                and exc.__context__
-                and isinstance(exc.__context__, FileExistsError)
-            ):
-                logger.debug("'%s' file already exists, skipping", to_info)
-            else:
-                raise
-        if from_fs.scheme != Schemes.LOCAL:
-            self._fs_cache[ReferenceHashFile.config_tuple(from_fs)] = from_fs

dvc/data/diff.py

@@ -1,133 +0,0 @@
-from dataclasses import dataclass, field
-from typing import TYPE_CHECKING, List, Optional, Tuple
-
-if TYPE_CHECKING:
-    from dvc.hash_info import HashInfo
-    from dvc.objects.file import HashFile
-
-ADD = "add"
-MODIFY = "modify"
-DELETE = "delete"
-UNCHANGED = "unchanged"
-
-
-@dataclass
-class TreeEntry:
-    in_cache: bool
-    key: Tuple[str]
-    oid: Optional["HashInfo"] = field(default=None)
-
-    def __bool__(self):
-        return bool(self.oid)
-
-    def __eq__(self, other):
-        if not isinstance(other, TreeEntry):
-            return False
-
-        if self.key != other.key:
-            return False
-
-        return self.oid == other.oid
-
-
-@dataclass
-class Change:
-    old: TreeEntry
-    new: TreeEntry
-
-    @property
-    def typ(self):
-        if not self.old and not self.new:
-            return UNCHANGED
-
-        if self.old and not self.new:
-            return DELETE
-
-        if not self.old and self.new:
-            return ADD
-
-        if self.old != self.new:
-            return MODIFY
-
-        return UNCHANGED
-
-    def __bool__(self):
-        return self.typ != UNCHANGED
-
-
-@dataclass
-class DiffResult:
-    added: List[Change] = field(default_factory=list, compare=True)
-    modified: List[Change] = field(default_factory=list, compare=True)
-    deleted: List[Change] = field(default_factory=list, compare=True)
-    unchanged: List[Change] = field(default_factory=list, compare=True)
-
-    def __bool__(self):
-        return bool(self.added or self.modified or self.deleted)
-
-
-ROOT = ("",)
-
-
-def diff(
-    old: Optional["HashFile"], new: Optional["HashFile"], cache
-) -> DiffResult:
-    from .tree import Tree
-
-    if old is None and new is None:
-        return DiffResult()
-
-    def _get_keys(obj):
-        if not obj:
-            return []
-        return [ROOT] + (
-            [key for key, _, _ in obj] if isinstance(obj, Tree) else []
-        )
-
-    old_keys = set(_get_keys(old))
-    new_keys = set(_get_keys(new))
-
-    def _get_oid(obj, key):
-        if not obj or key == ROOT:
-            return obj.hash_info if obj else None
-
-        entry_obj = obj.get(cache, key)
-        return entry_obj.hash_info if entry_obj else None
-
-    def _in_cache(oid, cache):
-        from dvc.objects.errors import ObjectFormatError
-
-        if not oid:
-            return False
-
-        try:
-            cache.check(oid)
-            return True
-        except (FileNotFoundError, ObjectFormatError):
-            return False
-
-    ret = DiffResult()
-    for key in old_keys | new_keys:
-        old_oid = _get_oid(old, key)
-        new_oid = _get_oid(new, key)
-
-        change = Change(
-            old=TreeEntry(_in_cache(old_oid, cache), key, old_oid),
-            new=TreeEntry(_in_cache(new_oid, cache), key, new_oid),
-        )
-
-        if change.typ == ADD:
-            ret.added.append(change)
-        elif change.typ == MODIFY:
-            ret.modified.append(change)
-        elif change.typ == DELETE:
-            ret.deleted.append(change)
-        else:
-            assert change.typ == UNCHANGED
-            if not change.new.in_cache and not (
-                change.new.oid and change.new.oid.isdir
-            ):
-                ret.modified.append(change)
-            else:
-                ret.unchanged.append(change)
-    return ret

dvc/data/gc.py

@@ -1,40 +0,0 @@
-def gc(odb, used, jobs=None, cache_odb=None, shallow=True):
-    from dvc.data.tree import Tree
-    from dvc.objects.errors import ObjectDBPermissionError
-
-    from ._progress import QueryingProgress
-
-    if odb.read_only:
-        raise ObjectDBPermissionError("Cannot gc read-only ODB")
-    if not cache_odb:
-        cache_odb = odb
-    used_hashes = set()
-    for hash_info in used:
-        used_hashes.add(hash_info.value)
-        if hash_info.isdir and not shallow:
-            tree = Tree.load(cache_odb, hash_info)
-            used_hashes.update(
-                entry_obj.hash_info.value for _, entry_obj in tree
-            )
-
-    def _is_dir_hash(_hash):
-        from dvc.hash_info import HASH_DIR_SUFFIX
-
-        return _hash.endswith(HASH_DIR_SUFFIX)
-
-    removed = False
-    # hashes must be sorted to ensure we always remove .dir files first
-
-    hashes = QueryingProgress(odb.all(jobs), name=odb.fs_path)
-    for hash_ in sorted(hashes, key=_is_dir_hash, reverse=True):
-        if hash_ in used_hashes:
-            continue
-        fs_path = odb.hash_to_path(hash_)
-        if _is_dir_hash(hash_):
-            # backward compatibility
-            # pylint: disable=protected-access
-            odb._remove_unpacked_dir(hash_)
-        odb.fs.remove(fs_path)
-        removed = True
-
-    return removed

dvc/data/meta.py

@@ -1,47 +0,0 @@
-from collections import OrderedDict
-from dataclasses import dataclass, field
-from typing import TYPE_CHECKING, Optional
-
-if TYPE_CHECKING:
-    from dvc.objects.db import ObjectDB
-    from dvc.objects.file import HashFile
-
-
-@dataclass
-class Meta:
-    PARAM_SIZE = "size"
-    PARAM_NFILES = "nfiles"
-    PARAM_ISEXEC = "isexec"
-
-    size: Optional[int] = field(default=None)
-    nfiles: Optional[int] = field(default=None)
-    isexec: Optional[bool] = field(default=False)
-
-    obj: Optional["HashFile"] = field(default=None)
-    odb: Optional["ObjectDB"] = field(default=None)
-    remote: Optional[str] = field(default=None)
-
-    @classmethod
-    def from_dict(cls, d):
-        if not d:
-            return cls()
-
-        size = d.pop(cls.PARAM_SIZE, None)
-        nfiles = d.pop(cls.PARAM_NFILES, None)
-        isexec = d.pop(cls.PARAM_ISEXEC, False)
-
-        return cls(size=size, nfiles=nfiles, isexec=isexec)
-
-    def to_dict(self):
-        ret = OrderedDict()
-
-        if self.size is not None:
-            ret[self.PARAM_SIZE] = self.size
-
-        if self.nfiles is not None:
-            ret[self.PARAM_NFILES] = self.nfiles
-
-        if self.isexec:
-            ret[self.PARAM_ISEXEC] = self.isexec
-
-        return ret

dvc/data/reference.py

@@ -1,113 +0,0 @@
-import errno
-import logging
-import os
-import pickle
-from typing import TYPE_CHECKING, Optional
-
-from dvc.objects.errors import ObjectFormatError
-from dvc.objects.file import HashFile
-
-if TYPE_CHECKING:
-    from dvc.fs.base import AnyFSPath, FileSystem
-    from dvc.hash_info import HashInfo
-    from dvc.objects.db import ObjectDB
-
-logger = logging.getLogger(__name__)
-
-
-class ReferenceHashFile(HashFile):
-    PARAM_PATH = "path"
-    PARAM_HASH = "hash"
-    PARAM_CHECKSUM = "checksum"
-    PARAM_FS_CONFIG = "fs_config"
-
-    def __init__(
-        self,
-        fs_path: "AnyFSPath",
-        fs: "FileSystem",
-        hash_info: "HashInfo",
-        checksum: Optional[str] = None,
-        **kwargs,
-    ):
-        super().__init__(fs_path, fs, hash_info, **kwargs)
-        self.checksum = checksum or fs.checksum(fs_path)
-
-    def __str__(self):
-        return f"ref object {self.hash_info} -> {self.fs_path}"
-
-    def check(self, odb: "ObjectDB", check_hash: bool = True):
-        assert self.fs
-        if not self.fs.exists(self.fs_path):
-            raise FileNotFoundError(
-                errno.ENOENT, os.strerror(errno.ENOENT), self.fs_path
-            )
-        if self.checksum != self._get_checksum():
-            raise ObjectFormatError(f"{self} is changed")
-        if check_hash:
-            self._check_hash(odb)
-
-    def _get_checksum(self) -> str:
-        assert self.fs
-        return self.fs.checksum(self.fs_path)
-
-    def to_bytes(self):
-        # NOTE: dumping reference FS's this way is insecure, as the
-        # fully parsed remote FS config will include credentials
-        #
-        # ReferenceHashFiles should currently only be serialized in
-        # memory and not to disk
-        fs_path = self.fs_path
-        dict_ = {
-            self.PARAM_PATH: fs_path,
-            self.PARAM_HASH: self.hash_info,
-            self.PARAM_CHECKSUM: self.checksum,
-            self.PARAM_FS_CONFIG: self.config_tuple(self.fs),
-        }
-        try:
-            return pickle.dumps(dict_)
-        except pickle.PickleError as exc:
-            raise ObjectFormatError(f"Could not pickle {self}") from exc
-
-    @classmethod
-    def from_bytes(cls, data: bytes, fs_cache: Optional[dict] = None):
-        from dvc.fs import get_fs_cls
-        from dvc.fs.repo import RepoFileSystem, _RepoFileSystem
-
-        try:
-            dict_ = pickle.loads(data)
-        except pickle.PickleError as exc:
-            raise ObjectFormatError("ReferenceHashFile is corrupted") from exc
-
-        try:
-            fs_path = dict_[cls.PARAM_PATH]
-            hash_info = dict_[cls.PARAM_HASH]
-        except KeyError as exc:
-            raise ObjectFormatError("ReferenceHashFile is corrupted") from exc
-
-        scheme, config_pairs = dict_.get(cls.PARAM_FS_CONFIG)
-        fs = fs_cache.get((scheme, config_pairs)) if fs_cache else None
-        if not fs:
-            config = dict(config_pairs)
-            if _RepoFileSystem.PARAM_REPO_URL in config:
-                fs_cls = RepoFileSystem
-            else:
-                fs_cls = get_fs_cls(config, scheme=scheme)
-            fs = fs_cls(**config)
-        return ReferenceHashFile(
-            fs_path,
-            fs,
-            hash_info,
-            checksum=dict_.get(cls.PARAM_CHECKSUM),
-        )
-
-    @staticmethod
-    def config_tuple(fs: "FileSystem"):
-        return (
-            fs.scheme,
-            tuple(
-                (key, value)
-                for key, value in sorted(
-                    fs.config.items(), key=lambda item: item[0]
-                )
-            ),
-        )

dvc/data/slow_link_detection.py

@@ -1,45 +0,0 @@
-# type: ignore
-import logging
-import sys
-import time
-from functools import wraps
-
-import colorama
-
-logger = logging.getLogger(__name__)
-this = sys.modules[__name__]
-
-this.timeout_seconds = 10.0
-this.already_displayed = False
-this.message = (
-    "You can cut execution time considerably by using a different "
-    "`cache.type` configuration.\n"
-    "See {blue}https://dvc.org/doc/commands-reference/config#cache{reset} "
-    "for more information.\n"
-    "To disable this message, run:\n"
-    "`dvc config cache.slow_link_warning false`".format(
-        blue=colorama.Fore.BLUE, reset=colorama.Fore.RESET
-    )
-)
-
-
-def slow_link_guard(f):
-    @wraps(f)
-    def wrapper(self, odb, *args, **kwargs):
-        if this.already_displayed:
-            return f(self, odb, *args, **kwargs)
-
-        if not odb.slow_link_warning or odb.cache_types:
-            return f(self, odb, *args, **kwargs)
-
-        start = time.time()
-        result = f(self, odb, *args, **kwargs)
-        delta = time.time() - start
-
-        if delta >= this.timeout_seconds:
-            logger.warning(this.message)
-            this.already_displayed = True
-
-        return result
-
-    return wrapper

dvc/data/stage.py

@@ -1,352 +0,0 @@
-import hashlib
-import logging
-import os
-from concurrent.futures import ThreadPoolExecutor
-from functools import partial
-from typing import TYPE_CHECKING, Dict, Optional, Tuple
-
-from dvc.exceptions import DvcIgnoreInCollectedDirError
-from dvc.hash_info import HashInfo
-from dvc.ignore import DvcIgnore
-from dvc.objects.file import HashFile
-from dvc.progress import Tqdm
-from dvc.utils import file_md5, is_exec
-
-from .db.reference import ReferenceObjectDB
-from .meta import Meta
-
-if TYPE_CHECKING:
-    from dvc.fs.base import AnyFSPath, FileSystem
-    from dvc.objects.db import ObjectDB
-
-logger = logging.getLogger(__name__)
-
-
-_STAGING_MEMFS_PATH = "dvc-staging"
-
-
-def _upload_file(from_fs_path, fs, odb, upload_odb, callback=None):
-    from dvc.fs._callback import FsspecCallback
-    from dvc.utils import tmp_fname
-    from dvc.utils.stream import HashedStreamReader
-
-    fs_path = upload_odb.fs.path
-    tmp_info = fs_path.join(upload_odb.fs_path, tmp_fname())
-    with fs.open(from_fs_path, mode="rb") as stream:
-        stream = HashedStreamReader(stream)
-        size = fs.size(from_fs_path)
-        with FsspecCallback.as_tqdm_callback(
-            callback,
-            desc=fs_path.name(from_fs_path),
-            bytes=True,
-            total=size,
-        ) as cb:
-            upload_odb.fs.upload(stream, tmp_info, size=size, callback=cb)
-
-    odb.add(tmp_info, upload_odb.fs, stream.hash_info)
-    meta = Meta(size=size)
-    return from_fs_path, meta, odb.get(stream.hash_info)
-
-
-def _adapt_info(info, scheme):
-    if scheme == "s3" and "ETag" in info:
-        info["etag"] = info["ETag"].strip('"')
-    elif scheme == "gs" and "etag" in info:
-        import base64
-
-        info["etag"] = base64.b64decode(info["etag"]).hex()
-    elif scheme.startswith("http") and (
-        "ETag" in info or "Content-MD5" in info
-    ):
-        info["checksum"] = info.get("ETag") or info.get("Content-MD5")
-    return info
-
-
-def _get_file_hash(fs_path, fs, name):
-    info = _adapt_info(fs.info(fs_path), fs.scheme)
-
-    if name in info:
-        assert not info[name].endswith(".dir")
-        hash_value = info[name]
-    elif hasattr(fs, name):
-        func = getattr(fs, name)
-        hash_value = func(fs_path)
-    elif name == "md5":
-        hash_value = file_md5(fs_path, fs)
-    else:
-        raise NotImplementedError
-
-    meta = Meta(size=info["size"], isexec=is_exec(info.get("mode", 0)))
-    hash_info = HashInfo(name, hash_value)
-    return meta, hash_info
-
-
-def get_file_hash(fs_path, fs, name, state=None):
-    if state:
-        meta, hash_info = state.get(  # pylint: disable=assignment-from-none
-            fs_path, fs
-        )
-        if hash_info:
-            return meta, hash_info
-
-    meta, hash_info = _get_file_hash(fs_path, fs, name)
-
-    if state:
-        assert ".dir" not in hash_info.value
-        state.save(fs_path, fs, hash_info)
-
-    return meta, hash_info
-
-
-def _stage_file(fs_path, fs, name, odb=None, upload_odb=None, dry_run=False):
-    state = odb.state if odb else None
-    meta, hash_info = get_file_hash(fs_path, fs, name, state=state)
-    if upload_odb and not dry_run:
-        assert odb and name == "md5"
-        return _upload_file(fs_path, fs, odb, upload_odb)
-
-    if dry_run:
-        obj = HashFile(fs_path, fs, hash_info)
-    else:
-        odb.add(fs_path, fs, hash_info, hardlink=False)
-        obj = odb.get(hash_info)
-
-    return fs_path, meta, obj
-
-
-def _build_objects(
-    fs_path,
-    fs,
-    name,
-    dvcignore=None,
-    jobs=None,
-    no_progress_bar=False,
-    **kwargs,
-):
-    if dvcignore:
-        walk_iterator = dvcignore.find(fs, fs_path)
-    else:
-        walk_iterator = fs.find(fs_path)
-    with Tqdm(
-        unit="md5",
-        desc="Computing file/dir hashes (only done once)",
-        disable=no_progress_bar,
-    ) as pbar:
-        worker = pbar.wrap_fn(
-            partial(
-                _stage_file,
-                fs=fs,
-                name=name,
-                **kwargs,
-            )
-        )
-        with ThreadPoolExecutor(
-            max_workers=jobs if jobs is not None else fs.hash_jobs
-        ) as executor:
-            yield from executor.map(worker, walk_iterator)
-
-
-def _iter_objects(fs_path, fs, name, **kwargs):
-    yield from _build_objects(fs_path, fs, name, **kwargs)
-
-
-def _build_tree(fs_path, fs, name, **kwargs):
-    from .tree import Tree
-
-    tree_meta = Meta(size=0, nfiles=0)
-    tree = Tree(None, None, None)
-    for file_fs_path, meta, obj in _iter_objects(fs_path, fs, name, **kwargs):
-        if DvcIgnore.DVCIGNORE_FILE == fs.path.name(file_fs_path):
-            raise DvcIgnoreInCollectedDirError(fs.path.parent(file_fs_path))
-
-        # NOTE: this is lossy transformation:
-        #   "hey\there" -> "hey/there"
-        #   "hey/there" -> "hey/there"
-        # The latter is fine filename on Windows, which
-        # will transform to dir/file on back transform.
-        #
-        # Yes, this is a BUG, as long as we permit "/" in
-        # filenames on Windows and "\" on Unix
-
-        key = fs.path.relparts(file_fs_path, fs_path)
-        assert key
-        tree.add(key, meta, obj.hash_info)
-
-        tree_meta.size += meta.size or 0
-        tree_meta.nfiles += 1
-
-    return tree_meta, tree
-
-
-def _stage_tree(fs_path, fs, fs_info, name, odb=None, **kwargs):
-    from .tree import Tree
-
-    value = fs_info.get(name)
-    if odb and value:
-        hash_info = HashInfo(name, value)
-        try:
-            tree = Tree.load(odb, hash_info)
-            return Meta(nfiles=len(tree)), tree
-        except FileNotFoundError:
-            pass
-
-    meta, tree = _build_tree(fs_path, fs, name, odb=odb, **kwargs)
-    state = odb.state if odb and odb.state else None
-    hash_info = None
-    if state:
-        _, hash_info = state.get(  # pylint: disable=assignment-from-none
-            fs_path, fs
-        )
-    tree.digest(hash_info=hash_info)
-    odb.add(tree.fs_path, tree.fs, tree.hash_info, hardlink=False)
-    raw = odb.get(tree.hash_info)
-    # cleanup unneeded memfs tmpfile and return tree based on the
-    # ODB fs/path
-    if odb.fs != tree.fs:
-        tree.fs.remove(tree.fs_path)
-    tree.fs = raw.fs
-    tree.fs_path = raw.fs_path
-    return meta, tree
-
-
-_url_cache: Dict[str, str] = {}
-
-
-def _make_staging_url(
-    fs: "FileSystem", odb: "ObjectDB", fs_path: Optional[str]
-):
-    from dvc.scheme import Schemes
-
-    url = f"{Schemes.MEMORY}://{_STAGING_MEMFS_PATH}"
-
-    if fs_path is not None:
-        if odb.fs.scheme == Schemes.LOCAL:
-            fs_path = os.path.abspath(fs_path)
-
-        if fs_path not in _url_cache:
-            _url_cache[fs_path] = hashlib.sha256(
-                fs_path.encode("utf-8")
-            ).hexdigest()
-
-        url = fs.path.join(url, _url_cache[fs_path])
-
-    return url
-
-
-def _get_staging(odb: "ObjectDB") -> "ObjectDB":
-    """Return an ODB that can be used for staging objects.
-
-    Staging will be a reference ODB stored in the the global memfs.
-    """
-
-    from dvc.fs.memory import MemoryFileSystem
-
-    fs = MemoryFileSystem()
-    fs_path = _make_staging_url(fs, odb, odb.fs_path)
-    state = odb.state
-    return ReferenceObjectDB(fs, fs_path, state=state)
-
-
-def _load_from_state(odb, staging, fs_path, fs, name):
-    from dvc.objects.errors import ObjectFormatError
-
-    from . import check, load
-    from .tree import Tree
-
-    state = odb.state
-    meta, hash_info = state.get(fs_path, fs)
-    if hash_info:
-        for odb_ in (odb, staging):
-            if odb_.exists(hash_info):
-                try:
-                    obj = load(odb_, hash_info)
-                    check(odb_, obj, check_hash=False)
-                    if isinstance(obj, Tree):
-                        meta.nfiles = len(obj)
-                    assert obj.hash_info.name == name
-                    return odb_, meta, obj
-                except (ObjectFormatError, FileNotFoundError):
-                    pass
-    raise FileNotFoundError
-
-
-def _stage_external_tree_info(odb, tree, name):
-    # NOTE: used only for external outputs. Initial reasoning was to be
-    # able to validate .dir files right in the workspace (e.g. check s3
-    # etag), but could be dropped for manual validation with regular md5,
-    # that would be universal for all clouds.
-    assert odb and name != "md5"
-
-    odb.add(tree.fs_path, tree.fs, tree.hash_info)
-    raw = odb.get(tree.hash_info)
-    _, hash_info = get_file_hash(raw.fs_path, raw.fs, name, state=odb.state)
-    tree.fs_path = raw.fs_path
-    tree.fs = raw.fs
-    tree.hash_info.name = hash_info.name
-    tree.hash_info.value = hash_info.value
-    if not tree.hash_info.value.endswith(".dir"):
-        tree.hash_info.value += ".dir"
-    return tree
-
-
-def stage(
-    odb: "ObjectDB",
-    fs_path: "AnyFSPath",
-    fs: "FileSystem",
-    name: str,
-    upload: bool = False,
-    dry_run: bool = False,
-    **kwargs,
-) -> Tuple["ObjectDB", "Meta", "HashFile"]:
-    """Stage (prepare) objects from the given path for addition to an ODB.
-
-    Returns at tuple of (staging_odb, object) where addition to the ODB can
-    be completed by transferring the object from staging to the dest ODB.
-
-    If dry_run is True, object hashes will be computed and returned, but file
-    objects themselves will not be added to the staging ODB (i.e. the resulting
-    file objects cannot transferred from staging to another ODB).
-
-    If upload is True, files will be uploaded to a temporary path on the dest
-    ODB filesystem, and staged objects will reference the uploaded path rather
-    than the original source path.
-    """
-    assert fs_path
-    # assert scheme(fs_path) == fs.scheme
-
-    details = fs.info(fs_path)
-    staging = _get_staging(odb)
-    if odb:
-        try:
-            return _load_from_state(odb, staging, fs_path, fs, name)
-        except FileNotFoundError:
-            pass
-
-    if details["type"] == "directory":
-        meta, obj = _stage_tree(
-            fs_path,
-            fs,
-            details,
-            name,
-            odb=staging,
-            upload_odb=odb if upload else None,
-            dry_run=dry_run,
-            **kwargs,
-        )
-        logger.debug("staged tree '%s'", obj)
-        if name != "md5":
-            obj = _stage_external_tree_info(odb, obj, name)
-    else:
-        _, meta, obj = _stage_file(
-            fs_path,
-            fs,
-            name,
-            odb=staging,
-            upload_odb=odb if upload else None,
-            dry_run=dry_run,
-        )
-
-    if odb and odb.state and obj.hash_info:
-        odb.state.save(fs_path, fs, obj.hash_info)
-
-    return staging, meta, obj

dvc/data/status.py

@@ -1,207 +0,0 @@
-import logging
-from typing import TYPE_CHECKING, Dict, Iterable, NamedTuple, Optional, Set
-
-from dvc.hash_info import HashInfo
-from dvc.scheme import Schemes
-
-from .tree import Tree
-
-if TYPE_CHECKING:
-    from dvc.objects.db import ObjectDB
-    from dvc.objects.file import HashFile
-
-    from .db.index import ObjectDBIndexBase
-
-logger = logging.getLogger(__name__)
-
-
-class StatusResult(NamedTuple):
-    exists: Set["HashInfo"]
-    missing: Set["HashInfo"]
-
-
-class CompareStatusResult(NamedTuple):
-    ok: Set["HashInfo"]
-    missing: Set["HashInfo"]
-    new: Set["HashInfo"]
-    deleted: Set["HashInfo"]
-
-
-def _indexed_dir_hashes(odb, index, dir_objs, name, cache_odb, jobs=None):
-    # Validate our index by verifying all indexed .dir hashes
-    # still exist on the remote
-    from ._progress import QueryingProgress
-
-    dir_hashes = set(dir_objs.keys())
-    indexed_dirs = set(index.dir_hashes())
-    indexed_dir_exists = set()
-    if indexed_dirs:
-        hashes = QueryingProgress(
-            odb.list_hashes_exists(indexed_dirs, jobs=jobs),
-            total=len(indexed_dirs),
-        )
-        indexed_dir_exists.update(hashes)
-        missing_dirs = indexed_dirs.difference(indexed_dir_exists)
-        if missing_dirs:
-            logger.debug(
-                "Remote cache missing indexed .dir hashes '%s', "
-                "clearing remote index",
-                ", ".join(missing_dirs),
-            )
-            index.clear()
-
-    # Check if non-indexed (new) dir hashes exist on remote
-    dir_exists = dir_hashes.intersection(indexed_dir_exists)
-    dir_missing = dir_hashes - dir_exists
-    dir_exists.update(
-        QueryingProgress(
-            odb.list_hashes_exists(dir_missing, jobs=jobs),
-            total=len(dir_missing),
-        )
-    )
-
-    # If .dir hash exists in the ODB, assume directory contents
-    # also exists
-    for dir_hash in dir_exists:
-        tree = dir_objs.get(dir_hash)
-        if not tree:
-            try:
-                tree = Tree.load(cache_odb, HashInfo(name, dir_hash))
-            except FileNotFoundError:
-                continue
-        file_hashes = [oid.value for _, _, oid in tree]
-        if dir_hash not in index:
-            logger.debug(
-                "Indexing new .dir '%s' with '%s' nested files",
-                dir_hash,
-                len(file_hashes),
-            )
-            index.update([dir_hash], file_hashes)
-        yield from file_hashes
-        yield tree.hash_info.value
-
-
-def status(
-    odb: "ObjectDB",
-    obj_ids: Iterable["HashInfo"],
-    name: Optional[str] = None,
-    index: Optional["ObjectDBIndexBase"] = None,
-    cache_odb: Optional["ObjectDB"] = None,
-    shallow: bool = True,
-    jobs: Optional[int] = None,
-    **kwargs,
-) -> "StatusResult":
-    """Return status of whether or not the specified objects exist odb.
-
-    If cache_odb is set, trees will be loaded from cache_odb instead of odb
-    when needed.
-
-    Status is returned as a tuple of:
-        exists: objs that exist in odb
-        missing: objs that do not exist in ODB
-    """
-    logger.debug("Preparing to collect status from '%s'", odb.fs_path)
-    if not name:
-        name = odb.fs.PARAM_CHECKSUM
-
-    if cache_odb is None:
-        cache_odb = odb
-
-    hash_infos: Dict[str, "HashInfo"] = {}
-    dir_objs: Dict[str, Optional["HashFile"]] = {}
-    for hash_info in obj_ids:
-        assert hash_info.value
-        if hash_info.isdir:
-            if shallow:
-                tree = None
-            else:
-                tree = Tree.load(cache_odb, hash_info)
-                for _, _, oid in tree:
-                    assert oid and oid.value
-                    hash_infos[oid.value] = oid
-            if index:
-                dir_objs[hash_info.value] = tree
-        hash_infos[hash_info.value] = hash_info
-
-    if odb.fs.scheme == Schemes.MEMORY:
-        # assume memfs staged objects already exist
-        return StatusResult(set(hash_infos.values()), set())
-
-    hashes: Set[str] = set(hash_infos.keys())
-    exists: Set[str] = set()
-
-    logger.debug("Collecting status from '%s'", odb.fs_path)
-    if index and hashes:
-        if dir_objs:
-            exists = hashes.intersection(
-                _indexed_dir_hashes(
-                    odb, index, dir_objs, name, cache_odb, jobs=jobs
-                )
-            )
-            hashes.difference_update(exists)
-        if hashes:
-            exists.update(index.intersection(hashes))
-            hashes.difference_update(exists)
-
-    if hashes:
-        from ._progress import QueryingProgress
-
-        with QueryingProgress(phase="Checking", name=odb.fs_path) as pbar:
-            exists.update(
-                odb.hashes_exist(hashes, jobs=jobs, progress=pbar.callback)
-            )
-    return StatusResult(
-        {hash_infos[hash_] for hash_ in exists},
-        {hash_infos[hash_] for hash_ in (hashes - exists)},
-    )
-
-
-def compare_status(
-    src: "ObjectDB",
-    dest: "ObjectDB",
-    obj_ids: Iterable["HashInfo"],
-    log_missing: bool = True,
-    check_deleted: bool = True,
-    src_index: Optional["ObjectDBIndexBase"] = None,
-    dest_index: Optional["ObjectDBIndexBase"] = None,
-    jobs: Optional[int] = None,
-    **kwargs,
-) -> "CompareStatusResult":
-    """Compare status for the specified objects between two ODBs.
-
-    Status is returned as a tuple of:
-        ok: hashes that exist in both src and dest
-        missing: hashes that do not exist in neither src nor dest
-        new: hashes that only exist in src
-        deleted: hashes that only exist in dest
-    """
-    if "cache_odb" not in kwargs:
-        kwargs["cache_odb"] = src
-    dest_exists, dest_missing = status(
-        dest, obj_ids, index=dest_index, jobs=jobs, **kwargs
-    )
-    # for transfer operations we can skip src status check when all objects
-    # already exist in dest
-    if dest_missing or check_deleted:
-        src_exists, src_missing = status(
-            src, obj_ids, index=src_index, jobs=jobs, **kwargs
-        )
-    else:
-        src_exists = dest_exists
-        src_missing = set()
-    result = CompareStatusResult(
-        src_exists & dest_exists,
-        src_missing & dest_missing,
-        src_exists - dest_exists,
-        dest_exists - src_exists,
-    )
-    if log_missing and result.missing:
-        missing_desc = "\n".join(
-            f"name: {hash_info.obj_name}, {hash_info}"
-            for hash_info in result.missing
-        )
-        logger.warning(
-            "Some of the cache files do not exist neither locally "
-            f"nor on remote. Missing cache files:\n{missing_desc}"
-        )
-    return result

dvc/data/transfer.py

@@ -1,179 +0,0 @@
-import errno
-import logging
-from functools import partial, wraps
-from typing import TYPE_CHECKING, Any, Callable, Iterable, Optional
-
-from funcy import split
-
-from dvc.progress import Tqdm
-from dvc.utils.threadpool import ThreadPoolExecutor
-
-if TYPE_CHECKING:
-    from dvc.hash_info import HashInfo
-    from dvc.objects.db import ObjectDB
-
-    from .db.index import ObjectDBIndexBase
-    from .tree import Tree
-
-logger = logging.getLogger(__name__)
-
-
-def _log_exceptions(func):
-    @wraps(func)
-    def wrapper(fs_path, *args, **kwargs):
-        try:
-            func(fs_path, *args, **kwargs)
-            return 0
-        except Exception as exc:  # pylint: disable=broad-except
-            # NOTE: this means we ran out of file descriptors and there is no
-            # reason to try to proceed, as we will hit this error anyways.
-            # pylint: disable=no-member
-            if isinstance(exc, OSError) and exc.errno == errno.EMFILE:
-                raise
-
-            logger.exception("failed to transfer '%s'", fs_path)
-            return 1
-
-    return wrapper
-
-
-def find_tree_by_obj_id(
-    odbs: Iterable[Optional["ObjectDB"]], obj_id: "HashInfo"
-) -> Optional["Tree"]:
-    from dvc.objects.errors import ObjectFormatError
-
-    from .tree import Tree
-
-    for odb in odbs:
-        if odb is not None:
-            try:
-                return Tree.load(odb, obj_id)
-            except (FileNotFoundError, ObjectFormatError):
-                pass
-    return None
-
-
-def _do_transfer(
-    src: "ObjectDB",
-    dest: "ObjectDB",
-    obj_ids: Iterable["HashInfo"],
-    missing_ids: Iterable["HashInfo"],
-    processor: Callable,
-    src_index: Optional["ObjectDBIndexBase"] = None,
-    dest_index: Optional["ObjectDBIndexBase"] = None,
-    cache_odb: Optional["ObjectDB"] = None,
-    **kwargs: Any,
-):
-    from dvc.exceptions import FileTransferError
-
-    dir_ids, file_ids = split(lambda hash_info: hash_info.isdir, obj_ids)
-    total_fails = 0
-    succeeded_dir_objs = []
-    all_file_ids = set(file_ids)
-
-    for dir_hash in dir_ids:
-        dir_obj = find_tree_by_obj_id([cache_odb, src], dir_hash)
-        assert dir_obj
-
-        entry_ids = {oid for _, _, oid in dir_obj}
-        bound_file_ids = all_file_ids & entry_ids
-        all_file_ids -= entry_ids
-
-        dir_fails = sum(processor(bound_file_ids))
-        if dir_fails:
-            logger.debug(
-                "failed to upload full contents of '%s', "
-                "aborting .dir file upload",
-                dir_obj.name,
-            )
-            logger.error(
-                "failed to upload '%s' to '%s'",
-                src.get(dir_obj.hash_info).fs_path,
-                dest.get(dir_obj.hash_info).fs_path,
-            )
-            total_fails += dir_fails + 1
-        elif entry_ids.intersection(missing_ids):
-            # if for some reason a file contained in this dir is
-            # missing both locally and in the remote, we want to
-            # push whatever file content we have, but should not
-            # push .dir file
-            logger.debug(
-                "directory '%s' contains missing files,"
-                "skipping .dir file upload",
-                dir_obj.name,
-            )
-        else:
-            is_dir_failed = sum(processor([dir_obj.hash_info]))
-            total_fails += is_dir_failed
-            if not is_dir_failed:
-                succeeded_dir_objs.append(dir_obj)
-
-    # insert the rest
-    total_fails += sum(processor(all_file_ids))
-    if total_fails:
-        if src_index:
-            src_index.clear()
-        raise FileTransferError(total_fails)
-
-    # index successfully pushed dirs
-    if dest_index:
-        for dir_obj in succeeded_dir_objs:
-            file_hashes = {oid.value for _, _, oid in dir_obj}
-            logger.debug(
-                "Indexing pushed dir '%s' with '%s' nested files",
-                dir_obj.hash_info,
-                len(file_hashes),
-            )
-            assert dir_obj.hash_info and dir_obj.hash_info.value
-            dest_index.update([dir_obj.hash_info.value], file_hashes)
-
-
-def transfer(
-    src: "ObjectDB",
-    dest: "ObjectDB",
-    obj_ids: Iterable["HashInfo"],
-    jobs: Optional[int] = None,
-    verify: bool = False,
-    hardlink: bool = False,
-    **kwargs,
-) -> int:
-    """Transfer (copy) the specified objects from one ODB to another.
-
-    Returns the number of successfully transferred objects
-    """
-    from .status import compare_status
-
-    logger.debug(
-        "Preparing to transfer data from '%s' to '%s'",
-        src.fs_path,
-        dest.fs_path,
-    )
-    if src == dest:
-        return 0
-
-    status = compare_status(
-        src, dest, obj_ids, check_deleted=False, jobs=jobs, **kwargs
-    )
-    if not status.new:
-        return 0
-
-    def func(hash_info: "HashInfo") -> None:
-        obj = src.get(hash_info)
-        return dest.add(
-            obj.fs_path,
-            obj.fs,
-            obj.hash_info,
-            verify=verify,
-            hardlink=hardlink,
-        )
-
-    total = len(status.new)
-    jobs = jobs or dest.fs.jobs
-    with Tqdm(total=total, unit="file", desc="Transferring") as pbar:
-        with ThreadPoolExecutor(max_workers=jobs) as executor:
-            wrapped_func = pbar.wrap_fn(_log_exceptions(func))
-            processor = partial(executor.imap_unordered, wrapped_func)
-            _do_transfer(
-                src, dest, status.new, status.missing, processor, **kwargs
-            )
-    return total

dvc/data/tree.py

@@ -1,293 +0,0 @@
-import json
-import logging
-import posixpath
-from typing import TYPE_CHECKING, Dict, Iterable, Optional, Tuple
-
-from funcy import cached_property
-
-from dvc.objects.errors import ObjectFormatError
-from dvc.objects.file import HashFile
-
-from .stage import get_file_hash
-
-if TYPE_CHECKING:
-    from dvc.hash_info import HashInfo
-    from dvc.objects.db import ObjectDB
-
-    from .meta import Meta
-
-logger = logging.getLogger(__name__)
-
-
-class TreeError(Exception):
-    pass
-
-
-def _try_load(
-    odbs: Iterable["ObjectDB"],
-    hash_info: "HashInfo",
-) -> Optional["HashFile"]:
-    for odb in odbs:
-        if not odb:
-            continue
-
-        try:
-            return Tree.load(odb, hash_info)
-        except (FileNotFoundError, ObjectFormatError):
-            pass
-
-    return None
-
-
-class Tree(HashFile):
-    PARAM_RELPATH = "relpath"
-
-    def __init__(self, *args, **kwargs):
-        super().__init__(*args, **kwargs)
-        self._dict: Dict[Tuple[str], Tuple["Meta", "HashInfo"]] = {}
-
-    @cached_property
-    def _trie(self):
-        from pygtrie import Trie
-
-        return Trie(self._dict)
-
-    def add(self, key: Tuple[str], meta: "Meta", oid: "HashInfo"):
-        self.__dict__.pop("trie", None)
-        self._dict[key] = (meta, oid)
-
-    def digest(self, hash_info: Optional["HashInfo"] = None):
-        from dvc.fs.memory import MemoryFileSystem
-        from dvc.utils import tmp_fname
-
-        memfs = MemoryFileSystem()
-        fs_path = "memory://{}".format(tmp_fname(""))
-        memfs.pipe_file(fs_path, self.as_bytes())
-        self.fs = memfs
-        self.fs_path = fs_path
-        if hash_info:
-            self.hash_info = hash_info
-        else:
-            _, self.hash_info = get_file_hash(fs_path, memfs, "md5")
-            assert self.hash_info.value
-            self.hash_info.value += ".dir"
-
-    def _load(self, key, meta, hash_info):
-        if hash_info and hash_info.isdir and not meta.obj:
-            meta.obj = _try_load([meta.odb, meta.remote], hash_info)
-            if meta.obj:
-                for ikey, value in meta.obj.iteritems():
-                    self._trie[key + ikey] = value
-                    self._dict[key + ikey] = value
-
-    def iteritems(self, prefix=None):
-        kwargs = {}
-        if prefix:
-            kwargs = {"prefix": prefix}
-            item = self._trie.longest_prefix(prefix)
-            if item:
-                key, (meta, hash_info) = item
-                self._load(key, meta, hash_info)
-
-        for key, (meta, hash_info) in self._trie.iteritems(**kwargs):
-            self._load(key, meta, hash_info)
-            yield key, (meta, hash_info)
-
-    def shortest_prefix(self, *args, **kwargs):
-        return self._trie.shortest_prefix(*args, **kwargs)
-
-    def __len__(self):
-        return len(self._dict)
-
-    def __iter__(self):
-        yield from (
-            (key, value[0], value[1]) for key, value in self._dict.items()
-        )
-
-    def as_dict(self):
-        return self._dict.copy()
-
-    def as_list(self):
-        from operator import itemgetter
-
-        # Sorting the list by path to ensure reproducibility
-        return sorted(
-            (
-                {
-                    # NOTE: not using hash_info.to_dict() because we don't want
-                    # size/nfiles fields at this point.
-                    oid.name: oid.value,
-                    self.PARAM_RELPATH: posixpath.sep.join(parts),
-                }
-                for parts, _, oid in self  # noqa: B301
-            ),
-            key=itemgetter(self.PARAM_RELPATH),
-        )
-
-    def as_bytes(self):
-        return json.dumps(self.as_list(), sort_keys=True).encode("utf-8")
-
-    @classmethod
-    def from_list(cls, lst):
-        from dvc.hash_info import HashInfo
-
-        tree = cls(None, None, None)
-        for _entry in lst:
-            entry = _entry.copy()
-            relpath = entry.pop(cls.PARAM_RELPATH)
-            parts = tuple(relpath.split(posixpath.sep))
-            hash_info = HashInfo.from_dict(entry)
-            tree.add(parts, None, hash_info)
-        return tree
-
-    @classmethod
-    def load(cls, odb, hash_info):
-        obj = odb.get(hash_info)
-
-        try:
-            with obj.fs.open(obj.fs_path, "r") as fobj:
-                raw = json.load(fobj)
-        except ValueError as exc:
-            raise ObjectFormatError(f"{obj} is corrupted") from exc
-
-        if not isinstance(raw, list):
-            logger.error(
-                "dir cache file format error '%s' [skipping the file]",
-                obj.fs_path,
-            )
-            raise ObjectFormatError(f"{obj} is corrupted")
-
-        tree = cls.from_list(raw)
-        tree.fs_path = obj.fs_path
-        tree.fs = obj.fs
-        tree.hash_info = hash_info
-
-        return tree
-
-    def filter(self, prefix: Tuple[str]) -> Optional["Tree"]:
-        """Return a filtered copy of this tree that only contains entries
-        inside prefix.
-
-        The returned tree will contain the original tree's hash_info and
-        fs_path.
-
-        Returns an empty tree if no object exists at the specified prefix.
-        """
-        tree = Tree(self.fs_path, self.fs, self.hash_info)
-        try:
-            for key, (meta, oid) in self._trie.items(prefix):
-                tree.add(key, meta, oid)
-        except KeyError:
-            pass
-        return tree
-
-    def get(self, odb, prefix: Tuple[str]) -> Optional[HashFile]:
-        """Return object at the specified prefix in this tree.
-
-        Returns None if no object exists at the specified prefix.
-        """
-        _, oid = self._dict.get(prefix) or (None, None)
-        if oid:
-            return odb.get(oid)
-
-        tree = Tree(None, None, None)
-        depth = len(prefix)
-        try:
-            for key, (meta, entry_oid) in self._trie.items(prefix):
-                tree.add(key[depth:], meta, entry_oid)
-        except KeyError:
-            return None
-        tree.digest()
-        return tree
-
-    def ls(self, prefix=None):
-        kwargs = {}
-        if prefix:
-            kwargs["prefix"] = prefix
-
-        meta, hash_info = self._trie.get(prefix, (None, None))
-        if hash_info and hash_info.isdir and meta and not meta.obj:
-            raise TreeError
-
-        ret = []
-
-        def node_factory(_, key, children, *args):
-            if key == prefix:
-                list(children)
-            else:
-                ret.append(key[-1])
-
-        self._trie.traverse(node_factory, **kwargs)
-
-        return ret
-
-
-def du(odb, tree):
-    try:
-        return sum(
-            odb.fs.size(odb.hash_to_path(oid.value)) for _, _, oid in tree
-        )
-    except FileNotFoundError:
-        return None
-
-
-def _diff(ancestor, other, allow_removed=False):
-    from dictdiffer import diff
-
-    from dvc.exceptions import MergeError
-
-    allowed = ["add"]
-    if allow_removed:
-        allowed.append("remove")
-
-    result = list(diff(ancestor, other))
-    for typ, _, _ in result:
-        if typ not in allowed:
-            raise MergeError(
-                "unable to auto-merge directories with diff that contains "
-                f"'{typ}'ed files"
-            )
-    return result
-
-
-def _merge(ancestor, our, their):
-    import copy
-
-    from dictdiffer import patch
-
-    our_diff = _diff(ancestor, our)
-    if not our_diff:
-        return copy.deepcopy(their)
-
-    their_diff = _diff(ancestor, their)
-    if not their_diff:
-        return copy.deepcopy(our)
-
-    # make sure there are no conflicting files
-    _diff(our, their, allow_removed=True)
-
-    return patch(our_diff + their_diff, ancestor)
-
-
-def merge(odb, ancestor_info, our_info, their_info):
-    from . import load
-
-    assert our_info
-    assert their_info
-
-    if ancestor_info:
-        ancestor = load(odb, ancestor_info)
-    else:
-        ancestor = Tree(None, None, None)
-
-    our = load(odb, our_info)
-    their = load(odb, their_info)
-
-    merged_dict = _merge(ancestor.as_dict(), our.as_dict(), their.as_dict())
-
-    merged = Tree(None, None, None)
-    for key, (meta, oid) in merged_dict.items():
-        merged.add(key, meta, oid)
-    merged.digest()
-
-    return merged

dvc/data_cloud.py

@@ -3,15 +3,38 @@
 import logging
 from typing import TYPE_CHECKING, Iterable, Optional
 
-from dvc.data.db import get_index
+from funcy import cached_property
+
+from dvc_data.hashfile.db import get_index
 
 if TYPE_CHECKING:
-    from dvc.hash_info import HashInfo
-    from dvc.objects.db import ObjectDB
+    from dvc_data.hashfile.db import HashFileDB
+    from dvc_data.hashfile.hash_info import HashInfo
+    from dvc_data.hashfile.status import CompareStatusResult
+    from dvc_data.hashfile.transfer import TransferResult
 
 logger = logging.getLogger(__name__)
 
 
+class Remote:
+    def __init__(self, path, fs, **config):
+        self.path = path
+        self.fs = fs
+
+        self.worktree = config.pop("worktree", False)
+        self.config = config
+
+    @cached_property
+    def odb(self):
+        from dvc_data.hashfile.db import get_odb
+
+        path = self.path
+        if self.worktree:
+            path = self.fs.path.join(path, ".dvc", "cache")
+
+        return get_odb(self.fs, path, **self.config)
+
+
 class DataCloud:
     """Class that manages dvc remotes.
 
@@ -26,18 +49,23 @@ class DataCloud:
     def __init__(self, repo):
         self.repo = repo
 
-    def get_remote_odb(
+    def get_remote(
         self,
         name: Optional[str] = None,
         command: str = "<command>",
-    ) -> "ObjectDB":
+    ) -> "Remote":
         from dvc.config import NoRemoteError
 
         if not name:
             name = self.repo.config["core"].get("remote")
 
         if name:
-            return self._init_odb(name)
+            from dvc.fs import get_cloud_fs
+
+            cls, config, fs_path = get_cloud_fs(self.repo, name=name)
+            fs = cls(**config)
+            config["tmp_dir"] = self.repo.index_db_dir
+            return Remote(fs_path, fs, **config)
 
         if bool(self.repo.config["remote"]):
             error_msg = (
@@ -54,21 +82,43 @@ class DataCloud:
 
         raise NoRemoteError(error_msg)
 
-    def _init_odb(self, name):
-        from dvc.data.db import get_odb
-        from dvc.fs import get_cloud_fs
+    def get_remote_odb(
+        self,
+        name: Optional[str] = None,
+        command: str = "<command>",
+    ) -> "HashFileDB":
+        remote = self.get_remote(name=name, command=command)
+        return remote.odb
 
-        cls, config, fs_path = get_cloud_fs(self.repo, name=name)
-        config["tmp_dir"] = self.repo.index_db_dir
-        return get_odb(cls(**config), fs_path, **config)
+    def _log_missing(self, status: "CompareStatusResult"):
+        if status.missing:
+            missing_desc = "\n".join(
+                f"name: {hash_info.obj_name}, {hash_info}"
+                for hash_info in status.missing
+            )
+            logger.warning(
+                "Some of the cache files do not exist neither locally "
+                f"nor on remote. Missing cache files:\n{missing_desc}"
+            )
+
+    def transfer(
+        self,
+        src_odb: "HashFileDB",
+        dest_odb: "HashFileDB",
+        objs: Iterable["HashInfo"],
+        **kwargs,
+    ) -> "TransferResult":
+        from dvc_data.hashfile.transfer import transfer
+
+        return transfer(src_odb, dest_odb, objs, **kwargs)
 
     def push(
         self,
         objs: Iterable["HashInfo"],
         jobs: Optional[int] = None,
         remote: Optional[str] = None,
-        odb: Optional["ObjectDB"] = None,
-    ):
+        odb: Optional["HashFileDB"] = None,
+    ) -> "TransferResult":
         """Push data items in a cloud-agnostic way.
 
         Args:
@@ -78,17 +128,15 @@ class DataCloud:
                 By default remote from core.remote config option is used.
             odb: optional ODB to push to. Overrides remote.
         """
-        from dvc.data.transfer import transfer
-
-        if not odb:
-            odb = self.get_remote_odb(remote, "push")
-        return transfer(
+        odb = odb or self.get_remote_odb(remote, "push")
+        return self.transfer(
             self.repo.odb.local,
             odb,
             objs,
             jobs=jobs,
             dest_index=get_index(odb),
             cache_odb=self.repo.odb.local,
+            validate_status=self._log_missing,
         )
 
     def pull(
@@ -96,8 +144,8 @@ class DataCloud:
         objs: Iterable["HashInfo"],
         jobs: Optional[int] = None,
         remote: Optional[str] = None,
-        odb: Optional["ObjectDB"] = None,
-    ):
+        odb: Optional["HashFileDB"] = None,
+    ) -> "TransferResult":
         """Pull data items in a cloud-agnostic way.
 
         Args:
@@ -107,11 +155,8 @@ class DataCloud:
                 By default remote from core.remote config option is used.
             odb: optional ODB to pull from. Overrides remote.
         """
-        from dvc.data.transfer import transfer
-
-        if not odb:
-            odb = self.get_remote_odb(remote, "pull")
-        return transfer(
+        odb = odb or self.get_remote_odb(remote, "pull")
+        return self.transfer(
             odb,
             self.repo.odb.local,
             objs,
@@ -119,6 +164,7 @@ class DataCloud:
             src_index=get_index(odb),
             cache_odb=self.repo.odb.local,
             verify=odb.verify,
+            validate_status=self._log_missing,
         )
 
     def status(
@@ -126,7 +172,7 @@ class DataCloud:
         objs: Iterable["HashInfo"],
         jobs: Optional[int] = None,
         remote: Optional[str] = None,
-        odb: Optional["ObjectDB"] = None,
+        odb: Optional["HashFileDB"] = None,
         log_missing: bool = True,
     ):
         """Check status of data items in a cloud-agnostic way.
@@ -141,7 +187,7 @@ class DataCloud:
             log_missing: log warning messages if file doesn't exist
                 neither in cache, neither in cloud.
         """
-        from dvc.data.status import compare_status
+        from dvc_data.hashfile.status import compare_status
 
         if not odb:
             odb = self.get_remote_odb(remote, "status")
@@ -157,5 +203,5 @@ class DataCloud:
 
     def get_url_for(self, remote, checksum):
         odb = self.get_remote_odb(remote)
-        path = odb.hash_to_path(checksum)
+        path = odb.oid_to_path(checksum)
         return odb.fs.unstrip_protocol(path)

dvc/dependency/__init__.py

@@ -1,7 +1,7 @@
 from collections import defaultdict
 from typing import Any, Mapping
 
-from dvc.output import ARTIFACT_SCHEMA, Output
+from dvc.output import ARTIFACT_SCHEMA, DIR_FILES_SCHEMA, Output
 
 from .base import Dependency
 from .param import ParamsDependency
@@ -15,10 +15,11 @@ SCHEMA: Mapping[str, Any] = {
     **ARTIFACT_SCHEMA,
     **RepoDependency.REPO_SCHEMA,
     **ParamsDependency.PARAM_SCHEMA,
+    Output.PARAM_FILES: [DIR_FILES_SCHEMA],
 }
 
 
-def _get(stage, p, info):
+def _get(stage, p, info, **kwargs):
     if info and info.get(RepoDependency.PARAM_REPO):
         repo = info.pop(RepoDependency.PARAM_REPO)
         return RepoDependency(repo, stage, p, info)
@@ -27,21 +28,30 @@ def _get(stage, p, info):
         params = info.pop(ParamsDependency.PARAM_PARAMS)
         return ParamsDependency(stage, p, params)
 
-    return Dependency(stage, p, info)
+    return Dependency(stage, p, info, **kwargs)
 
 
 def loadd_from(stage, d_list):
     ret = []
     for d in d_list:
         p = d.pop(Output.PARAM_PATH, None)
-        ret.append(_get(stage, p, d))
+        files = d.pop(Output.PARAM_FILES, None)
+        ret.append(_get(stage, p, d, files=files))
     return ret
 
 
-def loads_from(stage, s_list, erepo=None):
+def loads_from(stage, s_list, erepo=None, fs_config=None):
     assert isinstance(s_list, list)
     info = {RepoDependency.PARAM_REPO: erepo} if erepo else {}
-    return [_get(stage, s, info.copy()) for s in s_list]
+    return [
+        _get(
+            stage,
+            s,
+            info.copy(),
+            fs_config=fs_config,
+        )
+        for s in s_list
+    ]
 
 
 def _merge_params(s_list):

dvc/dependency/base.py

@@ -1,6 +1,7 @@
-from typing import Type
+from typing import Dict, Type
 
 from dvc.exceptions import DvcException
+from dvc.fs import download as fs_download
 from dvc.output import Output
 
 
@@ -30,5 +31,39 @@ class Dependency(Output):
     )  # type: Type[DvcException]
     IsStageFileError = DependencyIsStageFileError  # type: Type[DvcException]
 
+    def workspace_status(self) -> Dict[str, str]:
+        if self.fs.version_aware:
+            old_fs_path = self.fs_path
+            try:
+                self.fs_path = self.fs.path.version_path(self.fs_path, None)
+                if self.changed_meta():
+                    return {str(self): "update available"}
+            finally:
+                self.fs_path = old_fs_path
+        return super().workspace_status()
+
     def update(self, rev=None):
-        pass
+        if self.fs.version_aware:
+            self.fs_path = self.fs.path.version_path(self.fs_path, rev)
+            self.meta = self.get_meta()
+            self.def_path = self.fs.path.version_path(
+                self.def_path, self.meta.version_id
+            )
+            self.fs_path = self.fs.path.version_path(
+                self.fs_path, self.meta.version_id
+            )
+
+    def download(self, to, jobs=None):
+        fs_download(self.fs, self.fs_path, to, jobs=jobs)
+
+    def save(self):
+        super().save()
+        if self.fs.version_aware:
+            self.fs_path = self.fs.path.version_path(
+                self.fs_path, self.meta.version_id
+            )
+
+    def dumpd(self, **kwargs):
+        if self.fs.version_aware:
+            kwargs["with_files"] = True
+        return super().dumpd(**kwargs)

dvc/dependency/param.py

@@ -8,8 +8,8 @@ import dpath.util
 from voluptuous import Any
 
 from dvc.exceptions import DvcException
-from dvc.hash_info import HashInfo
-from dvc.utils.serialize import LOADERS, ParseError
+from dvc.utils.serialize import ParseError, load_path
+from dvc_data.hashfile.hash_info import HashInfo
 
 from .base import Dependency
 
@@ -39,14 +39,15 @@ class ParamsDependency(Dependency):
 
     def __init__(self, stage, path, params=None, repo=None):
         self.params = list(params) if params else []
-        info = (
-            {self.PARAM_PARAMS: params} if isinstance(params, dict) else None
-        )
+        hash_info = HashInfo()
+        if isinstance(params, dict):
+            hash_info = HashInfo(self.PARAM_PARAMS, params)
         repo = repo or stage.repo
         path = path or os.path.join(repo.root_dir, self.DEFAULT_PARAMS_FILE)
-        super().__init__(stage, path, info=info, repo=repo)
+        super().__init__(stage, path, repo=repo)
+        self.hash_info = hash_info
 
-    def dumpd(self):
+    def dumpd(self, **kwargs):
         ret = super().dumpd()
         if not self.hash_info:
             ret[self.PARAM_PARAMS] = self.params or {}
@@ -139,12 +140,9 @@ class ParamsDependency(Dependency):
             )
 
     def read_file(self):
-        _, ext = os.path.splitext(self.fs_path)
-        loader = LOADERS[ext]
-
         self.validate_filepath()
         try:
-            return loader(self.fs_path, fs=self.repo.fs)
+            return load_path(self.fs_path, self.repo.fs)
         except ParseError as exc:
             raise BadParamFileError(
                 f"Unable to read parameters from '{self}'"

dvc/dependency/repo.py

@@ -5,12 +5,15 @@ from typing import TYPE_CHECKING, Dict, Optional, Set, Tuple
 
 from voluptuous import Required
 
+from dvc.prompt import confirm
+from dvc_data.hashfile.meta import Meta
+
 from .base import Dependency
 
 if TYPE_CHECKING:
-    from dvc.hash_info import HashInfo
-    from dvc.objects.db import ObjectDB
-    from dvc.objects.file import HashFile
+    from dvc_data.hashfile.file import HashFile
+    from dvc_data.hashfile.hash_info import HashInfo
+    from dvc_objects.db import ObjectDB
 
 
 class RepoDependency(Dependency):
@@ -29,7 +32,8 @@ class RepoDependency(Dependency):
 
     def __init__(self, def_repo, stage, *args, **kwargs):
         self.def_repo = def_repo
-        self._staged_objs: Dict[str, "HashFile"] = {}
+        self._objs: Dict[str, "HashFile"] = {}
+        self._meta: Dict[str, Meta] = {}
         super().__init__(stage, *args, **kwargs)
 
     def _parse_path(self, fs, fs_path):
@@ -57,11 +61,11 @@ class RepoDependency(Dependency):
     def save(self):
         pass
 
-    def dumpd(self):
+    def dumpd(self, **kwargs):
         return {self.PARAM_PATH: self.def_path, self.PARAM_REPO: self.def_repo}
 
     def download(self, to, jobs=None):
-        from dvc.data.checkout import checkout
+        from dvc_data.hashfile.checkout import checkout
 
         for odb, objs in self.get_used_objs().items():
             self.repo.cloud.pull(objs, jobs=jobs, odb=odb)
@@ -72,8 +76,9 @@ class RepoDependency(Dependency):
             to.fs,
             obj,
             self.repo.odb.local,
-            dvcignore=None,
+            ignore=None,
             state=self.repo.state,
+            prompt=confirm,
         )
 
     def update(self, rev=None):
@@ -91,23 +96,21 @@ class RepoDependency(Dependency):
     def get_used_objs(
         self, **kwargs
     ) -> Dict[Optional["ObjectDB"], Set["HashInfo"]]:
-        used, _ = self._get_used_and_obj(**kwargs)
+        used, _, _ = self._get_used_and_obj(**kwargs)
         return used
 
     def _get_used_and_obj(
         self, obj_only=False, **kwargs
-    ) -> Tuple[Dict[Optional["ObjectDB"], Set["HashInfo"]], "HashFile"]:
+    ) -> Tuple[Dict[Optional["ObjectDB"], Set["HashInfo"]], Meta, "HashFile"]:
         from dvc.config import NoRemoteError
-        from dvc.data.stage import stage
-        from dvc.data.tree import Tree, TreeError
         from dvc.exceptions import NoOutputOrStageError, PathMissingError
         from dvc.utils import as_posix
+        from dvc_data.hashfile.build import build
+        from dvc_data.hashfile.tree import Tree, TreeError
 
         local_odb = self.repo.odb.local
         locked = kwargs.pop("locked", True)
-        with self._make_repo(
-            locked=locked, cache_dir=local_odb.cache_dir
-        ) as repo:
+        with self._make_repo(locked=locked, cache_dir=local_odb.path) as repo:
             used_obj_ids = defaultdict(set)
             rev = repo.get_rev()
             if locked and self.def_repo.get(self.PARAM_REV_LOCK) is None:
@@ -130,45 +133,47 @@ class RepoDependency(Dependency):
                     pass
 
             try:
-                staging, _, staged_obj = stage(
+                object_store, meta, obj = build(
                     local_odb,
                     as_posix(self.def_path),
-                    repo.repo_fs,
+                    repo.dvcfs,
                     local_odb.fs.PARAM_CHECKSUM,
                 )
             except (FileNotFoundError, TreeError) as exc:
                 raise PathMissingError(
                     self.def_path, self.def_repo[self.PARAM_URL]
                 ) from exc
-            staging = copy(staging)
-            staging.read_only = True
+            object_store = copy(object_store)
+            object_store.read_only = True
 
-            self._staged_objs[rev] = staged_obj
-            used_obj_ids[staging].add(staged_obj.hash_info)
-            if isinstance(staged_obj, Tree):
-                used_obj_ids[staging].update(oid for _, _, oid in staged_obj)
-            return used_obj_ids, staged_obj
+            self._objs[rev] = obj
+            self._meta[rev] = meta
+
+            used_obj_ids[object_store].add(obj.hash_info)
+            if isinstance(obj, Tree):
+                used_obj_ids[object_store].update(oid for _, _, oid in obj)
+            return used_obj_ids, meta, obj
 
     def _check_circular_import(self, odb, obj_ids):
-        from dvc.data.db.reference import ReferenceObjectDB
-        from dvc.data.tree import Tree
         from dvc.exceptions import CircularImportError
-        from dvc.fs.repo import RepoFileSystem
+        from dvc.fs.dvc import DVCFileSystem
+        from dvc_data.hashfile.db.reference import ReferenceHashFileDB
+        from dvc_data.hashfile.tree import Tree
 
-        if not isinstance(odb, ReferenceObjectDB):
+        if not isinstance(odb, ReferenceHashFileDB):
             return
 
         def iter_objs():
             for hash_info in obj_ids:
                 if hash_info.isdir:
                     tree = Tree.load(odb, hash_info)
-                    yield from (odb.get(oid) for _, _, oid in tree)
+                    yield from (odb.get(hi.value) for _, _, hi in tree)
                 else:
-                    yield odb.get(hash_info)
+                    yield odb.get(hash_info.value)
 
         checked_urls = set()
         for obj in iter_objs():
-            if not isinstance(obj.fs, RepoFileSystem):
+            if not isinstance(obj.fs, DVCFileSystem):
                 continue
             if (
                 obj.fs.repo_url in checked_urls
@@ -187,13 +192,23 @@ class RepoDependency(Dependency):
     def get_obj(self, filter_info=None, **kwargs):
         locked = kwargs.get("locked", True)
         rev = self._get_rev(locked=locked)
-        if rev in self._staged_objs:
-            return self._staged_objs[rev]
-        _, obj = self._get_used_and_obj(
+        if rev in self._objs:
+            return self._objs[rev]
+        _, _, obj = self._get_used_and_obj(
             obj_only=True, filter_info=filter_info, **kwargs
         )
         return obj
 
+    def get_meta(self, filter_info=None, **kwargs):
+        locked = kwargs.get("locked", True)
+        rev = self._get_rev(locked=locked)
+        if rev in self._meta:
+            return self._meta[rev]
+        _, meta, _ = self._get_used_and_obj(
+            obj_only=True, filter_info=filter_info, **kwargs
+        )
+        return meta
+
     def _make_repo(self, locked=True, **kwargs):
         from dvc.external_repo import external_repo
 

dvc/dvcfile.py

@@ -59,7 +59,7 @@ def is_lock_file(path):
 
 
 def is_git_ignored(repo, path):
-    from dvc.fs.local import LocalFileSystem
+    from dvc.fs import LocalFileSystem
     from dvc.scm import NoSCMError
 
     try:
@@ -172,7 +172,7 @@ class FileMixin:
     def dump(self, stage, **kwargs):
         raise NotImplementedError
 
-    def merge(self, ancestor, other):
+    def merge(self, ancestor, other, allowed=None):
         raise NotImplementedError
 
 
@@ -197,19 +197,19 @@ class SingleStageFile(FileMixin):
         assert not isinstance(stage, PipelineStage)
         if self.verify:
             check_dvcfile_path(self.repo, self.path)
-        logger.debug(f"Saving information to '{relpath(self.path)}'.")
-        dump_yaml(self.path, serialize.to_single_stage_file(stage))
+        logger.debug("Saving information to '%s'.", relpath(self.path))
+        dump_yaml(self.path, serialize.to_single_stage_file(stage, **kwargs))
         self.repo.scm_context.track_file(self.relpath)
 
     def remove_stage(self, stage):  # pylint: disable=unused-argument
         self.remove()
 
-    def merge(self, ancestor, other):
+    def merge(self, ancestor, other, allowed=None):
         assert isinstance(ancestor, SingleStageFile)
         assert isinstance(other, SingleStageFile)
 
         stage = self.stage
-        stage.merge(ancestor.stage, other.stage)
+        stage.merge(ancestor.stage, other.stage, allowed=allowed)
         self.dump(stage)
 
 
@@ -237,15 +237,17 @@ class PipelineFile(FileMixin):
             self._dump_pipeline_file(stage)
 
         if update_lock:
-            self._dump_lockfile(stage)
+            self._dump_lockfile(stage, **kwargs)
 
-    def _dump_lockfile(self, stage):
-        self._lockfile.dump(stage)
+    def _dump_lockfile(self, stage, **kwargs):
+        self._lockfile.dump(stage, **kwargs)
 
     @staticmethod
-    def _check_if_parametrized(stage):
+    def _check_if_parametrized(stage, action: str = "dump") -> None:
         if stage.raw_data.parametrized:
-            raise ParametrizedDumpError(f"cannot dump a parametrized {stage}")
+            raise ParametrizedDumpError(
+                f"cannot {action} a parametrized {stage}"
+            )
 
     def _dump_pipeline_file(self, stage):
         self._check_if_parametrized(stage)
@@ -291,6 +293,7 @@ class PipelineFile(FileMixin):
         self._lockfile.remove()
 
     def remove_stage(self, stage):
+        self._check_if_parametrized(stage, "remove")
         self._lockfile.remove_stage(stage)
         if not self.exists():
             return
@@ -307,7 +310,7 @@ class PipelineFile(FileMixin):
         else:
             super().remove()
 
-    def merge(self, ancestor, other):
+    def merge(self, ancestor, other, allowed=None):
         raise NotImplementedError
 
 
@@ -363,7 +366,7 @@ class Lockfile(FileMixin):
         return {SCHEMA_KWD: version}
 
     def dump(self, stage, **kwargs):
-        stage_data = serialize.to_lockfile(stage)
+        stage_data = serialize.to_lockfile(stage, **kwargs)
 
         with modify_yaml(self.path, fs=self.repo.fs) as data:
             version = LOCKFILE_VERSION.from_dict(data)
@@ -408,7 +411,7 @@ class Lockfile(FileMixin):
         else:
             self.remove()
 
-    def merge(self, ancestor, other):
+    def merge(self, ancestor, other, allowed=None):
         raise NotImplementedError
 
 

dvc/env.py

@@ -2,9 +2,6 @@ DVC_CHECKPOINT = "DVC_CHECKPOINT"
 DVC_DAEMON = "DVC_DAEMON"
 DVC_PAGER = "DVC_PAGER"
 DVC_ROOT = "DVC_ROOT"
-DVCLIVE_PATH = "DVCLIVE_PATH"
-DVCLIVE_SUMMARY = "DVCLIVE_SUMMARY"
-DVCLIVE_HTML = "DVCLIVE_HTML"
 DVCLIVE_RESUME = "DVCLIVE_RESUME"
 DVC_IGNORE_ISATTY = "DVC_IGNORE_ISATTY"
 DVC_EXP_GIT_REMOTE = "DVC_EXP_GIT_REMOTE"

dvc/exceptions.py

@@ -132,10 +132,17 @@ class NotDvcRepoError(DvcException):
 class CyclicGraphError(DvcException):
     def __init__(self, stages):
         assert isinstance(stages, list)
-        msg = "Pipeline has a cycle involving: {}.".format(
-            ", ".join(s.addressing for s in stages)
+        stage_part = "stage" if len(stages) == 1 else "stages"
+        msg = (
+            "Same item(s) are defined as both a dependency and an output "
+            "in {stage_part}: {stage}."
+        )
+        super().__init__(
+            msg.format(
+                stage_part=stage_part,
+                stage=", ".join(s.addressing for s in stages),
+            )
         )
-        super().__init__(msg)
 
 
 class ConfirmRemoveError(DvcException):
@@ -202,14 +209,6 @@ class FileMissingError(DvcException):
         )
 
 
-class DvcIgnoreInCollectedDirError(DvcException):
-    def __init__(self, ignore_dirname):
-        super().__init__(
-            ".dvcignore file should not be in collected dir path: "
-            "'{}'".format(ignore_dirname)
-        )
-
-
 class FileTransferError(DvcException):
     _METHOD = "transfer"
 
@@ -286,6 +285,11 @@ class PathMissingError(DvcException):
         self.dvc_only = dvc_only
 
 
+class URLMissingError(DvcException):
+    def __init__(self, url):
+        super().__init__(f"The path '{url}' does not exist")
+
+
 class RemoteCacheRequiredError(DvcException):
     def __init__(self, scheme, fs_path):
 

dvc/external_repo.py

@@ -33,7 +33,7 @@ def external_repo(
     from scmrepo.git import Git
 
     from dvc.config import NoRemoteError
-    from dvc.fs.git import GitFileSystem
+    from dvc.fs import GitFileSystem
 
     logger.debug("Creating external repo %s@%s", url, rev)
     path = _cached_clone(url, rev, for_write=for_write)
@@ -50,19 +50,21 @@ def external_repo(
     config.update(cache_config)
 
     if for_write:
+        scm = None
         root_dir = path
         fs = None
     else:
-        root_dir = os.path.realpath(path)
-        scm = Git(root_dir)
+        scm = Git(path)
         fs = GitFileSystem(scm=scm, rev=rev)
+        root_dir = "/"
 
     repo_kwargs = dict(
         root_dir=root_dir,
         url=url,
         fs=fs,
         config=config,
-        repo_factory=erepo_factory(url, cache_config),
+        repo_factory=erepo_factory(url, root_dir, cache_config),
+        scm=scm,
         **kwargs,
     )
 
@@ -92,14 +94,16 @@ def external_repo(
             _remove(path)
 
 
-def erepo_factory(url, cache_config, *args, **kwargs):
-    def make_repo(path, **_kwargs):
+def erepo_factory(url, root_dir, cache_config):
+    from dvc.fs import localfs
+
+    def make_repo(path, fs=None, **_kwargs):
         _config = cache_config.copy()
         if os.path.isdir(url):
-            rel = os.path.relpath(path, _kwargs["fs"].fs_args["scm"].root_dir)
-            repo_path = os.path.join(url, rel)
+            fs = fs or localfs
+            repo_path = os.path.join(url, *fs.path.relparts(path, root_dir))
             _config.update(_get_remote_config(repo_path))
-        return Repo(path, config=_config, **_kwargs)
+        return Repo(path, fs=fs, config=_config, **_kwargs)
 
     return make_repo
 
@@ -141,7 +145,7 @@ def _get_remote_config(url):
             name = "auto-generated-upstream"
             return {
                 "core": {"remote": name},
-                "remote": {name: {"url": repo.odb.local.cache_dir}},
+                "remote": {name: {"url": repo.odb.local.path}},
             }
 
         # Use original remote to make sure that we are using correct url,

dvc/fs/__init__.py

@@ -1,44 +1,70 @@
 from urllib.parse import urlparse
 
-from ..scheme import Schemes
-from . import utils  # noqa: F401
-from .azure import AzureFileSystem
-from .gdrive import GDriveFileSystem
-from .gs import GSFileSystem
-from .hdfs import HDFSFileSystem
-from .http import HTTPFileSystem
-from .https import HTTPSFileSystem
-from .local import LocalFileSystem
-from .oss import OSSFileSystem
-from .s3 import S3FileSystem
-from .ssh import SSHFileSystem
-from .webdav import WebDAVFileSystem, WebDAVSFileSystem
-from .webhdfs import WebHDFSFileSystem
+from dvc_http import HTTPFileSystem, HTTPSFileSystem  # noqa: F401
 
-FS_MAP = {
-    Schemes.AZURE: AzureFileSystem,
-    Schemes.GDRIVE: GDriveFileSystem,
-    Schemes.GS: GSFileSystem,
-    Schemes.HDFS: HDFSFileSystem,
-    Schemes.WEBHDFS: WebHDFSFileSystem,
-    Schemes.HTTP: HTTPFileSystem,
-    Schemes.HTTPS: HTTPSFileSystem,
-    Schemes.S3: S3FileSystem,
-    Schemes.SSH: SSHFileSystem,
-    Schemes.OSS: OSSFileSystem,
-    Schemes.WEBDAV: WebDAVFileSystem,
-    Schemes.WEBDAVS: WebDAVSFileSystem,
-    # NOTE: LocalFileSystem is the default
-}
+from dvc.config import ConfigError as RepoConfigError
+from dvc.config_schema import SCHEMA, Invalid
+
+# pylint: disable=unused-import
+from dvc_objects.fs import utils  # noqa: F401
+from dvc_objects.fs import (  # noqa: F401
+    LocalFileSystem,
+    MemoryFileSystem,
+    Schemes,
+    generic,
+    get_fs_cls,
+    known_implementations,
+    localfs,
+    registry,
+    system,
+)
+from dvc_objects.fs.base import AnyFSPath, FileSystem  # noqa: F401
+from dvc_objects.fs.errors import (  # noqa: F401
+    AuthError,
+    ConfigError,
+    RemoteMissingDepsError,
+)
+from dvc_objects.fs.path import Path  # noqa: F401
+
+from .callbacks import Callback
+from .data import DataFileSystem  # noqa: F401
+from .dvc import DVCFileSystem  # noqa: F401
+from .git import GitFileSystem  # noqa: F401
+
+known_implementations.update(
+    {
+        "dvc": {
+            "class": "dvc.fs.dvc.DVCFileSystem",
+            "err": "dvc is supported, but requires 'dvc' to be installed",
+        },
+        "git": {
+            "class": "dvc.fs.git.GitFileSystem",
+            "err": "git is supported, but requires 'dvc' to be installed",
+        },
+    }
+)
 
 
-def get_fs_cls(remote_conf, scheme=None):
-    if not scheme:
-        scheme = urlparse(remote_conf["url"]).scheme
-    return FS_MAP.get(scheme, LocalFileSystem)
+# pylint: enable=unused-import
 
 
-def get_fs_config(repo, config, **kwargs):
+def download(fs, fs_path, to, jobs=None):
+    with Callback.as_tqdm_callback(
+        desc=f"Downloading {fs.path.name(fs_path)}",
+        unit="files",
+    ) as cb:
+        fs.get(fs_path, to.fs_path, batch_size=jobs, callback=cb)
+
+
+def parse_external_url(url, config=None):
+    remote_config = dict(config) if config else {}
+    remote_config["url"] = url
+    fs_cls, fs_config, fs_path = get_cloud_fs(None, **remote_config)
+    fs = fs_cls(**fs_config)
+    return fs, fs_path
+
+
+def get_fs_config(config, **kwargs):
     name = kwargs.get("name")
     if name:
         try:
@@ -49,10 +75,10 @@ def get_fs_config(repo, config, **kwargs):
             raise RemoteNotFoundError(f"remote '{name}' doesn't exist")
     else:
         remote_conf = kwargs
-    return _resolve_remote_refs(repo, config, remote_conf)
+    return _resolve_remote_refs(config, remote_conf)
 
 
-def _resolve_remote_refs(repo, config, remote_conf):
+def _resolve_remote_refs(config, remote_conf):
     # Support for cross referenced remotes.
     # This will merge the settings, shadowing base ref with remote_conf.
     # For example, having:
@@ -78,30 +104,26 @@ def _resolve_remote_refs(repo, config, remote_conf):
     if parsed.scheme != "remote":
         return remote_conf
 
-    base = get_fs_config(repo, config, name=parsed.netloc)
-    cls, _, _ = get_cloud_fs(repo, **base)
+    base = get_fs_config(config, name=parsed.netloc)
+    cls, _, _ = _get_cloud_fs(config, **base)
     relpath = parsed.path.lstrip("/").replace("/", cls.sep)
     url = cls.sep.join((base["url"], relpath))
     return {**base, **remote_conf, "url": url}
 
 
 def get_cloud_fs(repo, **kwargs):
-    from dvc.config import ConfigError
-    from dvc.config_schema import SCHEMA, Invalid
-
     repo_config = repo.config if repo else {}
+    return _get_cloud_fs(repo_config, **kwargs)
+
+
+def _get_cloud_fs(repo_config, **kwargs):
     core_config = repo_config.get("core", {})
 
-    remote_conf = get_fs_config(repo, repo_config, **kwargs)
+    remote_conf = get_fs_config(repo_config, **kwargs)
     try:
         remote_conf = SCHEMA["remote"][str](remote_conf)
     except Invalid as exc:
-        raise ConfigError(str(exc)) from None
-
-    if "jobs" not in remote_conf:
-        jobs = core_config.get("jobs")
-        if jobs:
-            remote_conf["jobs"] = jobs
+        raise RepoConfigError(str(exc)) from None
 
     if "checksum_jobs" not in remote_conf:
         checksum_jobs = core_config.get("checksum_jobs")
@@ -110,11 +132,8 @@ def get_cloud_fs(repo, **kwargs):
 
     cls = get_fs_cls(remote_conf)
 
-    if cls == GDriveFileSystem and repo:
-        remote_conf["gdrive_credentials_tmp_dir"] = repo.tmp_dir
-
     url = remote_conf.pop("url")
-    if issubclass(cls, WebDAVFileSystem):
+    if cls.protocol in ["webdav", "webdavs"]:
         # For WebDAVFileSystem, provided url is the base path itself, so it
         # should be treated as being a root path.
         fs_path = cls.root_marker