69 lines
1.3 KiB
Python
69 lines
1.3 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
import argparse
|
||
|
|
import shlex
|
||
|
|
import shutil
|
||
|
|
import subprocess
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
parser = argparse.ArgumentParser(
|
||
|
|
prog="Remove venv v1.0.0",
|
||
|
|
formatter_class=argparse.RawTextHelpFormatter,
|
||
|
|
description="Remove all the venv-created files",
|
||
|
|
)
|
||
|
|
parser.add_argument(
|
||
|
|
"--freeze",
|
||
|
|
action="store_true",
|
||
|
|
help="Runs a venv freeze before deleting",
|
||
|
|
)
|
||
|
|
parser.add_argument(
|
||
|
|
"--dir",
|
||
|
|
type=Path,
|
||
|
|
help="Choose directory to remove venv inside",
|
||
|
|
)
|
||
|
|
args = parser.parse_args()
|
||
|
|
|
||
|
|
VENV_BASENAMES = [
|
||
|
|
"1",
|
||
|
|
"bin",
|
||
|
|
"include",
|
||
|
|
"lib",
|
||
|
|
"lib64",
|
||
|
|
"pyvenv.cfg",
|
||
|
|
]
|
||
|
|
|
||
|
|
|
||
|
|
def check_for_venv(directory):
|
||
|
|
global VENV_BASENAMES
|
||
|
|
|
||
|
|
for file in VENV_BASENAMES:
|
||
|
|
if not (directory / file).exists():
|
||
|
|
print(f"Missing file {file}")
|
||
|
|
return False
|
||
|
|
|
||
|
|
return True
|
||
|
|
|
||
|
|
|
||
|
|
def rm_venv(directory):
|
||
|
|
for file in VENV_BASENAMES:
|
||
|
|
p = directory / file
|
||
|
|
|
||
|
|
if p.is_dir():
|
||
|
|
shutil.rmtree(p)
|
||
|
|
else:
|
||
|
|
p.unlink()
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
p = Path().cwd() if args.dir == None else args.dir
|
||
|
|
|
||
|
|
if not check_for_venv(p):
|
||
|
|
print("Not a venv")
|
||
|
|
exit(1)
|
||
|
|
|
||
|
|
if args.freeze:
|
||
|
|
subprocess.run(
|
||
|
|
shlex.split("bash -c '. bin/activate && pip freeze > requirements.txt'")
|
||
|
|
)
|
||
|
|
|
||
|
|
rm_venv(p)
|