mirror of
https://github.com/rwinkhart/reflexify.git
synced 2026-08-28 04:46:39 -04:00
55 lines
2.4 KiB
Python
Executable File
55 lines
2.4 KiB
Python
Executable File
#!/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
|
|
|
|
|
|
def type_sort():
|
|
if file.__contains__('_Nat_'):
|
|
_type_folder = 'National/'
|
|
_type_identifier = '_Nat'
|
|
elif file.__contains__('_Sx_'):
|
|
_type_folder = 'Supercross/'
|
|
_type_identifier = '_Sx'
|
|
else:
|
|
_type_folder = 'Free Ride/'
|
|
_type_identifier = ''
|
|
_beta_slot_folder = sorting_dir + _type_folder + file.split('.')[0][-1] + '/' + \
|
|
file_details[0] + ' - ' + file_details[1] + '/'
|
|
Path(_beta_slot_folder).mkdir(mode=0o700, parents=True, exist_ok=True)
|
|
return _beta_slot_folder, _type_identifier
|
|
|
|
|
|
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, type_identifier = type_sort()
|
|
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, type_identifier = type_sort()
|
|
move(temp_folder + '/' + file, beta_slot_folder +
|
|
f"Beta{type_identifier}_Track_Slot_{file.split('.')[0][-1]}.dx9.{file.split('.')[-1]}")
|
|
rmtree(temp_folder)
|