Skip to content

Commit 3445ae8

Browse files
author
Samuel Hassine
committed
[examples] Enhance all examples
1 parent e10cf15 commit 3445ae8

File tree

4 files changed

+8
-13
lines changed

4 files changed

+8
-13
lines changed

examples/config.yml.sample

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,3 @@
11
opencti:
2-
api_url: 'http://opencti.example.com'
2+
url: 'http://opencti.example.com'
33
token : 'ChangeMe'

examples/observables/get_observables.py

Lines changed: 3 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -3,21 +3,16 @@
33
import os
44
import yaml
55

6-
from pycti.opencti_api_client import OpenCTI
6+
from pycti import OpenCTIApiClient
77

88
# Load configuration
99
config = yaml.load(open(os.path.dirname(__file__) + '/../config.yml'))
1010

11-
# File to import
12-
file_to_import = config['mitre']['repository_path_cti'] + '/enterprise-attack/enterprise-attack.json'
13-
1411
# OpenCTI initialization
15-
opencti = OpenCTI(config['opencti']['api_url'], config['opencti']['api_key'], config['opencti']['log_file'], config['opencti']['verbose'])
12+
opencti_api_client = OpenCTIApiClient(config['opencti']['url'], config['opencti']['token'], 'info')
1613

1714
# Get observables and their context
18-
observables = opencti.get_stix_observables(10)
19-
20-
opencti.health_check()
15+
observables = opencti_api_client.get_stix_observables(10)
2116

2217
for observable in observables:
2318
observable_value = observable['observable_value']

examples/stix2/export.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,10 +13,10 @@
1313
export_file = './report.json'
1414

1515
# OpenCTI initialization
16-
opencti = OpenCTIApiClient(config['opencti']['api_url'], config['opencti']['token'])
16+
opencti_api_client = OpenCTIApiClient(config['opencti']['url'], config['opencti']['token'])
1717

1818
# Import the bundle
19-
bundle = opencti.stix2_export_entity('report', '{REPORT_ID}', 'full')
19+
bundle = opencti_api_client.stix2_export_entity('report', '{REPORT_ID}', 'full')
2020

2121
with open(export_file, 'w') as file:
2222
json.dump(bundle, file, indent=4)

examples/stix2/import.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
file_to_import = './enterprise-attack.json'
1313

1414
# OpenCTI initialization
15-
opencti = OpenCTIApiClient(config['opencti']['api_url'], config['opencti']['token'])
15+
opencti_api_client = OpenCTIApiClient(config['opencti']['url'], config['opencti']['token'])
1616

1717
# Import the bundle
18-
opencti.stix2_import_bundle_from_file(file_to_import, False)
18+
opencti_api_client.stix2_import_bundle_from_file(file_to_import, False)

0 commit comments

Comments
 (0)