(self, block_size, sort_key=None)
| 24 | return self.block_filenames |
| 25 | |
| 26 | def split(self, block_size, sort_key=None): |
| 27 | i = 0 |
| 28 | with open(self.filename) as file: |
| 29 | while True: |
| 30 | lines = file.readlines(block_size) |
| 31 | |
| 32 | if lines == []: |
| 33 | break |
| 34 | |
| 35 | if sort_key is None: |
| 36 | lines.sort() |
| 37 | else: |
| 38 | lines.sort(key=sort_key) |
| 39 | |
| 40 | self.write_block("".join(lines), i) |
| 41 | i += 1 |
| 42 | |
| 43 | def cleanup(self): |
| 44 | map(os.remove, self.block_filenames) |