From 65313880cf7251a49ab53d403cef15a44a7f9bff Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Rapha=C3=ABl=20Vinot?= Date: Mon, 20 Feb 2023 14:27:57 +0100 Subject: [PATCH] fix: Properly handle missing parameter in CSV importer Fix #931 --- examples/load_csv.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/examples/load_csv.py b/examples/load_csv.py index 580791a..7732196 100755 --- a/examples/load_csv.py +++ b/examples/load_csv.py @@ -10,7 +10,7 @@ from pymisp import MISPEvent try: from keys import misp_url, misp_key, misp_verifycert - from pymisp import ExpandedPyMISP + from pymisp import PyMISP offline = False except ImportError as e: offline = True @@ -66,7 +66,7 @@ if __name__ == '__main__': if offline: print('You are in offline mode, quitting.') 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: event = MISPEvent() event.info = args.new_event @@ -80,7 +80,7 @@ if __name__ == '__main__': else: print('Something went wrong:') print(new_event) - else: + elif args.update_event: for o in objects: new_object = misp.add_object(args.update_event, o, pythonify=True) if isinstance(new_object, str): @@ -90,3 +90,5 @@ if __name__ == '__main__': else: print('Something went wrong:') 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)')