M9 flexing

flex.py

#!/usr/bin/python
import os
import subprocess
import sys

root = os.path.expanduser(f'~/arknights')

size = {
    'm9': '375x115+1240+330',
    'p6': '375x180+1240+265',
}

def do_work(kind, rows, cols):
    path = f'{root}/{kind}'
    _, _, files = next(os.walk(path))
    files.sort()
    
    images = iter(files)
    args = ['magick']
    
    for i in range(rows):
        args.append('(')
        
        for j in range(cols):
            image = next(images)
            image = f'{path}/{image}'
            args.append(image)
            args.append('-crop')
            args.append(size[kind])
        
        args.append('+smush')
        args.append('0')
        args.append(')')
    
    args.append('-smush')
    args.append('0')
    args.append(f'{path}.jpg')
    subprocess.run(args)

if __name__ == '__main__':
    kind = sys.argv[1]
    rows = int(sys.argv[2])
    cols = int(sys.argv[3])
    do_work(kind, rows, cols)