]> git.ipfire.org Git - thirdparty/zstd.git/commitdiff
remove error message for Ctrl+C
authorYann Collet <cyan@fb.com>
Mon, 13 Aug 2018 18:48:25 +0000 (11:48 -0700)
committerYann Collet <cyan@fb.com>
Mon, 13 Aug 2018 18:48:25 +0000 (11:48 -0700)
tests/rateLimiter.py

index 15222e0166953a9a930d10538a0e2653032a2fbd..d69523ca3b255d034c2e241d5623151cb2240cde 100755 (executable)
@@ -23,20 +23,26 @@ rate *= 1.4   # compensation for excluding i/o time (experimentally determined)
 start = time.time()
 total_read = 0
 
-buf = " "
-while len(buf):
-  now = time.time()
-  to_read = max(int(rate * (now - start) - total_read), 1)
-  max_buf_size = 1 * MB
-  to_read = min(to_read, max_buf_size)
-
-  read_start = time.time()
-  buf = sys.stdin.buffer.read(to_read)
-
-  write_start = read_end = time.time()
-  sys.stdout.buffer.write(buf)
-  write_end = time.time()
-
-  wait_time = max(read_end - read_start, write_end - write_start)
-  start += wait_time   # exclude delay of the slowest
-  total_read += len(buf)
+sys.stderr.close()  # remove error message, for Ctrl+C
+
+try:
+  buf = " "
+  while len(buf):
+    now = time.time()
+    to_read = max(int(rate * (now - start) - total_read), 1)
+    max_buf_size = 1 * MB
+    to_read = min(to_read, max_buf_size)
+
+    read_start = time.time()
+    buf = sys.stdin.buffer.read(to_read)
+
+    write_start = read_end = time.time()
+    sys.stdout.buffer.write(buf)
+    write_end = time.time()
+
+    wait_time = max(read_end - read_start, write_end - write_start)
+    start += wait_time   # exclude delay of the slowest
+    total_read += len(buf)
+
+except (KeyboardInterrupt, BrokenPipeError) as e:
+    pass