Skip to content

Commit 82ed23e

Browse files
committed
Use logging for create_sample_data script
1 parent e29cce2 commit 82ed23e

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

server/devel/create_sample_data.py

+7-4
Original file line numberDiff line numberDiff line change
@@ -18,13 +18,16 @@
1818
allow you to use the production server.
1919
"""
2020

21+
import logging
2122
import random
2223
import sys
2324
from argparse import ArgumentParser, Namespace
2425
from typing import Iterator, Optional, Tuple
2526

2627
import requests
2728

29+
logging.basicConfig(level=logging.INFO)
30+
2831

2932
def get_args() -> Namespace:
3033
"""Parse command line arguments
@@ -223,24 +226,24 @@ def main():
223226

224227
queues = QueueDataGenerator(num_queues=args.queues)
225228
testflinger_client.post_queue_data(queues=queues)
226-
print(f"Created {args.queues} queues")
229+
logging.info("Created %s queues", args.queues)
227230

228231
valid_queue_names = extract_queue_names(queues=queues)
229232

230233
agents = AgentDataGenerator(
231234
num_agents=args.agents, queue_list=valid_queue_names
232235
)
233236
testflinger_client.post_agent_data(agents=agents)
234-
print(f"Created {args.agents} agents")
237+
logging.info("Created %s agents", args.agents)
235238

236239
jobs = JobDataGenerator(num_jobs=args.jobs, queue_list=valid_queue_names)
237240
testflinger_client.post_job_data(jobs=jobs)
238-
print(f"Created {args.jobs} jobs")
241+
logging.info("Created %s jobs", args.jobs)
239242

240243

241244
if __name__ == "__main__":
242245
try:
243246
main()
244247
except requests.exceptions.RequestException as error:
245-
print(f"Error: {error}")
248+
logging.error(error)
246249
sys.exit(1)

0 commit comments

Comments
 (0)