newb question: file searching

[jaysherby]

I'm new at Python and I need a little advice. Part of the script I'm
trying to write needs to be aware of all the files of a certain
extension in the script's path and all sub-directories. Can someone
set me on the right path to what modules and calls to use to do that?
You'd think that it would be a fairly simple proposition, but I can't
find examples anywhere. Thanks.

[hiaips]

Oops, what I wrote above isn't quite correct. As another poster pointed
out, you'd want to do
for file in x[2]:
....

--dave

[jaysherby]

So far, so good. I just have a few more questions.

Here's my code so far:

import os
top = '/home/USERNAME/'
images = []
for files in os.walk(top, topdown=True):
images += [file for file in files[2] if file.endswith(' jpg')]
print images

I still need to know how I can dynamically get the name of the
directory that my script is in. Also, how can I get it to add the full
path of the file to "images" instead of just the file name. See, when
I go to use a particular image name later on, it won't do me much good
if I don't have the path to it.

[Simon Forman]

ja*******@gmail .com wrote:

hiaips wrote:

ja*******@gmail .com wrote:
I'm new at Python and I need a little advice. Part of the script I'm
trying to write needs to be aware of all the files of a certain
extension in the script's path and all sub-directories. Can someone
set me on the right path to what modules and calls to use to do that?
You'd think that it would be a fairly simple proposition, but I can't
find examples anywhere. Thanks.

dir_name = 'mydirectory'
extension = 'my extension'
import os
files = os.listdir(dir_ name)
files_with_ext = [file for file in files if file.endswith(e xtension)]

That will only do the top level (not subdirectories) , but you can use
the os.walk procedure (or some of the other procedures in the os and
os.path modules) to do that.

--Dave

Thanks, Dave. That's exactly what I was looking for, well, except for
a few small alterations I'll make to achieve the desired effect. I
must ask, in the interest of learning, what is

[file for file in files if file.endswith(e xtension)]

actually doing? I know that 'file' is a type, but what's with the set
up and the brackets and all? Can someone run down the syntax for me on
that? And also, I'm still not sure I know exactly how os.walk() works.
And, finally, the python docs all note that symbols like . and ..
don't work with these commands. How can I grab the directory that my
script is residing in?

Thanks.

'file' *is* the name of the file type, so one shouldn't reuse it as the
name of a variable as Dave did in his example.

the brackets are a "list comprehension" and they work like so:

files_with_ext = []
for filename in files:
if filename.endswi th(extension):
files_with_ext. append(filename )

The above is exactly equivalent to the [filename for filename in
files... ] form.

AFAIK, os.listdir('.') works fine, but you can also use
os.listdir(os.g etcwd()). However, both of those return the current
directory, which can be different than the directory your script's
residing in if you've called os.chdir() or if you start your script
from a different directory.
HTH.

(Also, if you're not already, be aware of os.path.isfile( ) and
os.path.isdir() . They should probably be helpful to you.)

Peace,
~Simon

[Steve M]

ja*******@gmail .com wrote:

I must ask, in the interest of learning, what is

[file for file in files if file.endswith(e xtension)]

actually doing? I know that 'file' is a type, but what's with the set
up and the brackets and all?

Other people explained the list comprehension, but you might be
confused about the unfortunate choice of 'file' as the name in this
example. 'file' is a built-in as you remarked. It is allowed, but a bad
idea, to use names that are the same as built-ins. Some people
characterize this as shadowing the built-in.

A similar, common case is when people use the name 'list' for a list
object. They shouldn't.

The example could have been written as:

[f for f in files if f.endswith(exte nsion)]

[jaysherby]

Okay. This is almost solved. I just need to know how to have each
entry in my final list have the full path, not just the file name.
Also, I've noticed that files are being found within hidden
directories. I'd like to exclude hidden directories from the walk, or
at least not add anything within them. Any advice?

Here's what I've got so far:

def getFileList():
import os
imageList = []
for files in os.walk(os.getc wd(), topdown=True):
imageList += [file for file in files[2] if file.endswith(' jpg') or
file.endswith(' gif')]
return imageList

[Gabriel Genellina]

At Tuesday 8/8/2006 17:55, ja*******@gmail .com wrote:

>I must ask, in the interest of learning, what is

[file for file in files if file.endswith(e xtension)]

actually doing? I know that 'file' is a type, but what's with the set
up and the brackets and all? Can someone run down the syntax for me on
that?

Note: "file" is really a bad name here; rebinding builtins is never a
good idea. But unfortunately names like "file", "dir", "dict" are
very handy when you need a dummy variable like this, so people tend
to use them.
This has exactly the same meaning:

[x for x in files if x.endswith(exte nsion)]

>I still need to know how I can dynamically get the name of the
directory that my script is in.

os.path.dirname (os.path.abspat h(sys.modules[__name__].__file__))

>Also, how can I get it to add the full
path of the file to "images" instead of just the file name. See, when
I go to use a particular image name later on, it won't do me much good
if I don't have the path to it.

Look at os.path.join()

Gabriel Genellina
Softlab SRL

