x/stacktach-notigen
Code Issues Proposed changes

fixed up event_pump to use template dir supplied

Browse Source
This commit is contained in:
Sandy Walsh 2014-08-14 17:31:59 +00:00
parent b208de357c
commit fa5aced98b
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
bin/event_pump.py
View File

@ -8,6 +8,7 @@ pip install --pre notification_utils
import datetime import datetime
import sys
from notabene import kombu_driver as driver from notabene import kombu_driver as driver
import notification_utils import notification_utils
@ -22,16 +23,20 @@ queue = driver.create_queue(queue_name, exchange, queue_name,
channel=connection.channel()) channel=connection.channel())
queue.declare() 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() now = datetime.datetime.utcnow()
start = now start = now
end = now + datetime.timedelta(days=1)
nevents = 0 nevents = 0
while nevents < 10000: while now < end:
e = g.generate(now) e = g.generate(now)
if e: if e:
nevents += len(e) nevents += len(e)
for event in e: for event in e:
driver.send_notification(event, queue_name, connection, exchange) 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() now = datetime.datetime.utcnow()