mirror of
https://github.com/aquasecurity/kube-hunter.git
synced 2026-05-06 01:08:11 +00:00
0f1739262f
Fix linting issues with flake8 and black. Add pre-commit congifuration, update documnetation for it. Apply linting check in Travis CI.
61 lines
1.5 KiB
Python
61 lines
1.5 KiB
Python
from subprocess import check_call
|
|
from pkg_resources import parse_requirements
|
|
from configparser import ConfigParser
|
|
from setuptools import setup, Command
|
|
|
|
|
|
class ListDependenciesCommand(Command):
|
|
"""A custom command to list dependencies"""
|
|
|
|
description = "list package dependencies"
|
|
user_options = []
|
|
|
|
def initialize_options(self):
|
|
pass
|
|
|
|
def finalize_options(self):
|
|
pass
|
|
|
|
def run(self):
|
|
cfg = ConfigParser()
|
|
cfg.read("setup.cfg")
|
|
requirements = cfg["options"]["install_requires"]
|
|
print(requirements)
|
|
|
|
|
|
class PyInstallerCommand(Command):
|
|
"""A custom command to run PyInstaller to build standalone executable."""
|
|
|
|
description = "run PyInstaller on kube-hunter entrypoint"
|
|
user_options = []
|
|
|
|
def initialize_options(self):
|
|
pass
|
|
|
|
def finalize_options(self):
|
|
pass
|
|
|
|
def run(self):
|
|
cfg = ConfigParser()
|
|
cfg.read("setup.cfg")
|
|
command = [
|
|
"pyinstaller",
|
|
"--clean",
|
|
"--onefile",
|
|
"--name",
|
|
"kube-hunter",
|
|
]
|
|
setup_cfg = cfg["options"]["install_requires"]
|
|
requirements = parse_requirements(setup_cfg)
|
|
for r in requirements:
|
|
command.extend(["--hidden-import", r.key])
|
|
command.append("kube_hunter/__main__.py")
|
|
print(" ".join(command))
|
|
check_call(command)
|
|
|
|
|
|
setup(
|
|
use_scm_version={"fallback_version": "noversion"},
|
|
cmdclass={"dependencies": ListDependenciesCommand, "pyinstaller": PyInstallerCommand},
|
|
)
|