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.
53 lines
1.2 KiB
53 lines
1.2 KiB
import os
|
|
import shutil
|
|
import sys
|
|
|
|
from ext import foldername
|
|
|
|
FILE_NAME = "organize.py"
|
|
EXT_NAME = "ext.py"
|
|
|
|
|
|
def organize_files(path):
|
|
if not os.path.exists(path):
|
|
print("ERROR! Invalid location")
|
|
return
|
|
|
|
files = os.listdir(path)
|
|
extns = {os.path.splitext(file)[1].strip(".") for file in files}
|
|
|
|
# Create Folders
|
|
for ext in extns:
|
|
folder = foldername(ext) or ''
|
|
new = os.path.join(path, folder)
|
|
if folder and not os.path.exists(new):
|
|
os.makedirs(new)
|
|
|
|
# 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(path, file)
|
|
dest = os.path.join(path, 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 {path}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
try:
|
|
location = sys.argv[1]
|
|
organize_files(location)
|
|
except Exception as e:
|
|
print(f"Error: {e}")
|
|
print("USAGE: python organize.py <location>")
|