chg: Improve sharing groups, bring back organsations included and ability to get specific SG

pull/795/head
Tom King 2021-10-04 11:41:36 +01:00
parent 0a3f2e68ef
commit c120db02b8
2 changed files with 43 additions and 0 deletions

View File

@ -1921,6 +1921,21 @@ class PyMISP:
to_return.append(s)
return to_return
def get_sharing_group(self, sharing_group: Union[MISPSharingGroup, int, str, UUID], pythonify: bool = False) -> Union[Dict, MISPSharingGroup]:
"""Get a sharing group
:param sharing_group: sharing group to find
:param pythonify: Returns a PyMISP Object instead of the plain json output
"""
sharing_group_id = get_uuid_or_id_from_abstract_misp(sharing_group)
r = self._prepare_request('GET', f'sharing_groups/view/{sharing_group_id}')
sharing_group = self._check_json_response(r)
if not (self.global_pythonify or pythonify) or 'errors' in sharing_group:
return sharing_group
s = MISPSharingGroup()
s.from_dict(**sharing_group)
return s
def add_sharing_group(self, sharing_group: MISPSharingGroup, pythonify: bool = False) -> Union[Dict, MISPSharingGroup]:
"""Add a new sharing group

View File

@ -126,12 +126,40 @@ class MISPOrganisation(AbstractMISP):
class MISPSharingGroup(AbstractMISP):
def __init__(self):
super().__init__()
self.SharingGroupOrg: List[MISPOrganisation] = []
@property
def orgs(self) -> List[MISPOrganisation]:
return self.SharingGroupOrg
@orgs.setter
def orgs(self, orgs: List[MISPOrganisation]):
"""Set a list of prepared MISPSighting."""
if all(isinstance(x, MISPSighting) for x in orgs):
self.SharingGroupOrg = orgs
else:
raise PyMISPError('All the attributes have to be of type MISPOrganisation.')
def add_org(self, org):
misp_org = MISPOrganisation()
misp_org.from_dict(**org)
self.SharingGroupOrg.append(misp_org)
return misp_org
def from_dict(self, **kwargs):
if 'SharingGroupOrg' in kwargs:
[self.add_org(org) for org in kwargs.pop('SharingGroupOrg')]
if 'SharingGroup' in kwargs:
kwargs = kwargs['SharingGroup']
super().from_dict(**kwargs)
def __repr__(self) -> str:
if hasattr(self, 'name'):
return f'<{self.__class__.__name__}(name={self.name})'
return f'<{self.__class__.__name__}(NotInitialized)'
class MISPShadowAttribute(AbstractMISP):