diff --git a/monkey/infection_monkey/credential_collectors/credential_components/i_credential_component.py b/monkey/infection_monkey/credential_collectors/credential_components/i_credential_component.py index 566d3ed05..43ebb9aca 100644 --- a/monkey/infection_monkey/credential_collectors/credential_components/i_credential_component.py +++ b/monkey/infection_monkey/credential_collectors/credential_components/i_credential_component.py @@ -1,10 +1,10 @@ from abc import ABC from dataclasses import dataclass -from ..credential_types import CredentialTypes +from ..credential_type import CredentialType @dataclass class ICredentialComponent(ABC): - type: CredentialTypes + type: CredentialType content: dict diff --git a/monkey/infection_monkey/credential_collectors/credential_components/password.py b/monkey/infection_monkey/credential_collectors/credential_components/password.py index 3691ee69a..87abe1575 100644 --- a/monkey/infection_monkey/credential_collectors/credential_components/password.py +++ b/monkey/infection_monkey/credential_collectors/credential_components/password.py @@ -1,7 +1,7 @@ -from ..credential_types import CredentialTypes +from ..credential_type import CredentialType from .i_credential_component import ICredentialComponent class Password(ICredentialComponent): def __init__(self, password: str): - super().__init__(type=CredentialTypes.PASSWORD, content={"password": password}) + super().__init__(type=CredentialType.PASSWORD, content={"password": password}) diff --git a/monkey/infection_monkey/credential_collectors/credential_components/ssh_keypair.py b/monkey/infection_monkey/credential_collectors/credential_components/ssh_keypair.py index 09d0d2f01..040256ee0 100644 --- a/monkey/infection_monkey/credential_collectors/credential_components/ssh_keypair.py +++ b/monkey/infection_monkey/credential_collectors/credential_components/ssh_keypair.py @@ -1,7 +1,7 @@ -from ..credential_types import CredentialTypes +from ..credential_type import CredentialType from .i_credential_component import ICredentialComponent class SSHKeypair(ICredentialComponent): def __init__(self, content: dict): - super().__init__(type=CredentialTypes.SSH_KEYPAIR, content=content) + super().__init__(type=CredentialType.SSH_KEYPAIR, content=content) diff --git a/monkey/infection_monkey/credential_collectors/credential_components/username.py b/monkey/infection_monkey/credential_collectors/credential_components/username.py index 4ccf0ddea..98c7b0b3d 100644 --- a/monkey/infection_monkey/credential_collectors/credential_components/username.py +++ b/monkey/infection_monkey/credential_collectors/credential_components/username.py @@ -1,7 +1,7 @@ -from ..credential_types import CredentialTypes +from ..credential_type import CredentialType from .i_credential_component import ICredentialComponent class Username(ICredentialComponent): def __init__(self, username: str): - super().__init__(type=CredentialTypes.USERNAME, content={"username": username}) + super().__init__(type=CredentialType.USERNAME, content={"username": username}) diff --git a/monkey/infection_monkey/credential_collectors/credential_types.py b/monkey/infection_monkey/credential_collectors/credential_type.py similarity index 76% rename from monkey/infection_monkey/credential_collectors/credential_types.py rename to monkey/infection_monkey/credential_collectors/credential_type.py index 2354664a1..5e8a9e6ea 100644 --- a/monkey/infection_monkey/credential_collectors/credential_types.py +++ b/monkey/infection_monkey/credential_collectors/credential_type.py @@ -1,7 +1,7 @@ from enum import Enum -class CredentialTypes(Enum): +class CredentialType(Enum): SSH_KEYPAIR = 1 USERNAME = 2 PASSWORD = 3