You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
88 lines
2.5 KiB
88 lines
2.5 KiB
import os
|
|
import platform
|
|
import shutil
|
|
import sys
|
|
|
|
from ext import foldername
|
|
|
|
FILE_NAME = "organize.py"
|
|
EXT_NAME = "ext.py"
|
|
|
|
|
|
def organize_files(drive):
|
|
# check os
|
|
operating_system = platform.system()
|
|
if operating_system == 'Darwin': # Mac
|
|
drive = os.path.expanduser(drive)
|
|
if not os.path.exists(drive):
|
|
print(f"ERROR! {drive} is not a valid location")
|
|
return
|
|
files = os.listdir(drive)
|
|
extns = {os.path.splitext(file)[1].strip('.') for file in files}
|
|
# Create Folders
|
|
for ext in extns:
|
|
folder = foldername(ext)
|
|
if folder and not os.path.exists(os.path.join(drive, folder)):
|
|
os.makedirs(os.path.join(drive, folder))
|
|
|
|
# Move Files To Folders
|
|
for file in files:
|
|
if file in [FILE_NAME, EXT_NAME]:
|
|
continue
|
|
ext = os.path.splitext(file)[1].strip('.')
|
|
folder = foldername(ext)
|
|
if not folder:
|
|
continue
|
|
|
|
src = os.path.join(drive, file)
|
|
dest = os.path.join(drive, folder, file)
|
|
|
|
if not os.path.exists(dest):
|
|
shutil.move(src, dest)
|
|
print(f"Moved {file} to {folder}")
|
|
|
|
print(f"\nSUCCESS! All files organized in {drive}")
|
|
|
|
elif operating_system == 'Windows': # Windows
|
|
drive = drive + '\\'
|
|
if not os.path.exists(drive):
|
|
print("ERROR! Invalid location")
|
|
return
|
|
files = os.listdir(drive)
|
|
extns = {os.path.splitext(file)[1].strip('.') for file in files}
|
|
|
|
# Create Folders
|
|
for ext in extns:
|
|
folder = foldername(ext)
|
|
if folder and not os.path.exists(drive + folder):
|
|
os.makedirs(drive + folder)
|
|
|
|
# Move Files To Folders
|
|
for file in files:
|
|
if file in [FILE_NAME, EXT_NAME]:
|
|
continue
|
|
ext = os.path.splitext(file)[1].strip('.')
|
|
folder = foldername(ext)
|
|
if not folder:
|
|
continue
|
|
|
|
src = drive + file
|
|
dest = drive + folder + '/' + file
|
|
|
|
if not os.path.exists(dest):
|
|
shutil.move(src, dest)
|
|
print(f"Moved {file} to {folder}")
|
|
|
|
print(f"\nSUCCESS! All files organized in {drive}")
|
|
else:
|
|
print(f'{operating_system} not currently supported')
|
|
|
|
|
|
if __name__ == "__main__":
|
|
try:
|
|
location = sys.argv[1]
|
|
organize_files(location)
|
|
except Exception as e:
|
|
print("USAGE: python organize.py <location>")
|
|
print(e)
|