Skip to content

Commit

Permalink
Merge pull request #6241 from pypa/trim-click
Browse files Browse the repository at this point in the history
remove click echo from utils.resolver
  • Loading branch information
matteius authored Sep 13, 2024
2 parents 9911630 + 66b921d commit 080031d
Showing 1 changed file with 15 additions and 19 deletions.
34 changes: 15 additions & 19 deletions pipenv/utils/resolver.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,12 +24,11 @@
from pipenv.patched.pip._internal.req.req_file import parse_requirements
from pipenv.patched.pip._internal.req.req_install import InstallRequirement
from pipenv.patched.pip._internal.utils.temp_dir import global_tempdir_manager
from pipenv.patched.pip._vendor import rich
from pipenv.patched.pip._vendor.packaging.utils import canonicalize_name
from pipenv.project import Project
from pipenv.utils import console, err
from pipenv.utils.fileutils import create_tracked_tempdir
from pipenv.utils.requirements import normalize_name
from pipenv.vendor import click

from .dependencies import (
HackedPythonVersion,
Expand All @@ -51,9 +50,6 @@
else:
import importlib.metadata as importlib_metadata

console = rich.console.Console()
err = rich.console.Console(stderr=True)


def get_package_finder(
install_cmd=None,
Expand Down Expand Up @@ -531,11 +527,10 @@ def resolve_constraints(self):
if result.req:
result.req.marker = marker
except TypeError as e:
click.echo(
err.print(
f"Error generating python marker for {candidate}. "
f"Is the specifier {requires_python} incorrectly quoted or otherwise wrong?"
f"Full error: {e}",
err=True,
)
new_tree.add(result)

Expand Down Expand Up @@ -706,15 +701,14 @@ def actually_resolve_deps(

def resolve(cmd, st, project):
from pipenv.cmdparse import Script
from pipenv.vendor.click import echo

c = subprocess_run(Script.parse(cmd).cmd_args, block=False, env=os.environ.copy())
is_verbose = project.s.is_verbose()
err = ""
errors = ""
for line in iter(c.stderr.readline, ""):
if not line.rstrip():
continue
err += line
errors += line
if is_verbose:
st.console.print(line.rstrip())

Expand All @@ -723,13 +717,13 @@ def resolve(cmd, st, project):
out = c.stdout.read()
if returncode != 0:
st.console.print(environments.PIPENV_SPINNER_FAIL_TEXT.format("Locking Failed!"))
echo(out.strip(), err=True)
err.print(out.strip())
if not is_verbose:
echo(err, err=True)
err.print(err)
raise RuntimeError("Failed to lock Pipfile.lock!")
if is_verbose:
echo(out.strip(), err=True)
return subprocess.CompletedProcess(c.args, returncode, out, err)
err.print(out.strip())
return subprocess.CompletedProcess(c.args, returncode, out, errors)


def venv_resolve_deps(
Expand Down Expand Up @@ -868,8 +862,8 @@ def venv_resolve_deps(
with open(target_file.name) as fh:
results = json.load(fh)
except (IndexError, json.JSONDecodeError):
click.echo(c.stdout.strip(), err=True)
click.echo(c.stderr.strip(), err=True)
err.print(c.stdout.strip())
err.print(c.stderr.strip())
if os.path.exists(target_file.name):
os.unlink(target_file.name)
raise RuntimeError("There was a problem with locking.")
Expand All @@ -879,13 +873,15 @@ def venv_resolve_deps(
environments.PIPENV_SPINNER_OK_TEXT.format("Success!")
)
if not project.s.is_verbose() and c.stderr.strip():
click.echo(click.style(f"Warning: {c.stderr.strip()}"), err=True)
err.print(
f"Warning: {c.stderr.strip()}", overflow="ignore", crop=False
)
else:
st.console.print(
environments.PIPENV_SPINNER_FAIL_TEXT.format("Locking Failed!")
)
click.echo(f"Output: {c.stdout.strip()}", err=True)
click.echo(f"Error: {c.stderr.strip()}", err=True)
err.print(f"Output: {c.stdout.strip()}")
err.print(f"Error: {c.stderr.strip()}")
if lockfile_section not in lockfile:
lockfile[lockfile_section] = {}
return prepare_lockfile(
Expand Down

0 comments on commit 080031d

Please sign in to comment.