added misp object templates path argument

pull/171/head
Tobias Mainka 2018-01-11 12:39:19 +01:00
parent 85bc693a63
commit 716fd2723c
1 changed files with 6 additions and 3 deletions

View File

@ -814,9 +814,12 @@ class MISPObject(AbstractMISP):
super(MISPObject, self).__init__(**kwargs)
self.__strict = strict
self.name = name
self.__misp_objects_path = os.path.join(
os.path.abspath(os.path.dirname(sys.modules['pymisp'].__file__)),
'data', 'misp-objects', 'objects')
if kwargs.get('misp_objects_path', None):
self.__misp_objects_path = kwargs.get('misp_objects_path', None)
else:
self.__misp_objects_path = os.path.join(
os.path.abspath(os.path.dirname(sys.modules['pymisp'].__file__)),
'data', 'misp-objects', 'objects')
if os.path.exists(os.path.join(self.__misp_objects_path, self.name, 'definition.json')):
self.__known_template = True
else: