[SOLVED] Search for a File Python

metallica1973 · 10-24-2013, 11:19 AM

I am back at it with Python and have run into a little stupid hurdle. My goal is to simply search for the GeoIP.dat database and add the path to a couple of variables. So for example:

Code:

geopath=os.system('find /usr/share -iname GeoIP.dat')
geobase = pygeoip.GeoIP(geopath, pygeoip.MEMORY_CACHE)

which will load the GeoIP database so I can begin adding addition logic to my script. The issue I am having is when I simply execute using ipython:

Code:

geopath=os.system('find /usr/share -iname GeoIP.dat')
geopath
Out[15]: 0

it returns "0" in which I assume mean successful. So my stupid question is how do I see what was returned? Why doesnt it return the results instead of a "0"?

dugan · 10-24-2013, 11:29 AM

I think you want "subprocess.check_output". See here:

http://andymckay.github.io/presentat...index.html#/37

Or, you could do it more pythonically:

Code:

import os


def find_file(root, name):

    for root, dirs, files in os.walk(root):
        for filename in files:
            if filename.lower() == name:
                return os.path.join(root, name)

geopath = find_file('/usr/share', 'GeoIP.dat')
geobase = pygeoip.GeoIP(geopath, pygeoip.MEMORY_CACHE)

mina86 · 10-24-2013, 11:40 AM

You should not use os.system in that way. Reading output of find line-by-line is not error-prof. Furthermore, there is no need to spawn a new process to look for files with given name since this can be done in Python. What you need is os.walk:

Code:

import os

geopaths = []
for dirname, _, filenames in os.walk('/ush/share'):
    for filename in filenames:
        if filename.lower() == 'geoip.dat':
            geopaths.append(os.path.join(dirname, filename))

Quote:

Originally Posted by dugan

Code:

def find_file(root, name):
    for root, dirs, files in os.walk(root):
        for filename in files:
            if filename.lower() == name:
                return os.path.join(root, name)

geopath = find_file('/usr/share', 'GeoIP.dat')

Either pass “geoip.dat” or do not use “lower()” in the function.

dugan · 10-24-2013, 11:48 AM

Quote:

Originally Posted by mina86

Either pass “geoip.dat” or do not use “lower()” in the function.

Revising:

Code:

def find_file(root, name):

    for root, dirs, files in os.walk(root):
        for filename in files:
            if filename.lower() == name.lower():
                return os.path.join(root, name)

That should give you the same results as "find -iname" now.

mina86 · 10-24-2013, 12:18 PM

Quote:

Originally Posted by dugan

Code:

if filename.lower() == name.lower():

Better yet, add “name = name.lower()” at the beginning of the function.

Quote:

Originally Posted by dugan

Code:

return os.path.join(root, name)

This should read: “return os.path.join(root, filename)”.

metallica1973 · 10-24-2013, 01:23 PM

Awesome Many Thanks I will test this stuff and give you an update.

metallica1973 · 10-24-2013, 02:17 PM

I know I will recieve a scolding but the easiest method is Dugan:

Code:

caca = subprocess.check_output(['sudo', 'find', '/usr/share', '-iname', 'GeoIP.dat'])
[sudo] password for me: 

In [75]: caca
Out[75]: '/usr/share/GeoIP/GeoIP.dat\n'

If I can remove the \n, this will satisfy, I believe this is what I need and the less code the better. Ok let me have it.

this did the trick:

Code:

mod = caca[:-1].strip()
'/usr/share/GeoIP/GeoIP.dat'

long story short:

Code:

caca = subprocess.check_output(['sudo', 'find', '/usr/share', '-iname', 'GeoIP.dat'])
mod = caca[:-1].strip()
geobase = pygeoip.GeoIP(mod, pygeoip.MEMORY_CACHE)

dugan · 10-24-2013, 02:37 PM

Quote:

Originally Posted by metallica1973

If I can remove the \n

Code:

caca.strip()

(There are other ways, but this IMHO makes the most sense).

mina86 · 10-25-2013, 03:42 AM

Quote:

Originally Posted by metallica1973

I know I will recieve a scolding but the easiest method is Dugan:

As long as you are aware it's incorrect and use it by yourself, I don't care what you're using.

metallica1973 · 10-25-2013, 04:10 PM