From fa5aced98b847df27e960e08f8650920df9e377f Mon Sep 17 00:00:00 2001 From: Sandy Walsh Date: Thu, 14 Aug 2014 17:31:59 +0000 Subject: [PATCH] fixed up event_pump to use template dir supplied --- bin/event_pump.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/bin/event_pump.py b/bin/event_pump.py index 77efab9..64d16c9 100644 --- a/bin/event_pump.py +++ b/bin/event_pump.py @@ -8,6 +8,7 @@ pip install --pre notification_utils import datetime +import sys from notabene import kombu_driver as driver import notification_utils @@ -22,16 +23,20 @@ queue = driver.create_queue(queue_name, exchange, queue_name, channel=connection.channel()) queue.declare() -g = notigen.EventGenerator(1000) # Number of operations per minute +template_dir = sys.argv[1] +print "Using template dir:", template_dir + +g = notigen.EventGenerator(template_dir, 1) now = datetime.datetime.utcnow() start = now +end = now + datetime.timedelta(days=1) nevents = 0 -while nevents < 10000: +while now < end: e = g.generate(now) if e: nevents += len(e) for event in e: driver.send_notification(event, queue_name, connection, exchange) - print nevents, event['when'], event['event_type'] + print nevents, event['timestamp'], event['event_type'] now = datetime.datetime.utcnow()