Go Premium for a chance to win a PS4. Enter to Win

x
Solved

Python - Script to find if subfolder exists or not

Posted on 2009-05-13
Medium Priority
1,193 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 2000 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

Variable is a place holder or reserved memory locations to store any value. Which means whenever we create a variable, indirectly we are reserving some space in the memory. The interpreter assigns or allocates some space in the memory based on the d…
Dictionaries contain key:value pairs. Which means a collection of tuples with an attribute name and an assigned value to it. The semicolon present in between each key and values and attribute with values are delimited with a comma.  In python we can…
Learn the basics of strings in Python: declaration, operations, indices, and slicing. Strings are declared with quotations; for example: s = "string": Strings are immutable.: Strings may be concatenated or multiplied using the addition and multiplic…
Learn the basics of if, else, and elif statements in Python 2.7. Use "if" statements to test a specified condition.: The structure of an if statement is as follows: (CODE) Use "else" statements to allow the execution of an alternative, if the …
Suggested Courses
Course of the Month8 days, 4 hours left to enroll