sort into folders

Python/dpt.py

@@ -0,0 +1,100 @@
+#!/bin/python3
+
+import sys
+import getopt
+import os
+from pathlib import Path
+import subprocess
+
+ARCHIVE_TYPES = [
+	".zip",
+	".rar",
+	".7z"
+]
+MAX_OUTPUT_LEN = 150
+
+input_path = ""
+do_delete = False
+
+def usage(name):
+	print('''{script} [options]
+	-i <path>  : specify input dir
+	-d         : do delete
+	-h         : print help and quit'''\
+	.format(script=name))
+
+def opts(argv):
+	global input_path, do_delete
+	try:
+		opts, args = getopt.getopt(argv[1:], 'i:dh')
+	except getopt.GetoptError as e:
+			print("Unkown option. Quiting.")
+			exit(1)
+	for opt, arg in opts:
+		if opt == '-i':
+			input_path = arg
+		elif opt == '-d':
+			do_delete = True
+		elif opt == '-h':
+			usage(argv[0])
+			exit(0)
+
+def is_password_protected(file):
+	command = ['7z', 't', '-p', file]
+	process = subprocess.run(command, stdout=subprocess.PIPE, stderr=subprocess.DEVNULL)
+	try:
+		process.wait()
+	except:
+		pass
+	return 0 != process.returncode
+
+def extract(file):
+	if is_password_protected(file):
+		#print('Ignoring ', file, '.', sep='')
+		return
+	global do_delete
+	output_dir = str(file.parent) + '/' + str(file.stem)
+	try:
+		os.mkdir(output_dir)
+	except:
+		pass
+	output_option = '-o' + str(output_dir)
+	command = ['7z', 'x', file, output_option]
+	try:
+		print('Extracting: ', file, '... ', sep='', end='')
+		process = subprocess.Popen(command, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL, stdin=subprocess.PIPE)
+		process.communicate(input='Y\n'.encode())
+		process.wait()
+		if process.returncode == 0:
+			print('Success.')
+			if do_delete:
+				os.remove(file)
+		else:
+			print('Fail.')
+	except:
+		print('\nError while forking 7z. Is it installed correctly?')
+
+def scandir(dir_):
+	#print('scandir', dir_)
+	global MAX_OUTPUT_LEN
+	global ARCHIVE_TYPES
+	if len(str(dir_)) >= MAX_OUTPUT_LEN:
+		return
+	for entry in Path(dir_).iterdir():
+		if entry.is_dir():
+			scandir(entry)
+		else:
+			for t in ARCHIVE_TYPES:
+				if t == entry.suffix:
+					extract(entry)
+
+def main(argv):
+	global input_path, do_delete
+	opts(argv)
+	if input_path == '':
+		print("No input directory specified. Quiting.")
+		exit(1)
+	scandir(input_path)
+
+if __name__ == "__main__":
+	main(sys.argv)