mirror of
https://github.com/peass-ng/PEASS-ng.git
synced 2026-08-01 00:20:50 -07:00
DanteandGitHub
fa5578b2ff
This pull request introduces a set of improvements to the peasLoaded.py file, aimed at enhancing the readability, maintainability, and performance of the code. The key changes include: - Indentation Correction: Fixed the indentation to comply with Python standards, ensuring proper code block recognition and avoiding potential runtime errors. - List Comprehension: Implemented list comprehension for the creation of FileRecord instances, which simplifies the code structure and improves readability. - Configuration Handling: Streamlined the access to the config dictionary by extracting it once at the beginning of the loop, reducing repetitive code and potential access errors. - Default Value Usage: Utilized the .get() method with default values from DEFAULTS for both `auto_check` and `exec` keys. These changes do not alter the core functionality of the code but provide a cleaner and more efficient approach to the existing logic. Please review the changes and let me know if there are any concerns or further improvements that can be made.
30 lines
985 B
Python
30 lines
985 B
Python
from .fileRecord import FileRecord
|
|
from .peassRecord import PEASRecord
|
|
from .yamlGlobals import YAML_LOADED, DEFAULTS
|
|
|
|
class PEASLoaded:
|
|
def __init__(self):
|
|
to_search = YAML_LOADED["search"]
|
|
self.peasrecords = []
|
|
|
|
for record in to_search:
|
|
record_value = record["value"]
|
|
config = record_value.get("config", {})
|
|
|
|
if "linpeas" in config.get("disable", "").lower():
|
|
continue
|
|
|
|
filerecords = [
|
|
FileRecord(regex=filerecord["name"], **filerecord["value"])
|
|
for filerecord in record_value["files"]
|
|
]
|
|
|
|
self.peasrecords.append(
|
|
PEASRecord(
|
|
name=record["name"],
|
|
auto_check=config.get("auto_check", DEFAULTS["auto_check"]),
|
|
exec=config.get("exec", DEFAULTS["exec"]),
|
|
filerecords=filerecords
|
|
)
|
|
)
|