MISP
/
cti-python-stix2
mirror of https://github.com/MISP/cti-python-stix2
2
0
Fork 0
Code Issues Releases Wiki Activity

Merge pull request #112 from samcornwell/get_type_from_id 

Add get_type_from_id utility function
stix2.0
Greg Back 2017-11-20 15:19:24 +00:00 committed by GitHub
parent ea11163ae6 eff51a2bb8
commit f481788e1e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
stix2/test/test_utils.py
View File

@ -74,3 +74,11 @@ def test_get_dict(data):
def test_get_dict_invalid(data): def test_get_dict_invalid(data):
with pytest.raises(ValueError): with pytest.raises(ValueError):
stix2.utils.get_dict(data) stix2.utils.get_dict(data)
@pytest.mark.parametrize('stix_id, typ', [
('malware--d69c8146-ab35-4d50-8382-6fc80e641d43', 'malware'),
('intrusion-set--899ce53f-13a0-479b-a0e4-67d46e241542', 'intrusion-set')
])
def test_get_type_from_id(stix_id, typ):
assert stix2.utils.get_type_from_id(stix_id) == typ

4
stix2/utils.py
View File

@ -255,3 +255,7 @@ def get_class_hierarchy_names(obj):
for cls in obj.__class__.__mro__: for cls in obj.__class__.__mro__:
names.append(cls.__name__) names.append(cls.__name__)
return names return names
def get_type_from_id(stix_id):
return stix_id.split('--', 1)[0]