fix: Properly handle missing parameter in CSV importer

Fix #931
pull/934/head
Raphaël Vinot 2023-02-20 14:27:57 +01:00
parent d6460eb9b0
commit 65313880cf
1 changed files with 5 additions and 3 deletions

View File

@ -10,7 +10,7 @@ from pymisp import MISPEvent
try: try:
from keys import misp_url, misp_key, misp_verifycert from keys import misp_url, misp_key, misp_verifycert
from pymisp import ExpandedPyMISP from pymisp import PyMISP
offline = False offline = False
except ImportError as e: except ImportError as e:
offline = True offline = True
@ -66,7 +66,7 @@ if __name__ == '__main__':
if offline: if offline:
print('You are in offline mode, quitting.') print('You are in offline mode, quitting.')
else: else:
misp = ExpandedPyMISP(url=misp_url, key=misp_key, ssl=misp_verifycert) misp = PyMISP(url=misp_url, key=misp_key, ssl=misp_verifycert)
if args.new_event: if args.new_event:
event = MISPEvent() event = MISPEvent()
event.info = args.new_event event.info = args.new_event
@ -80,7 +80,7 @@ if __name__ == '__main__':
else: else:
print('Something went wrong:') print('Something went wrong:')
print(new_event) print(new_event)
else: elif args.update_event:
for o in objects: for o in objects:
new_object = misp.add_object(args.update_event, o, pythonify=True) new_object = misp.add_object(args.update_event, o, pythonify=True)
if isinstance(new_object, str): if isinstance(new_object, str):
@ -90,3 +90,5 @@ if __name__ == '__main__':
else: else:
print('Something went wrong:') print('Something went wrong:')
print(new_event) print(new_event)
else:
print('you need to pass either a event info field (flag -i), or the event ID you want to update (flag -u)')