Python | Arrange the files in directories according to extensions

In this article, we will learn how to arrange the files in different directories according to their extensions. When you have a lot of files with different extension than you can simply use the script for your help.

Below is the Python implementation –

import os  # Using system functions 
  
import shutil # Moving file to different folders. 
  
# Changing the folder location. 
os.chdir(os.getcwd()) 
cwd = os.getcwd() 
  
# For listing files from the folder. 
  
l = [f for f in os.listdir(cwd) if os.path.isfile(f)]  
l2 = [] 
  
# Get the extension of file from list l. 
  
for  value in l: 
    s = value.split('.')[1]  
    l2.append(s) 
print(l, l2) 
  
# We remove duplicate values from  
# list l2 and checks if directory  
# exists otherwise we create new one 
  
for extension in set(l2): 
    dirname = extension  
    if os.path.exists(cwd+'\'+extension): 
        pass
    else: 
        os.makedirs(dirname) 
  
# We use zip function and list l and 
# l2 passed as arguments. 
# If extension in file is same and 
# file not exists then we move the file. 
  
for files, extension in zip(l, l2): 
    if extension in files: 
        if os.path.exists(cwd+'\'+extension+'\'+files): 
            pass
        else: 
            shutil.move(cwd+'\'+files, cwd+'\'+extension) 
        print(extension, files) 
    else : 
        print('error') 

Files Not Arranged:

Files Arranged:

Output:

['arrange.py', 'mydata.pdf', 'python.exe', 'resume.doc'] ['py', 'pdf', 'exe', 'doc']
py arrange.py
pdf mydata.pdf
exe python.exe
doc resume.doc

My Personal Notes

Recommended Posts: