Skip to content

Commit 479a2fc

Browse files
committed
feat: add --shuffle-files cli argument
1 parent 3923d2f commit 479a2fc

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

cutstitch/cli.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
import argparse
2+
import random
23

34
from PIL import Image
45

@@ -12,12 +13,15 @@ def main():
1213
parser.add_argument("--mode", choices=list(CUTOUT_MODES.keys()), default="vertical", help="cutout mode (default: vertical)")
1314
parser.add_argument("--angle", type=float, default=45, help="angle for diagonal mode")
1415
parser.add_argument("--output", default="stitched.png", help="output file name")
16+
parser.add_argument("--shuffle-files", type=bool, default=False, help="randomly shuffle input files before stitching")
1517
args = parser.parse_args()
1618

1719
if len(args.files) < 2:
1820
raise RuntimeError("Must provide at least two image files")
1921

2022
images = [Image.open(path).convert("RGBA") for path in args.files]
23+
if args.shuffle_files:
24+
random.shuffle(images)
2125
stitch_images(images, args.mode, angle=args.angle).save(args.output)
2226
print(f"Saved stitched image to {args.output}")
2327

0 commit comments

Comments
 (0)