Filter files by prefix and suffix.
(prefix, suffix, files, remove_prefix=None)
| 502 | |
| 503 | |
| 504 | def filter_files(prefix, suffix, files, remove_prefix=None): |
| 505 | """ |
| 506 | Filter files by prefix and suffix. |
| 507 | """ |
| 508 | filtered, rest = [], [] |
| 509 | match = re.compile(prefix + r'.*' + suffix + r'\Z').match |
| 510 | if remove_prefix: |
| 511 | ind = len(prefix) |
| 512 | else: |
| 513 | ind = 0 |
| 514 | for file in [x.strip() for x in files]: |
| 515 | if match(file): |
| 516 | filtered.append(file[ind:]) |
| 517 | else: |
| 518 | rest.append(file) |
| 519 | return filtered, rest |
| 520 | |
| 521 | |
| 522 | def get_prefix(module): |
no test coverage detected