47 lines
1 KiB
Python
Executable file
47 lines
1 KiB
Python
Executable file
#!/usr/bin/env python
|
|
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
|
|
from shutil import which
|
|
|
|
if not os.getenv("VIRTUAL_ENV"):
|
|
sys.exit("Ensure your virtualenv is activated.")
|
|
|
|
file_extensions = (".py",)
|
|
autoflake = os.getenv("AUTOFLAKE_BIN", "autoflake")
|
|
options = [
|
|
"--in-place",
|
|
"--remove-unused-variables",
|
|
"--remove-all-unused-imports",
|
|
"--ignore-init-module-imports"
|
|
]
|
|
|
|
if not which(autoflake):
|
|
sys.exit("Autoflake binary not found")
|
|
|
|
diff = subprocess.check_output(
|
|
["git", "diff", "--cached", "--name-only", "--diff-filter", "ACM"],
|
|
universal_newlines=True
|
|
)
|
|
|
|
python_files = [
|
|
item for item in diff.splitlines() if item.endswith(file_extensions)
|
|
]
|
|
|
|
if not python_files:
|
|
sys.exit(0)
|
|
|
|
try:
|
|
output = subprocess.check_output(
|
|
[autoflake, *options, *python_files],
|
|
universal_newlines=True,
|
|
stderr=subprocess.STDOUT
|
|
)
|
|
except subprocess.CalledProcessError as e:
|
|
print("Formatting was unsuccessful:")
|
|
sys.exit(e.stdout)
|
|
|
|
subprocess.run(["git", "add", "-u", *python_files])
|
|
sys.exit(0)
|