Python | Arrange the files in directories according to extensions

Last Updated : 09 Sep, 2019

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

Attention geek! Strengthen your foundations with the Python Programming Foundation Course and learn the basics.

To begin with, your interview preparations Enhance your Data Structures concepts with the Python DS Course.

My Personal Notes

Related Articles