Solved

# Python - Script to find if subfolder exists or not

Posted on 2009-05-13
1,169 Views
I'm new to python.

I need to run a script which will delete files/directorys within a subfolder structure only if certain directorys exist.

Parent directory (p:\*fund*\) could be any directory name, for example:
p:\abc\letters\reject\
p:\def\letters\printed\
p:\ghi\letters\

So in above scenario, need to search the root directory (p:\*.*\letters\) for directorys called either reject or printed, then if exist delete the file contents of reject or printed.

I have many task like this one to do, hope that someone can provide me one example then I can tinker with it from then on.

Any help would be apprciated!!

0
Question by:Rich55555

LVL 3

Expert Comment

ID: 24381019
Running the python code at the end of this post you will delete all subfolders that their root have names included in DIRS_TO_DEL and are under ROOT_DIR as you asked.

Remove the # in last line from #shutil.rmtree(td) to actually delete the subfolders (else it only prints them to the console)

TEST:

Beggining with a dir structure like this (plus some files):

Directory of C:\area51
New Text Document (2).txt
New Text Document (3).txt
New Text Document.txt

Directory of C:\area51\printed
New Text Document (2).txt
New Text Document.txt

Directory of C:\area51\printed\a
New Text Document.txt

Directory of C:\area51\printed\b

Directory of C:\area51\printed\c

Directory of C:\area51\printed\reject

Directory of C:\area51\test

Directory of C:\area51\test\a

Directory of C:\area51\test\a\reject

Directory of C:\area51\test\b

Directory of C:\area51\test\printed

Directory of C:\area51\test\reject

Directory of C:\area51\test\reject\a

and The end result is this:

Directory of C:\area51

New Text Document (2).txt
New Text Document (3).txt
New Text Document.txt

Directory of C:\area51\test

Directory of C:\area51\test\a

Directory of C:\area51\test\b
#reject.py
import os
import shutil

ROOT_DIR = 'c:\\area51'
DIRS_TO_DEL = ['printed', 'reject']

todel=[]

for p,d,f in os.walk(ROOT_DIR):
for dtd in DIRS_TO_DEL:
try:
d.remove(dtd)
todel.append(os.path.join(p,dtd))
except:
pass

for td in todel:
print td
#shutil.rmtree(td)

0

LVL 29

Accepted Solution

pepr earned 500 total points
ID: 24382618
In my opinion, it is better to do the second loop through the found-subdirectories list (they exist) and only then to check then against the list of directories that should be deleted. The above solution also does not solve the first part of the question.

See the snippet below. Make a test directory, copy the attached makeTestDirs.py to it and save the snippet as a.py. Then run the makeTestDirs.py and then a.py twice. You can observe the following in the cmd window:

======================================================
c:\tmp\___python\Rich55555>makeTestDirs.py

c:\tmp\___python\Rich55555>a.py

from A\B1
remove A\B1\letters

from A\B2
remove A\B2\letters

from A\C3
remove A\C3\gossips
remove A\C3\letters

from A\letters

c:\tmp\___python\Rich55555>a.py

from A\B1

from A\B2

from A\C3

from A\letters

c:\tmp\___python\Rich55555>

======================================================

You can see that there is nothing to remove after the first run.

import glob
import os
import shutil

# The following generator is a part of the solution.
# It takes one root directory (the first argument,
# could be relative), and returns the wanted
# subdirectories (the other arguments) joined with the root.
#
def myWalker(root, *subdirs):
for dirpath, dirnames, filenames in os.walk(root):
for d in dirnames:
if d in subdirs:
yield os.path.join(dirpath, d)

if __name__ == '__main__':

# Get the list of roots and pass each of them to the generator
# if they are directories. Here the glob module is probably
# the easies to form such list.
#
for root in glob.glob('A/*'):
if os.path.isdir(root):
print '\nfrom', root
for directory in myWalker(root, 'letters', 'gossips'):
print 'remove', directory

# Use the same approach for removing the real removing the directories.
for root in glob.glob('A/*'):
if os.path.isdir(root):
for directory in myWalker(root, 'letters', 'gossips'):
shutil.rmtree(directory)


makeTestDirs.py.txt
0

## Featured Post

Question has a verified solution.

If you are experiencing a similar issue, please ask a related question

### Suggested Solutions

This article will show the steps for installing Python on Ubuntu Operating System. I have created a virtual machine with Ubuntu Operating system 8.10 and this installing process also works with upgraded version of Ubuntu OS. For installing Py…
Introduction of Lists in Python: There are six built-in types of sequences. Lists and tuples are the most common one. In this article we will see how to use Lists in python and how we can utilize it while doing our own program. In general we can al…
Learn the basics of lists in Python. Lists, as their name suggests, are a means for ordering and storing values. : Lists are declared using brackets; for example: t = [1, 2, 3]: Lists may contain a mix of data types; for example: t = ['string', 1, T…
Learn the basics of modules and packages in Python. Every Python file is a module, ending in the suffix: .py: Modules are a collection of functions and variables.: Packages are a collection of modules.: Module functions and variables are accessed us…