fix: [api] Notes, Opinions and Relationships are no longer AnalystDataBehaviorMixin

main
Christian Studer 2024-12-22 14:08:05 +01:00
parent 201bf5afea
commit ea8fd62c41
No known key found for this signature in database
GPG Key ID: 6BBED1B63A6D639F
1 changed files with 5 additions and 5 deletions

View File

@ -32,7 +32,7 @@ from .mispevent import MISPEvent, MISPAttribute, MISPSighting, MISPLog, MISPObje
MISPRole, MISPServer, MISPFeed, MISPEventDelegation, MISPCommunity, MISPUserSetting, \ MISPRole, MISPServer, MISPFeed, MISPEventDelegation, MISPCommunity, MISPUserSetting, \
MISPInbox, MISPEventBlocklist, MISPOrganisationBlocklist, MISPEventReport, \ MISPInbox, MISPEventBlocklist, MISPOrganisationBlocklist, MISPEventReport, \
MISPGalaxyCluster, MISPGalaxyClusterRelation, MISPCorrelationExclusion, MISPDecayingModel, \ MISPGalaxyCluster, MISPGalaxyClusterRelation, MISPCorrelationExclusion, MISPDecayingModel, \
MISPNote, MISPOpinion, MISPRelationship, AnalystDataBehaviorMixin MISPNote, MISPOpinion, MISPRelationship, MISPAnalystData
from .abstract import pymisp_json_default, MISPTag, AbstractMISP, describe_types from .abstract import pymisp_json_default, MISPTag, AbstractMISP, describe_types
@ -621,14 +621,14 @@ class PyMISP:
# ## END Galaxy Cluster ### # ## END Galaxy Cluster ###
# ## BEGIN Analyst Data ###a # ## BEGIN Analyst Data ###a
def get_analyst_data(self, analyst_data: AnalystDataBehaviorMixin | int | str | UUID, def get_analyst_data(self, analyst_data: MISPAnalystData | int | str | UUID,
pythonify: bool = False) -> dict[str, Any] | MISPNote | MISPOpinion | MISPRelationship: pythonify: bool = False) -> dict[str, Any] | MISPNote | MISPOpinion | MISPRelationship:
"""Get an analyst data from a MISP instance """Get an analyst data from a MISP instance
:param analyst_data: analyst data to get :param analyst_data: analyst data to get
:param pythonify: Returns a list of PyMISP Objects instead of the plain json output. Warning: it might use a lot of RAM :param pythonify: Returns a list of PyMISP Objects instead of the plain json output. Warning: it might use a lot of RAM
""" """
if isinstance(analyst_data, AnalystDataBehaviorMixin): if isinstance(analyst_data, MISPAnalystData):
analyst_data_type = analyst_data.analyst_data_object_type analyst_data_type = analyst_data.analyst_data_object_type
else: else:
analyst_data_type = 'all' analyst_data_type = 'all'
@ -666,7 +666,7 @@ class PyMISP:
:param analyst_data_id: analyst data ID to update :param analyst_data_id: analyst data ID to update
:param pythonify: Returns a PyMISP Object instead of the plain json output :param pythonify: Returns a PyMISP Object instead of the plain json output
""" """
if isinstance(analyst_data, AnalystDataBehaviorMixin): if isinstance(analyst_data, MISPAnalystData):
analyst_data_type = analyst_data.analyst_data_object_type analyst_data_type = analyst_data.analyst_data_object_type
else: else:
analyst_data_type = 'all' analyst_data_type = 'all'
@ -685,7 +685,7 @@ class PyMISP:
:param analyst_data: analyst data to delete :param analyst_data: analyst data to delete
""" """
if isinstance(analyst_data, AnalystDataBehaviorMixin): if isinstance(analyst_data, MISPAnalystData):
analyst_data_type = analyst_data.analyst_data_object_type analyst_data_type = analyst_data.analyst_data_object_type
else: else:
analyst_data_type = 'all' analyst_data_type = 'all'