mirror of
https://github.com/aquasecurity/kube-hunter.git
synced 2026-08-23 22:26:23 +00:00
Yehuda Chikvashvili
a0127659b7
* Make config initialized explicitly * Add mypy linting * Make tests run individually Resolve #341
51 lines
1.3 KiB
Python
51 lines
1.3 KiB
Python
from dataclasses import dataclass
|
|
from typing import Any, Optional
|
|
|
|
|
|
@dataclass
|
|
class Config:
|
|
""" Config is a configuration container.
|
|
It contains the following fields:
|
|
- active: Enable active hunters
|
|
- cidr: Network subnets to scan
|
|
- dispatcher: Dispatcher object
|
|
- include_patched_version: Include patches in version comparison
|
|
- interface: Interface scanning mode
|
|
- list_hunters: Print a list of existing hunters
|
|
- log_level: Log level
|
|
- mapping: Report only found components
|
|
- network_timeout: Timeout for network operations
|
|
- pod: From pod scanning mode
|
|
- quick: Quick scanning mode
|
|
- remote: Hosts to scan
|
|
- report: Output format
|
|
- statistics: Include hunters statistics
|
|
"""
|
|
|
|
active: bool = False
|
|
cidr: Optional[str] = None
|
|
dispatcher: Optional[Any] = None
|
|
include_patched_versions: bool = False
|
|
interface: bool = False
|
|
mapping: bool = False
|
|
network_timeout: float = 5.0
|
|
pod: bool = False
|
|
quick: bool = False
|
|
remote: Optional[str] = None
|
|
reporter: Optional[Any] = None
|
|
statistics: bool = False
|
|
|
|
|
|
_config: Optional[Config] = None
|
|
|
|
|
|
def get_config() -> Config:
|
|
if not _config:
|
|
raise ValueError("Configuration is not initialized")
|
|
return _config
|
|
|
|
|
|
def set_config(new_config: Config) -> None:
|
|
global _config
|
|
_config = new_config
|