From f9a54888af7021f7dbd5646baa134ed2d56ca6eb Mon Sep 17 00:00:00 2001
From: Kenneth Reitz <me@kennethreitz.org>
Date: Thu, 2 Feb 2017 21:22:20 -0500
Subject: [PATCH] better

---
 pipenv/cli.py | 25 ++++++++++++++++++++++---
 1 file changed, 22 insertions(+), 3 deletions(-)

diff --git a/pipenv/cli.py b/pipenv/cli.py
index 9527dbfe..2a1ceed6 100644
--- a/pipenv/cli.py
+++ b/pipenv/cli.py
@@ -674,13 +674,14 @@ def easter_egg(package_name):
 @click.group(invoke_without_command=True)
 @click.option('--where', is_flag=True, default=False, help="Output project home information.")
 @click.option('--venv', is_flag=True, default=False, help="Output virtualenv information.")
+@click.option('--rm', is_flag=True, default=False, help="Remove the virtualenv.")
 @click.option('--bare', is_flag=True, default=False, help="Minimal output.")
 @click.option('--three/--two', is_flag=True, default=None, help="Use Python 3/2 when creating virtualenv.")
 @click.option('--python', default=False, nargs=1, help="Specify which version of Python virtualenv should use.")
 @click.option('--help', '-h', is_flag=True, default=None, help="Show this message then exit.")
 @click.version_option(prog_name=crayons.yellow('pipenv'), version=__version__)
 @click.pass_context
-def cli(ctx, where=False, venv=False, bare=False, three=False, python=False, help=False):
+def cli(ctx, where=False, venv=False, rm=False, bare=False, three=False, python=False, help=False):
 
     if ctx.invoked_subcommand is None:
         # --where was passed...
@@ -692,12 +693,30 @@ def cli(ctx, where=False, venv=False, bare=False, three=False, python=False, hel
         elif venv:
 
             # There is no virtualenv yet.
-            if not project.virtualenv_location:
+            with spinner():
+                loc = project.virtualenv_location
+
+            if not loc:
                 click.echo(crayons.red('No virtualenv has been created for this project yet!'), err=True)
+                sys.exit(1)
             else:
                 click.echo(project.virtualenv_location)
+                sys.exit(0)
 
-            sys.exit(1)
+        elif rm:
+
+            with spinner():
+                loc = project.virtualenv_location
+
+            if loc:
+                click.echo(crayons.yellow('{0} ({1})...'.format(crayons.yellow('Removing virtualenv'), crayons.green(loc))))
+                with spinner():
+                    # Remove the virtualenv.
+                    shutil.rmtree(project.virtualenv_location)
+                sys.exit(0)
+            else:
+                click.echo(crayons.red('No virtualenv has been created for this project yet!'), err=True)
+                sys.exit(1)
 
         # --two / --three was passed.
         if (python) or (three is not None):