32a4d9e459
- Made the same renaming in the yaml laydown config files. - Added Type hints wherever I've been. - Added a custom NodeType in custom_typing.py to encompass the Union of ActiveNode, PassiveNode, ServiceNode.
29 lines
826 B
Python
29 lines
826 B
Python
# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence.
|
|
"""The Service class."""
|
|
|
|
from primaite.common.enums import SoftwareState
|
|
|
|
|
|
class Service(object):
|
|
"""Service class."""
|
|
|
|
def __init__(self, name: str, port: str, software_state: SoftwareState):
|
|
"""
|
|
Init.
|
|
|
|
:param name: The service name.
|
|
:param port: The service port.
|
|
:param software_state: The service SoftwareState.
|
|
"""
|
|
self.name = name
|
|
self.port = port
|
|
self.software_state = software_state
|
|
self.patching_count = 0
|
|
|
|
def reduce_patching_count(self):
|
|
"""Reduces the patching count for the service."""
|
|
self.patching_count -= 1
|
|
if self.patching_count <= 0:
|
|
self.patching_count = 0
|
|
self.software_state = SoftwareState.GOOD
|