summaryrefslogtreecommitdiff
path: root/scripts/ln_atr503_to_subset.py
blob: bb3c4d25284f278fb499034166359e6d3136d481 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
import argparse
from pathlib import Path

parser = argparse.ArgumentParser()
parser.add_argument('input', type=Path)
parser.add_argument('output', type=Path)
parser.add_argument('--prefix', default='')
argument = parser.parse_args()

input = argument.input  # type: Path
output = argument.output  # type: Path

paths = list(sorted(input.glob('*'), key=lambda p: int(''.join(filter(str.isdigit, p.name)))))
assert len(paths) == 503

output.mkdir(exist_ok=True)

names = ['{}{:02d}'.format(s, n + 1) for s in 'ABCDEFGHIJ' for n in range(50)]
names += ['J51', 'J52', 'J53']

for p, n in zip(paths, names):
    out = output / (argument.prefix + n + p.suffix)
    out.symlink_to(p)