浏览代码

refactor imports in Attack/AttackParameters.py

Jens Keim 6 年之前
父节点
当前提交
8c0f8918ec
共有 1 个文件被更改,包括 3 次插入3 次删除
  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.