Skip to content

Commit e3965bb

Browse files
committed
[client] Export required functions for import
1 parent cc0697f commit e3965bb

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

pyobas/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,8 @@
1010
)
1111
from pyobas.client import OpenBAS # noqa: F401
1212
from pyobas.exceptions import * # noqa: F401,F403,F405
13+
from pyobas.helpers import * # noqa: F401,F403,F405
14+
from pyobas.utils import * # noqa: F401,F403,F405
1315

1416
__all__ = [
1517
"__author__",

pyobas/_contracts/contract_config.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4,8 +4,7 @@
44
from enum import Enum
55
from typing import List
66

7-
from utils import EnhancedJSONEncoder
8-
7+
from pyobas import utils
98
from pyobas._contracts.contract_utils import ContractCardinality, ContractVariable
109
from pyobas._contracts.variable_helper import VariableHelper
1110

@@ -138,7 +137,7 @@ def prepare_contracts(contracts):
138137
"contract_id": c.contract_id,
139138
"contract_labels": c.label,
140139
"contract_attack_patterns": c.attack_patterns,
141-
"contract_content": json.dumps(c, cls=EnhancedJSONEncoder),
140+
"contract_content": json.dumps(c, cls=utils.EnhancedJSONEncoder),
142141
},
143142
contracts,
144143
)

0 commit comments

Comments
 (0)