82 lines
2.5 KiB
Python
82 lines
2.5 KiB
Python
""" handles moving tv downloads """
|
|
|
|
import os
|
|
import subprocess
|
|
from time import sleep
|
|
|
|
|
|
def get_pending(tv_downpath):
|
|
""" return how many shows are pending """
|
|
pending = len(os.listdir(tv_downpath))
|
|
return pending
|
|
|
|
|
|
def move_to_sort(tv_downpath, sortpath, ext):
|
|
""" move tv files to sortpath """
|
|
for dirpath, _, filenames in os.walk(tv_downpath):
|
|
for filename in filenames:
|
|
path = os.path.join(dirpath, filename)
|
|
_, extenstion = os.path.splitext(path)
|
|
f_size = os.stat(path).st_size
|
|
if extenstion.lower() in ext and 'sample' not in filename and f_size > 50000000:
|
|
move_to = os.path.join(sortpath, filename)
|
|
os.rename(path, move_to)
|
|
if os.listdir(sortpath):
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
|
|
def move_to_archive(sortpath, tvpath):
|
|
""" moves the renamed files to the archive """
|
|
print()
|
|
for dirpath, _, filenames in os.walk(sortpath):
|
|
for show in sorted(filenames):
|
|
print(show)
|
|
input('\ncontinue?')
|
|
# apply
|
|
for dirpath, _, filenames in os.walk(sortpath):
|
|
for show in filenames:
|
|
old_file = os.path.join(sortpath, dirpath, show)
|
|
show_name = dirpath.split('/')[-2]
|
|
season_name = dirpath.split('/')[-1]
|
|
new_folder = os.path.join(tvpath, show_name, season_name)
|
|
new_file = os.path.join(new_folder, show)
|
|
os.makedirs(new_folder, exist_ok=True)
|
|
os.rename(old_file, new_file)
|
|
|
|
|
|
def clean_up(sortpath, tv_downpath):
|
|
""" clean up download and sort folder """
|
|
to_clean_list = os.listdir(sortpath)
|
|
for to_clean in to_clean_list:
|
|
to_trash = os.path.join(sortpath, to_clean)
|
|
subprocess.call(["trash", to_trash])
|
|
to_clean_list = os.listdir(tv_downpath)
|
|
for to_clean in to_clean_list:
|
|
to_trash = os.path.join(tv_downpath, to_clean)
|
|
subprocess.call(["trash", to_trash])
|
|
|
|
|
|
def main(config, tvsort_id):
|
|
""" main function to sort tv shows """
|
|
# parse config
|
|
tv_downpath = config['tv_downpath']
|
|
tvpath = config['tvpath']
|
|
sortpath = config['sortpath']
|
|
ext = config['ext']
|
|
# stop here if nothing to do
|
|
pending = get_pending(tv_downpath)
|
|
if not pending:
|
|
print('no tv shows to sort')
|
|
sleep(2)
|
|
return
|
|
|
|
# move files
|
|
to_sort = move_to_sort(tv_downpath, sortpath, ext)
|
|
if to_sort:
|
|
renamed = tvsort_id.episode_rename(config)
|
|
if renamed:
|
|
move_to_archive(sortpath, tvpath)
|
|
clean_up(sortpath, tv_downpath)
|