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.
49 lines
1.1 KiB
49 lines
1.1 KiB
4 years ago
|
import os
|
||
|
|
import sys
|
||
|
|
import shutil
|
||
|
|
import argparse
|
||
|
|
from ext import foldername
|
||
|
|
|
||
|
|
FILE_NAME = "organize.py"
|
||
|
|
EXT_NAME = "ext.py"
|
||
|
|
|
||
|
|
|
||
|
|
def organize_files(path):
|
||
|
|
path = path+'\\'
|
||
|
|
if not os.path.exists(path):
|
||
|
|
print("ERROR! Invalid location")
|
||
|
|
return
|
||
|
|
files = os.listdir(path)
|
||
|
|
extns = set([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(path+folder):
|
||
|
|
os.makedirs(path+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 = path+file
|
||
|
|
dest = 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}")
|
||
|
|
|
||
|
|
|
||
|
|
try:
|
||
|
|
location = sys.argv[1]
|
||
|
|
organize_files(location)
|
||
|
|
except Exception:
|
||
|
|
print("USAGE: python organize.py <location>")
|