dvc-utils 0.0.5__py3-none-any.whl → 0.0.7__py3-none-any.whl

dvc_utils/main.py

@@ -2,15 +2,12 @@ from functools import cache
 from os import environ as env, getcwd
 from os.path import join, relpath
 import shlex
-from subprocess import Popen, PIPE
 from typing import Optional, Tuple
 
 from click import option, argument, group
 import click
 import yaml
-from utz import process, singleton, err
-
-from dvc_utils.named_pipes import named_pipes
+from utz import diff_cmds, process, err, singleton
 
 
 @group()
@@ -69,86 +66,6 @@ def dvc_cache_path(ref: str, dvc_path: Optional[str] = None, log: bool = False)
     return join(dvc_cache_dir(log=log), 'files', 'md5', dirname, basename)
 
 
-def diff_cmds(
-    cmds1: list[str],
-    cmds2: list[str],
-    verbose: bool = False,
-    color: bool = False,
-    unified: int | None = None,
-    ignore_whitespace: bool = False,
-    **kwargs,
-):
-    """Run two sequences of piped commands and diff their output.
-
-    Args:
-        cmds1: First sequence of commands to pipe together
-        cmds2: Second sequence of commands to pipe together
-        verbose: Whether to print commands being executed
-        color: Whether to show colored diff output
-        unified: Number of unified context lines, or None
-        ignore_whitespace: Whether to ignore whitespace changes
-        **kwargs: Additional arguments passed to subprocess.Popen
-
-    Each command sequence will be piped together before being compared.
-    For example, if cmds1 = ['cat foo.txt', 'sort'], the function will
-    execute 'cat foo.txt | sort' before comparing with cmds2's output.
-
-    Adapted from https://stackoverflow.com/a/28840955"""
-    with named_pipes(n=2) as pipes:
-        (pipe1, pipe2) = pipes
-        diff_cmd = [
-            'diff',
-            *(['-w'] if ignore_whitespace else []),
-            *(['-U', str(unified)] if unified is not None else []),
-            *(['--color=always'] if color else []),
-            pipe1,
-            pipe2,
-        ]
-        diff = Popen(diff_cmd)
-        processes = []
-
-        for pipe, cmds in ((pipe1, cmds1), (pipe2, cmds2)):
-            if verbose:
-                err(f"Running pipeline: {' | '.join(cmds)}")
-
-            # Create the pipeline of processes
-            prev_process = None
-            for i, cmd in enumerate(cmds):
-                is_last = i + 1 == len(cmds)
-
-                # For the first process, take input from the original source
-                stdin = None if prev_process is None else prev_process.stdout
-
-                # For the last process, output to the named pipe
-                if is_last:
-                    with open(pipe, 'wb', 0) as pipe_fd:
-                        proc = Popen(
-                            cmd,
-                            stdin=stdin,
-                            stdout=pipe_fd,
-                            close_fds=True,
-                            **kwargs
-                        )
-                # For intermediate processes, output to a pipe
-                else:
-                    proc = Popen(
-                        cmd,
-                        stdin=stdin,
-                        stdout=PIPE,
-                        close_fds=True,
-                        **kwargs
-                    )
-
-                if prev_process is not None:
-                    prev_process.stdout.close()
-
-                processes.append(proc)
-                prev_process = proc
-
-        for p in [diff] + processes:
-            p.wait()
-
-
 @cli.command('diff', short_help='Diff a DVC-tracked file at two commits (or one commit vs. current worktree), optionally passing both through another command first')
 @option('-c', '--color', is_flag=True, help='Colorize the output')
 @option('-r', '--refspec', default='HEAD', help='<commit 1>..<commit 2> (compare two commits) or <commit> (compare <commit> to the worktree)')

{dvc_utils-0.0.5.dist-info → dvc_utils-0.0.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dvc-utils
-Version: 0.0.5
+Version: 0.0.7
 Summary: CLI for diffing DVC files at two commits (or one commit vs. current worktree), optionally passing both through another command first
 Home-page: https://github.com/runsascoded/dvc-utils
 Author: Ryan Williams
@@ -15,7 +15,7 @@ CLI for diffing [DVC] files, optionally passing both through another command fir
 <!-- toc -->
 - [Installation](#installation)
 - [Usage](#usage)
-    - [`dvc-utils diff`](#dvc-utils-diff)
+    - [`dvc-diff`](#dvc-diff)
 - [Examples](#examples)
     - [Parquet file](#parquet-diff)
         - [Schema diff](#parquet-schema-diff)
@@ -42,11 +42,13 @@ dvc-utils --help
 #         worktree), optionally passing both through another command first
 ```
 
-### `dvc-utils diff` <a id="dvc-utils-diff"></a>
-<!-- `bmdf -- dvc-utils diff --help` -->
+The single subcommand, `dvc-utils diff`, is also exposed directly as `dvc-dff`:
+
+### `dvc-diff` <a id="dvc-diff"></a>
+<!-- `bmdf -- dvc-diff --help` -->
 ```bash
-dvc-utils diff --help
-# Usage: dvc-utils diff [OPTIONS] [cmd...] <path>
+dvc-diff --help
+# Usage: dvc-diff [OPTIONS] [exec_cmd...] <path>
 #
 #   Diff a file at two commits (or one commit vs. current worktree), optionally
 #   passing both through `cmd` first
@@ -69,6 +71,8 @@ dvc-utils diff --help
 #                            to `diff`)
 #   -v, --verbose            Log intermediate commands to stderr
 #   -w, --ignore-whitespace  Ignore whitespace differences (pass `-w` to `diff`)
+#   -x, --exec-cmd TEXT      Command(s) to execute before diffing; alternate
+#                            syntax to passing commands as positional arguments
 #   --help                   Show this message and exit.
 ```
 
@@ -91,7 +95,7 @@ parquet_schema() {
     parquet2json "$1" schema
 }
 export -f parquet_schema
-dvc-utils diff -r $commit^..$commit parquet_schema $path
+dvc-diff -r $commit^..$commit parquet_schema $path
 ```
 <details><summary>Output</summary>
 
@@ -149,7 +153,7 @@ pretty_print_first_row() {
     parquet2json "$1" cat -l 1 | jq .
 }
 export -f pretty_print_first_row
-dvc-utils diff -r $commit^..$commit pretty_print_first_row $path
+dvc-diff -r $commit^..$commit pretty_print_first_row $path
 ```
 
 <details><summary>Output</summary>
@@ -208,7 +212,7 @@ parquet_row_count() {
     parquet2json "$1" rowcount
 }
 export -f parquet_row_count
-dvc-utils diff -r $commit^..$commit parquet_row_count $path
+dvc-diff -r $commit^..$commit parquet_row_count $path
 ```
 
 This time we get no output; [the given `$commit`][commit] didn't change the row count in the DVC-tracked Parquet file [`$path`][commit path].

dvc_utils-0.0.7.dist-info/RECORD

@@ -0,0 +1,8 @@
+dvc_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+dvc_utils/main.py,sha256=1eiIL96OPY-dP12cNzHHDFB5KyGI2baauZgeDDjyxHs,5002
+dvc_utils-0.0.7.dist-info/LICENSE,sha256=ZS8AReay7xmQzBAHwxIuTouGXz3SKgUa2_Sz8Ip0EzQ,1070
+dvc_utils-0.0.7.dist-info/METADATA,sha256=kqz3QZm4TZOjhr2GsHjJPyfx0n0HBshk-O8fm3fjzs0,7107
+dvc_utils-0.0.7.dist-info/WHEEL,sha256=bFJAMchF8aTQGUgMZzHJyDDMPTO3ToJ7x23SLJa1SVo,92
+dvc_utils-0.0.7.dist-info/entry_points.txt,sha256=0JqFeb29N5ZrvwmIJrg4ii32xNFOXRul-HMq-3GOIw4,90
+dvc_utils-0.0.7.dist-info/top_level.txt,sha256=jT0-PJa2t_eFRE9rn-52AjdnZ8nQeEHllf2kJmaGh80,10
+dvc_utils-0.0.7.dist-info/RECORD,,

{dvc_utils-0.0.5.dist-info → dvc_utils-0.0.7.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.44.0)
+Generator: bdist_wheel (0.45.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

{dvc_utils-0.0.5.dist-info → dvc_utils-0.0.7.dist-info}/entry_points.txt

@@ -1,2 +1,3 @@
 [console_scripts]
+dvc-diff = dvc_utils.main:dvc_utils_diff
 dvc-utils = dvc_utils.main:cli

dvc_utils/named_pipes.py

@@ -1,19 +0,0 @@
-import os
-import shutil
-import tempfile
-from contextlib import contextmanager
-
-
-@contextmanager
-def named_pipes(n: int = 1):
-    """Yield a list of paths to named pipes that are created and destroyed
-
-    From https://stackoverflow.com/a/28840955"""
-    dirname = tempfile.mkdtemp()
-    try:
-        paths = [os.path.join(dirname, 'named_pipe' + str(i)) for i in range(n)]
-        for path in paths:
-            os.mkfifo(path)
-        yield paths
-    finally:
-        shutil.rmtree(dirname)

dvc_utils-0.0.5.dist-info/RECORD

@@ -1,9 +0,0 @@
-dvc_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-dvc_utils/main.py,sha256=UhCm28Zu_xwsmfzOKgmAYJrCrfumryid_10F_-Rg-2M,7850
-dvc_utils/named_pipes.py,sha256=VQ2t9BYCazFq_-MABj4t2HS7GHDvSqXXx8fOLz5DsTc,492
-dvc_utils-0.0.5.dist-info/LICENSE,sha256=ZS8AReay7xmQzBAHwxIuTouGXz3SKgUa2_Sz8Ip0EzQ,1070
-dvc_utils-0.0.5.dist-info/METADATA,sha256=y46AoDbejlmv318zUD4aKiY9SprlRh9hdCBcoGZ4H4M,6924
-dvc_utils-0.0.5.dist-info/WHEEL,sha256=eOLhNAGa2EW3wWl_TU484h7q1UNgy0JXjjoqKoxAAQc,92
-dvc_utils-0.0.5.dist-info/entry_points.txt,sha256=W9OuZ6CX8QF9ojbqLtfXFo8Q2hnJ-zlcGY4_7nO8paM,49
-dvc_utils-0.0.5.dist-info/top_level.txt,sha256=jT0-PJa2t_eFRE9rn-52AjdnZ8nQeEHllf2kJmaGh80,10
-dvc_utils-0.0.5.dist-info/RECORD,,