Jelajahi Sumber

refactor imports in Attack/AttackParameters.py

Jens Keim 6 tahun lalu
induk
melakukan
8c0f8918ec
1 mengubah file dengan 3 tambahan dan 3 penghapusan
  1. 3 3
      code/Attack/AttackParameters.py

+ 3 - 3
code/Attack/AttackParameters.py

@@ -1,7 +1,7 @@
-from enum import Enum
+import enum
 
 
-class Parameter(Enum):
+class Parameter(enum.Enum):
     """
     Defines the shortname for attack parameters. The shortname may be used for attack parameter specification
     when calling ID2T via the command line.
@@ -46,7 +46,7 @@ class Parameter(Enum):
     CUSTOM_PAYLOAD_FILE = 'custom.payload.file'  # file that contains custom payload for ftp exploits
 
 
-class ParameterTypes(Enum):
+class ParameterTypes(enum.Enum):
     """
     Defines types for parameters. These types may be used in the specification of allowed parameters within the
     individual attack classes. The type is used to verify the validity of the given value.