random_file_move.py

# Script Name   : random_file_move.py
# Author(s)     : Akash Jain
# Created       : 1 September 2020
# Last Modified : 1 September 2020
# Version       : 1.0
# Description   : This will move specified number of files(given in ratio) from the src directory to dest directory.


import os, random
import argparse


def check_ratio(x):
    try:
        x = float(x)
    except ValueError:
        raise argparse.ArgumentTypeError("%r not a floating-point literal" % (x,))

    if x < 0.0 or x > 1.0:
        raise argparse.ArgumentTypeError("%r not in range [0.0, 1.0]" % (x))
    return x


desc = "Script to move specified number of files(given in ratio) from the src directory to dest directory."
usage = "python random_file_move.py -src [SRC] -dest [DEST] -ratio [RATIO]"

parser = argparse.ArgumentParser(usage=usage, description=desc)
parser.add_argument(
    "-src",
    "--src",
    type=str,
    required=True,
    help="(REQUIRED) Path to directory from which we cut files. Space not allowed in path.",
)
parser.add_argument(
    "-dest",
    "--dest",
    type=str,
    required=True,
    help="(REQUIRED) Path to directory to which we move files. Space not allowed in path.",
)
parser.add_argument(
    "-ratio",
    "--ratio",
    type=check_ratio,
    required=True,
    help="(REQUIRED) Ratio of files in 'src' and 'dest' directory.",
)

args = parser.parse_args()

src = args.src
dest = args.dest
ratio = args.ratio

files = os.listdir(src)
size = int(ratio * len(files))

print("Move {} files from {} to {} ? [y/n]".format(size, src, dest))
if input().lower() == "y":
    for f in random.sample(files, size):
        try:
            os.rename(os.path.join(src, f), os.path.join(dest, f))
        except Exception as e:
            print(e)
    print("Successful")
else:
    print("Cancelled")