Skip to content

Commit 47259da

Browse files
committed
✨ feat(contrat_config): Add new ContractOutputType ExpectationSignature
1 parent adc201b commit 47259da

1 file changed

Lines changed: 9 additions & 12 deletions

File tree

pyoaev/contracts/contract_config.py

Lines changed: 9 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -57,6 +57,7 @@ class ContractOutputType(str, Enum):
5757
Sid: str = "sid"
5858
Vulnerability: str = "vulnerability"
5959
AccountWithPasswordNotRequired: str = "account_with_password_not_required"
60+
ExpectationSignature: str = "expectation_signature"
6061
AsreproastableAccount: str = "asreproastable_account"
6162
KerberoastableAccount: str = "kerberoastable_account"
6263

@@ -152,12 +153,14 @@ class Contract:
152153
config: ContractConfig
153154
manual: bool
154155
variables: List[ContractVariable] = field(
155-
default_factory=lambda: [
156-
VariableHelper.user_variable(),
157-
VariableHelper.exercise_variable(),
158-
VariableHelper.team_variable(),
159-
]
160-
+ VariableHelper.uri_variables()
156+
default_factory=lambda: (
157+
[
158+
VariableHelper.user_variable(),
159+
VariableHelper.exercise_variable(),
160+
VariableHelper.team_variable(),
161+
]
162+
+ VariableHelper.uri_variables()
163+
)
161164
)
162165
contract_attack_patterns_external_ids: List[str] = field(default_factory=list)
163166
contract_vulnerability_external_ids: List[str] = field(default_factory=list)
@@ -212,7 +215,6 @@ def get_type(self) -> str:
212215

213216
@dataclass
214217
class ContractText(ContractCardinalityElement):
215-
216218
defaultValue: str = ""
217219

218220
@property
@@ -253,7 +255,6 @@ def get_type(self) -> str:
253255

254256
@dataclass
255257
class ContractTextArea(ContractCardinalityElement):
256-
257258
defaultValue: str = ""
258259
richText: bool = False
259260

@@ -264,7 +265,6 @@ def get_type(self) -> str:
264265

265266
@dataclass
266267
class ContractCheckbox(ContractElement):
267-
268268
defaultValue: bool = False
269269

270270
@property
@@ -274,7 +274,6 @@ def get_type(self) -> str:
274274

275275
@dataclass
276276
class ContractAttachment(ContractCardinalityElement):
277-
278277
@property
279278
def get_type(self) -> str:
280279
return ContractFieldType.Attachment.value
@@ -292,7 +291,6 @@ def get_type(self) -> str:
292291

293292
@dataclass
294293
class ContractSelect(ContractCardinalityElement):
295-
296294
choices: dict[str, str] = None
297295

298296
@property
@@ -320,7 +318,6 @@ def get_type(self) -> str:
320318

321319
@dataclass
322320
class ContractPayload(ContractCardinalityElement):
323-
324321
@property
325322
def get_type(self) -> str:
326323
return ContractFieldType.Payload.value

0 commit comments

Comments
 (0)