diff --git a/pymisp/api.py b/pymisp/api.py index f0f4cf0..6efff1e 100644 --- a/pymisp/api.py +++ b/pymisp/api.py @@ -1134,7 +1134,10 @@ class PyMISP: """ taxonomy_id = get_uuid_or_id_from_abstract_misp(taxonomy) t = self.get_taxonomy(taxonomy_id) - if not t['Taxonomy']['enabled']: + if isinstance(t, MISPTaxonomy) and not t.enabled: + # Can happen if global pythonify is enabled. + raise PyMISPError(f"The taxonomy {t.name} is not enabled.") + elif not t['Taxonomy']['enabled']: raise PyMISPError(f"The taxonomy {t['Taxonomy']['name']} is not enabled.") url = urljoin(self.root_url, 'taxonomies/addTag/{}'.format(taxonomy_id)) response = self._prepare_request('POST', url) diff --git a/pymisp/mispevent.py b/pymisp/mispevent.py index 5117334..218dea0 100644 --- a/pymisp/mispevent.py +++ b/pymisp/mispevent.py @@ -2054,6 +2054,9 @@ class MISPWarninglist(AbstractMISP): class MISPTaxonomy(AbstractMISP): + name: str + enabled: bool + def from_dict(self, **kwargs): if 'Taxonomy' in kwargs: kwargs = kwargs['Taxonomy']