From a599afa81cd39012e5ada14edc7a3b41026326dc Mon Sep 17 00:00:00 2001 From: Elijah DeLee Date: Wed, 27 May 2020 18:29:11 -0400 Subject: [PATCH] Also apply time delta to job This way they all happened on the same day --- tools/scripts/firehose.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/tools/scripts/firehose.py b/tools/scripts/firehose.py index b99158c4e2..42bdcd9a87 100755 --- a/tools/scripts/firehose.py +++ b/tools/scripts/firehose.py @@ -63,8 +63,8 @@ class YieldedRows(StringIO): def __init__(self, job_id, rows, time_delta, *args, **kwargs): self.rows = rows - created_time = datetime.datetime.today() - datetime.timedelta(days=time_delta[0], hours=time_delta[1], seconds=5) - modified_time = datetime.datetime.today() - datetime.timedelta(days=time_delta[0], hours=time_delta[1], seconds=0) + created_time = datetime.datetime.today() - time_delta - datetime.timedelta(seconds=5) + modified_time = datetime.datetime.today() - time_delta created_stamp = created_time.strftime("%Y-%m-%d %H:%M:%S") modified_stamp = modified_time.strftime("%Y-%m-%d %H:%M:%S") self.row = "\t".join([ @@ -124,7 +124,7 @@ def cleanup(sql): conn.close() -def generate_jobs(jobs, batch_size): +def generate_jobs(jobs, batch_size, time_delta): print(f'inserting {jobs} job(s)') sys.path.insert(0, pkg_resources.get_distribution('awx').module_path) from awx import prepare_env @@ -161,7 +161,7 @@ def generate_jobs(jobs, batch_size): jobs = [ Job( status=STATUS_OPTIONS[i % len(STATUS_OPTIONS)], - started=now(), created=now(), modified=now(), finished=now(), + started=now()-time_delta, created=now()-time_delta, modified=now()-time_delta, finished=now()-time_delta, elapsed=0., **jt_defaults) for i in range(N) ] @@ -300,8 +300,9 @@ if __name__ == '__main__': params = parser.parse_args() jobs = params.jobs time_delta = params.days_delta, params.hours_delta + time_delta = datetime.timedelta(days=time_delta[0], hours=time_delta[1], seconds=0) events = params.events batch_size = params.batch_size print(datetime.datetime.utcnow().isoformat()) - created = generate_jobs(jobs, batch_size=batch_size) + created = generate_jobs(jobs, batch_size=batch_size, time_delta=time_delta) generate_events(events, str(created.pk), time_delta)