picnoir
/
Systemd
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

log-generator: make message size configurable, add short options 

Also remove "b''" from the generated MESSAGE= field.
This commit is contained in:
Zbigniew Jędrzejewski-Szmek 2018-05-16 12:05:43 +02:00
parent 864876ec0f
commit 757ed4fd2e
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/journal-remote/log-generator.py
View File

@ -5,7 +5,8 @@ import argparse
PARSER = argparse.ArgumentParser()
PARSER.add_argument('n', type=int)
PARSER.add_argument('--dots', action='store_true')
PARSER.add_argument('--data-size', type=int, default=4000)
PARSER.add_argument('-m', '--message-size', type=int, default=200)
PARSER.add_argument('-d', '--data-size', type=int, default=4000)
PARSER.add_argument('--data-type', choices={'random', 'simple'})
OPTIONS = PARSER.parse_args()
@ -42,7 +43,9 @@ bytes = 0
counter = 0
for i in range(OPTIONS.n):
message = repr(src.read(2000))
message = src.read(OPTIONS.message_size)
message = repr(message)[2:-1]
if OPTIONS.data_type == 'random':
data = repr(src.read(OPTIONS.data_size))
else: