Open In App
Related Articles

Python | Arrange the files in directories according to extensions

Improve Article
Improve
Save Article
Save
Like Article
Like

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

Whether you're preparing for your first job interview or aiming to upskill in this ever-evolving tech landscape, GeeksforGeeks Courses are your key to success. We provide top-quality content at affordable prices, all geared towards accelerating your growth in a time-bound manner. Join the millions we've already empowered, and we're here to do the same for you. Don't miss out - check it out now!

Last Updated : 09 Sep, 2019
Like Article
Save Article
Previous
Next
Similar Reads
Complete Tutorials