Skip to content

Commit

Permalink
Refactoring Organise.py
Browse files Browse the repository at this point in the history
  • Loading branch information
@Vegas007
Vegas007 authored Feb 13, 2018
1 parent f9573d0 commit 4c69f54
Showing 1 changed file with 40 additions and 178 deletions.
218 changes: 40 additions & 178 deletions Organise.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -2,193 +2,55 @@
import sys
import shutil

Music = ['MP3', 'WAV', 'WMA', 'MKA', 'AAC', 'MID', 'RA', 'RAM', 'RM', 'OGG']
Codes = ['CPP', 'RB', 'PY', 'HTML', 'CSS', 'JS']
Compressed = ['RAR', 'JAR', 'ZIP', 'TAR', 'MAR', 'ISO', 'LZ', '7ZIP', 'TGZ', 'GZ', 'BZ2']
Documents = ['DOC', 'DOCX', 'PPT', 'PPTX', 'PAGES', 'PDF', 'ODT', 'ODP', 'XLSX', 'XLS', 'ODS', 'TXT', 'IN', 'OUT', 'MD']
Images = ['JPG', 'JPEG', 'GIF', 'PNG', 'SVG']
Executables = ['LNK','DEB', 'EXE', 'SH', 'BUNDLE']
Video = ['FLV', 'WMV', 'MOV', 'MP4', 'MPEG', '3GP', 'MKV','AVI']


def getVideo():
return Video

def getMusic():
return Music

def getCodes():
return Codes

def getCompressed():
return Compressed

def getImages():
return Images

def getExe():
return Executables

def getDoc():
return Documents

# taking the location of the Folder to Arrange
EXT_VIDEO_LIST = ['FLV', 'WMV', 'MOV', 'MP4', 'MPEG', '3GP', 'MKV','AVI']
EXT_IMAGE_LIST = ['JPG', 'JPEG', 'GIF', 'PNG', 'SVG']
EXT_DOCUMENT_LIST = ['DOC', 'DOCX', 'PPT', 'PPTX', 'PAGES', 'PDF', 'ODT', 'ODP', 'XLSX', 'XLS', 'ODS', 'TXT', 'IN', 'OUT', 'MD']
EXT_MUSIC_LIST = ['MP3', 'WAV', 'WMA', 'MKA', 'AAC', 'MID', 'RA', 'RAM', 'RM', 'OGG']
EXT_CODE_LIST = ['CPP', 'RB', 'PY', 'HTML', 'CSS', 'JS']
EXT_EXECUTABLE_LIST = ['LNK','DEB', 'EXE', 'SH', 'BUNDLE']
EXT_COMPRESSED_LIST = ['RAR', 'JAR', 'ZIP', 'TAR', 'MAR', 'ISO', 'LZ', '7ZIP', 'TGZ', 'GZ', 'BZ2']

# Taking the location of the Folder to Arrange
try:
arrange_dir = str(sys.argv[1])
destLocation = str(sys.argv[1])
except IndexError:
arrange_dir = str(input("Enter the Path of directory: "))
destLocation = str(input('Enter the Path of directory: '))

# when we make a folder that already exist then WindowsError happen
# changing directory may give WindowsError

def change(direc):
# When we make a folder that already exist then WindowsError happen
# Changing directory may give WindowsError
def ChangeDirectory(dir):
try:
os.chdir(direc)
#print ("path changed")
os.chdir(dir)
except WindowsError:
print ("Error! Cannot change the Directory")
print ("Enter a valid directory!")
direc = str(input("Enter the Path of directory: "))
change(direc)

change(arrange_dir)

# now we will get the list of all the directories in the folder

list_dir = os.listdir(os.getcwd())

#print (list_dir)
print ('Error! Cannot change the Directory')
print ('Enter a valid directory!')
ChangeDirectory(str(input('Enter the Path of directory: ')))
ChangeDirectory(destLocation)

#check_Folder = False # for organising Folders
check_Music = False
check_Video = False
check_Exe = False
check_Code = False
check_Compressed = False
check_Img = False
check_Docs = False
def Organize(dirs, name):
try:
os.mkdir(name)
print ('{} Folder Created'.format(name))
except WindowsError:
print ('{} Folder Exist'.format(name))

src = '{}\\{}'.format(destLocation, dirs)
dest = '{}\{}'.format(destLocation, name)

os.chdir(dest)
shutil.move(src, '{}\\{}'.format(dest, dirs))

main_names = ['Video','Folders','Images','Documents','Music','Codes','Executables','Compressed']
print os.getcwd()
os.chdir(destLocation)

for name in list_dir:
#print (name.split('.'))
TYPES_LIST = ['Video', 'Images', 'Documents', 'Music', 'Codes', 'Executables', 'Compressed']
for dirs in os.listdir(os.getcwd()):
if 1:

if name.split('.')[-1].upper() in getVideo():
try:
os.mkdir("Video")
print ("Video Folder Created")
except WindowsError:
print ("Images Folder Exists")

old_dir = arrange_dir + "\\" + name
new_dir = arrange_dir + "\Video"
os.chdir(new_dir)
shutil.move(old_dir, new_dir + "\\" + name)
print os.getcwd()
os.chdir(arrange_dir)
#print ("It is a folder")
elif name.split('.')[-1].upper() in getImages():
try:
os.mkdir("Images")
print ("Images Folder Created")
except WindowsError:
print ("Images Folder Exists")

old_dir = arrange_dir + "\\" + name
new_dir = arrange_dir + "\Images"
os.chdir(new_dir)
shutil.move(old_dir, new_dir + "\\" + name)
print (os.getcwd())
os.chdir(arrange_dir)
#print ("It is a folder")
elif name.split('.')[-1].upper() in getMusic():
try:
os.mkdir("Music")
print ("Music Folder Created")
except WindowsError:
print ("Music Folder Exists")

old_dir = arrange_dir + "\\" + name
new_dir = arrange_dir + "\Music"
os.chdir(new_dir)
shutil.move(old_dir, new_dir + "\\" + name)
print (os.getcwd())
os.chdir(arrange_dir)
#print ("It is a folder")
elif name.split('.')[-1].upper() in getDoc():
try:
os.mkdir("Documents")
print ("Documents Folder Created")
except WindowsError:
print ("Documents Folder Exists")

old_dir = arrange_dir + "\\" + name
new_dir = arrange_dir + "\Documents"
os.chdir(new_dir)
shutil.move(old_dir, new_dir + "\\" + name)
print (os.getcwd())
os.chdir(arrange_dir)
#print ("It is a folder")
elif name.split('.')[-1].upper() in getCodes():
try:
os.mkdir("Codes")
print ("Codes Folder Created")
except WindowsError:
print ("Codes Folder Exists")

old_dir = arrange_dir + "\\" + name
new_dir = arrange_dir + "\Codes"
os.chdir(new_dir)
shutil.move(old_dir, new_dir + "\\" + name)
print (os.getcwd())
os.chdir(arrange_dir)
#print ("It is a folder")
elif name.split('.')[-1].upper() in getCompressed():
try:
os.mkdir("Compressed")
print ("Compressed Folder Created")
except WindowsError:
print ("Compressed Folder Exists")

old_dir = arrange_dir + "\\" + name
new_dir = arrange_dir + "\Compressed"
os.chdir(new_dir)
shutil.move(old_dir, new_dir + "\\" + name)
print (os.getcwd())
os.chdir(arrange_dir)
#print ("It is a folder")
elif name.split('.')[-1].upper() in getExe():
try:
os.mkdir("Executables")
print ("Executables Folder Created")
except WindowsError:
print ("Executables Folder Exists")

old_dir = arrange_dir + "\\" + name
new_dir = arrange_dir + "\Executables"
os.chdir(new_dir)
shutil.move(old_dir, new_dir + "\\" + name)
print (os.getcwd())
os.chdir(arrange_dir)
#print ("It is a folder")
#if file type is not identified
for name, extensions_list in zip(TYPES_LIST, [EXT_VIDEO_LIST, EXT_IMAGE_LIST, EXT_DOCUMENT_LIST, EXT_MUSIC_LIST, EXT_CODE_LIST, EXT_EXECUTABLE_LIST, EXT_COMPRESSED_LIST]):
if dirs.split('.')[-1].upper() in extensions_list:
Organize(dirs, name)
else:
if name not in main_names:
try:
os.mkdir("Folders")
print ("Folders Folder Created")
except WindowsError:
print ("Folders Folder Exists")

old_dir = arrange_dir + "\\" + name
new_dir = arrange_dir + "\Folders"
os.chdir(new_dir)
shutil.move(old_dir, new_dir + "\\" + name)
print (os.getcwd())
os.chdir(arrange_dir)


if dirs not in TYPES_LIST:
Organize(dirs, 'Folders')

print ("Done Arranging Files and Folder in your specified directory")
print ('Done Arranging Files and Folder in your specified directory')

0 comments on commit 4c69f54

Please sign in to comment.