_______________ _______________ _______________ _____
Preguntá. Respondé. Descubrí.
Todo lo que querías saber, y lo que ni imaginabas,
está en Yahoo! Respuestas (Beta).
¡Probalo ya!
http://www.yahoo.com.ar/respuestas

[John Machin]

ja*******@gmail .com wrote:

And, finally, the python docs all note that symbols like . and ..
don't work with these commands.

Where did you read that? The docs for os.listdir do say that '.' and
'..' are (sensibly) not returned as *results*. AFAIK there is nothing
to stop you using '.' or '..' as a path *argument* where
appropriate/useful.

How can I grab the directory that my
script is residing in?

C:\junk>type scriptdir.py
import sys, os.path
scriptpath = sys.argv[0] # read the docs for sys.argv
print "script is", scriptpath
scriptdir = os.path.split(s criptpath)[0] # read the docs for
os.path.split
print "script lives in", scriptdir
C:\junk>scriptd ir.py
script is C:\junk\scriptd ir.py
script lives in C:\junk

C:\junk>move scriptdir.py \bin

C:\junk>scriptd ir.py
script is c:\bin\scriptdi r.py
script lives in c:\bin

[Curiosity: drive letter is upper-case in first example, lower in
second]

HTH,
John

[Steve M]

ja*******@gmail .com wrote:

Okay. This is almost solved. I just need to know how to have each
entry in my final list have the full path, not just the file name.

from http://docs.python.org/lib/os-file-dir.html:

walk() generates the file names in a directory tree, by walking the
tree either top down or bottom up. For each directory in the tree
rooted at directory top (including top itself), it yields a 3-tuple
(dirpath, dirnames, filenames).

dirpath is a string, the path to the directory. dirnames is a list of
the names of the subdirectories in dirpath (excluding '.' and '..').
filenames is a list of the names of the non-directory files in dirpath.
Note that the names in the lists contain no path components. To get a
full path (which begins with top) to a file or directory in dirpath, do
os.path.join(di rpath, name).

--------

So walk yields a 3-tuple, not just a filename. You seem to be somewhat
aware of this where you refer to files[2] in your list comprehension,
but I believe that is not constructed correctly.

Try this (untested):

def get_image_filep aths(target_fol der):
"""Return a list of filepaths (path plus filename) for all images
in target_folder or any subfolder"""
import os
images = []
for dirpath, dirnames, filenames in os.walk(target_ folder):
for filename in filenames:
normalized_file name = filename.lower( )
if normalized_file name.endswith(' .jpg') or
normalized_file name.endswith(' .gif'):
filepath = os.path.join(di rpath, filename)
images.append(f ilepath)
return images

import os
images = get_image_filep aths(os.getcwd( ))

Also, I've noticed that files are being found within hidden
directories. I'd like to exclude hidden directories from the walk, or
at least not add anything within them. Any advice?

Decide how you identify a hidden directory and test dirpath before
adding it to the images list. E.g., test whether dirpath starts with
'.' and skip it if so.

>
Here's what I've got so far:

def getFileList():
import os
imageList = []
for files in os.walk(os.getc wd(), topdown=True):
imageList += [file for file in files[2] if file.endswith(' jpg') or
file.endswith(' gif')]
return imageList

[infidel]

Also, I've noticed that files are being found within hidden

directories. I'd like to exclude hidden directories from the walk, or
at least not add anything within them. Any advice?

The second item in the tuple yielded by os.walk() is a list of
subdirectories inside the directory indicated by the first item in the
tuple. You can remove values from this list at runtime to have os.walk
skip over them.

[thebjorn]

ja*******@gmail .com wrote:
[...]

that? And also, I'm still not sure I know exactly how os.walk() works.
And, finally, the python docs all note that symbols like . and ..
don't work with these commands. How can I grab the directory that my
script is residing in?

os.getcwd() will get you the directory your script is in (at least as
long as you're running the script from the current directory :-)

Here's my version of how to it (comments below):

def collect_ext(*ex tensions):
"""Return a list of files from current directory and downwards
that have given extensions. Call it like collect_ext('.p y',
'.pyw')
"""
res = []
for dirname, _, files in os.walk(os.getc wd()):
for fname in files:
name, ext = os.path.splitex t(fname)
if ext in extensions:
res.append(os.p ath.join(dirnam e, fname))
return res

Each time around the outer for-loop, os.walk() gives us all the
filenames (as a list into the files variable) in a sub-directory. We
need to run through this list (inner for-loop), and save all files with
one of the right extensions. (os.walk() also gives us a list of all
subdirectories, but since we don't need it, I map it to the _ (single
underscore) variable which is convention for "I'm not using this
part").

note1: notice the "*" in the def line, it makes the function accept a
variable number of arguments, so you can call it as collect_ext('.p y')
but also as collect_ext('.p y', '.pyw', '.pyo'). Inside the function,
what you've called it with is a list (or a tuple, I forget), which
means I can use the _in_ operator in the if test.

note2: I use os.path.join() to attach the directory name to the
filename before appending it to the result. It seems that might be
useful ;-)

hth,
-- bjorn