랜덤으로 파일 추출하는 파이썬 스크립트

# source directory 지정 target directory 지정

import os
import random
from shutil import copyfile

def main():
    print("select source directory")
    ldir = os.listdir()
    ldir = [l for l in ldir if os.path.isdir(l)]
    print(ldir)
    source = input()

    print("select target directory")
    target = input()

    ldir2 = os.listdir(source)
    print(f"There are {len(ldir2)} items in source directory")
    print("how many items do you want to select randomly?")

    nums = input()
    val = 0
    try:
        val = int(nums)
    except ValueError:
        print("Not number!")
        return
    print(f"You want to select {nums} in the directory")
    selected = random.sample(ldir2, val)

    print(f"{len(selected)} selected!!")
    print("press enter to see")
    input()
    print(selected)

    for fl in selected:
        copyfile(f"./{source}/{fl}", f"./{target}/{fl}")


if __name__ == '__main__':
    main()