Merge pull request geekcomputers#801 from akash720/master

Add random file mover
SKAUL05 · Sep 13, 2020 · 26233e3 · 26233e3
2 parents 3ce49cc + 6c5ad96
commit 26233e3
Showing 1 changed file with 51 additions and 0 deletions.
diff --git a/random_file_move.py b/random_file_move.py
@@ -0,0 +1,51 @@
+# 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')