From 4c9e6d0ec8502f19e01ed4efceca60642ae67d86 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Rapha=C3=ABl=20Vinot?= Date: Wed, 26 Dec 2018 18:27:48 +0100 Subject: [PATCH] fix: Create massive event using ExpandedPyMISP --- examples/events/create_massive_dummy_events.py | 10 +++++++--- examples/events/tools.py | 10 +++++++--- 2 files changed, 14 insertions(+), 6 deletions(-) diff --git a/examples/events/create_massive_dummy_events.py b/examples/events/create_massive_dummy_events.py index 12a2826..7829ad6 100755 --- a/examples/events/create_massive_dummy_events.py +++ b/examples/events/create_massive_dummy_events.py @@ -1,8 +1,12 @@ #!/usr/bin/env python # -*- coding: utf-8 -*- -from pymisp import PyMISP -from keys import url, key +from pymisp import ExpandedPyMISP +try: + from keys import url, key +except ImportError: + url = 'http://localhost:8080' + key = '8h0gHbhS0fv6JUOlTED0AznLXFbf83TYtQrCycqb' import argparse import tools @@ -13,7 +17,7 @@ if __name__ == '__main__': parser.add_argument("-a", "--attribute", type=int, help="Number of attributes per event (default 3000)") args = parser.parse_args() - misp = PyMISP(url, key, True, 'json') + misp = ExpandedPyMISP(url, key, True) if args.limit is None: args.limit = 1 diff --git a/examples/events/tools.py b/examples/events/tools.py index 76b70a2..94f5d91 100644 --- a/examples/events/tools.py +++ b/examples/events/tools.py @@ -4,6 +4,7 @@ import random from random import randint import string +from pymisp import MISPEvent def randomStringGenerator(size, chars=string.ascii_lowercase + string.digits): @@ -63,13 +64,16 @@ def create_dummy_event(misp): def create_massive_dummy_events(misp, nbattribute): - event = misp.new_event(0, 4, 0, 'massive dummy event') - eventid = event['Event']['id'] + event = MISPEvent() + event.info = 'massive dummy event' + event = misp.add_event(event) + print(event) + eventid = event.id distribution = '0' functions = [floodtxt, floodip, flooddomain, flooddomainip, floodemail, floodattachment] for i in range(nbattribute): choice = randint(0, 5) if choice == 5: - floodattachment(misp, eventid, distribution, False, 'Payload delivery', '', event['Event']['info'], event['Event']['analysis'], event['Event']['threat_level_id']) + floodattachment(misp, eventid, distribution, False, 'Payload delivery', '', event.info, event.analysis, event.threat_level_id) else: functions[choice](misp, event)