File Management in Python - Directories
(Page 3 of 5 )

 

Directories, like regular files, are easy to work with. Let's start by listing the contents of a directory: 

import os

for fileName in os.listdir ( '/' ):

   print fileName 

As you can see, this is extremely simple, and it can be done in three lines.

 

Creating a directory is also simple: 

import os

os.mkdir ( 'testDirectory' ) 

It is equally as easy to delete the directory we just created: 

import os

os.rmdir ( 'testDirectory ) 

We can also create multiple directories at a time: 

import os

os.makedirs ( 'I/will/show/you/how/deep/the/rabbit/hole/goes' ) 

Assuming we add nothing to the directories we just created, we can also delete them all at once: 

import os

os.removedirs ( 'I/will/show/you/how/deep/the/rabbit/hole/goes' ) 

Suppose we want to perform a specific action when a specific file type is reached. This can easily be done with the "fnmatch" module. Let's print the contents of all the ".txt" files we encounter and print the filename of any ".exe" files we encounter: 

import fnmatch

import os

for fileName in os.listdir ( '/' ):

   if fnmatch.fnmath ( fileName, '*.txt' ):

      print open ( fileName ).read()

   elif fnmatch.fnmatch ( fileName, '*.exe' ):

      print fileName 

The asterisk character can represent any amount of characters. If we want to match just one character, we can use the question mark: 

import fnmatch

import os

for fileName in os.listdir ( '/' ):

   if fnmatch.fnmatch ( fileName, '?.txt' ):

      print 'Text file.' 

It is also possible to create a regular expression using the "fnmatch" module, matching filenames with the "re" module:

 

import fnmatch

import os

import re

filePattern = fnmatch.translate ( '*.txt' )

for fileName in os.listdir ( '/' ):

   if re.match ( filePattern, fileName ):

      print 'Text file.'

 

If we're just looking for one type of filename, it is a lot easier to use the "glob" module. Its patterns are similar to those used in "fnmatch": 

import glob

for fileName in glob.glob ( '*.txt' ):

   print 'Text file.' 

It is also possible to use ranges of characters in the patterns, just as you would in regular expressions. Suppose you want to print the names of text files with one digit before the extension: 

import glob

for fileName in glob.glob ( '[0-9].txt' ):

   print fileName 

The "glob" module makes use of the "fnmatch" module.

 

Next: Pickling Data >>
 

More Python Articles
More By Peyton McCullough

Comments On: File Management in Python

· Hello!I wrote this article with a "less talk and more code" approach. Does...    · Good article, I like the code-heavy approach. Another module I find useful is...    · I like the more-code approach. A nice article.When you're starting out, it's...    · fileHandle = open ( 'usernames.txt','ab' )fileHandle.write (raw_input("Please...    · fileHandle.write ( raw_input ( 'Please enter user name to add: ' ) + '\n' )...    · Hey. I am a python newbie, and reading the "File Management" article helped me...    · following the nice file i/o tutorial here, i have only one remaining quesiton: how...    · The article may be good, but it does not explain how to avoid program exit when file...    · Be consistent with your backslashes, this will solve your problem!    · thats a more general issue of exception handling, IMHO    · I get errors when I try to run the first example in - Getting Information on...   >>> Post your comment now!