dispatch.py

from shutil import move
from glob import glob
from os import path
from collections import Counter


HOME = '/home/evuez/'

RESERVED = ('QUEUED', 'TMP', 'DONE')

TORNT = ('.torrent',)
MUSIC = ('.mp3', '.ogg', '.wav')
MOVIE = ('.avi', '.mkv', '.mpg', '.mpeg')
IMAGE = ('.png', '.gif', '.jpg', '.jpeg', '.svg')
DCMNT = ('.pdf', '.doc', '.docx', '.ods', '.odp', '.odt', '.txt', '.epub', '.pptx')


def get_dir(f):
	if f.endswith(TORNT):
		return path.join(HOME, 'downloads/QUEUED')
	elif f.endswith(MUSIC):
		return path.join(HOME, 'music')
	elif f.endswith(MOVIE):
		return path.join(HOME, 'videos')
	elif f.endswith(IMAGE):
		return path.join(HOME, 'images')
	elif f.endswith(DCMNT):
		return path.join(HOME, 'documents')


for x in glob(path.join(HOME, 'downloads/*')):
	if path.isfile(x):
			new_dir = get_dir(x)
	elif path.isdir(x):
		filetypes = []
		if x.endswith(RESERVED):
			continue
		for _ in glob(path.join(x, '*')):
			filetypes.append(path.splitext(_)[1])
		new_dir = get_dir(Counter(filetypes).most_common(1)[0][0])
	if new_dir is not None:
		move(x, new_dir)