#!/usr/bin/env python3
from os import listdir, scandir
from os.path import getsize
from pathlib import Path
from shutil import move, rmtree
from subprocess import run

archive_dir = input('archive location (full path): ').rstrip('/') + '/'
sorting_dir = input('target (sorted) directory (full path): ').rstrip('/') + '/'

file_list = sorted(listdir(archive_dir))

# ensure sorting location exists
Path(sorting_dir).mkdir(mode=0o700, parents=True, exist_ok=True)

for filename in file_list:
    if getsize(archive_dir + filename) > 4718592:
        file_details = filename.split(' -\x1f- ')
        temp_folder = sorting_dir + '/temp/' + file_details[0] + ' - ' + file_details[1]
        run(['7z', 'x', '-o' + temp_folder, archive_dir + filename])
        sub_folders = [f.path for f in scandir(temp_folder) if f.is_dir()]
        for subfolder in sub_folders:
            sub_files = [f for f in listdir(subfolder)]
            for file in sub_files:
                if file.endswith('.database') or file.endswith('.level') or file.endswith('.package') or \
                        file.endswith('.scene'):
                    beta_slot_folder = sorting_dir + file_details[-1] + '/' + file.split('.')[0][-1] + '/' +\
                        file_details[0] + ' - ' + file_details[1] + '/'
                    Path(beta_slot_folder).mkdir(mode=0o700, parents=True, exist_ok=True)
                    if file_details[-1] == 'National':
                        type_identifier = '_Nat'
                    elif file_details[-1] == 'Supercross':
                        type_identifier = '_Sx'
                    else:
                        type_identifier = ''
                    move(subfolder + '/' + file, beta_slot_folder +
                         f"Beta{type_identifier}_Track_Slot_{file.split('.')[0][-1]}.dx9.{file.split('.')[-1]}")
        files = [f for f in listdir(temp_folder)]
        for file in files:
            if file.endswith('.database') or file.endswith('.level') or file.endswith('.package') or \
                    file.endswith('.scene'):
                beta_slot_folder = sorting_dir + file_details[-1] + '/' + file.split('.')[0][-1] + '/' + \
                               file_details[0] + ' - ' + file_details[1] + '/'
                Path(beta_slot_folder).mkdir(mode=0o700, parents=True, exist_ok=True)
                if file_details[-1] == 'National':
                    type_identifier = '_Nat'
                elif file_details[-1] == 'Supercross':
                    type_identifier = '_Sx'
                else:
                    type_identifier = ''
                move(temp_folder + '/' + file, beta_slot_folder +
                     f"Beta{type_identifier}_Track_Slot_{file.split('.')[0][-1]}.dx9.{file.split('.')[-1]}")
        rmtree(temp_folder)