()
| 81 | os.mkdir(inputs_path) |
| 82 | |
| 83 | def stream_inputs(): |
| 84 | for i in range(n_files): |
| 85 | file_path = inputs_path / f"{i}.csv" |
| 86 | if data_format == "json": |
| 87 | payload = {"k": str(i), "v": i} |
| 88 | with open(file_path, "w") as streamed_file: |
| 89 | json.dump(payload, streamed_file) |
| 90 | elif data_format == "csv": |
| 91 | data = """ |
| 92 | k | v |
| 93 | {} | {} |
| 94 | """.format( |
| 95 | i, i |
| 96 | ) |
| 97 | write_csv(file_path, data) |
| 98 | elif data_format == "plaintext": |
| 99 | with open(file_path, "w") as f: |
| 100 | f.write(f"{i}") |
| 101 | else: |
| 102 | raise ValueError(f"Unknown format: {data_format}") |
| 103 | |
| 104 | time.sleep(stream_interval) |
| 105 | |
| 106 | inputs_thread = threading.Thread(target=stream_inputs, daemon=True) |
| 107 | inputs_thread.start() |
nothing calls this directly
no test coverage detected