Utilizing objects in another class in Python

Question

In code(pseudo) like this

def path():
    dirList = ['c:\\', 'y:\\', 'z:\\']
    home_folder = 'peter.txt'
    complete = [s + home_folder for s in dirList]
    print complete

def fileWrite():
    filename = 'c:\peter.txt'
    text = 'Hello World'
    file = open(filename, 'w')
    file.write(text)
    file.close()

I can make both work. I want all the items from the first to be iterated and run in the second. I am not entirely sure how to do that. Any help, much appreciated.

Answer 1

If I understand question correclty - you can add additional parameter to fileWrite like fileWrite(filename) and simply iterate over 'complete' sequence.

Answer 2

import os

def paths(filename):
    dirList = ['c:\\', 'y:\\', 'z:\\']
    complete = [os.path.join(s, filename) for s in dirList]
    return complete

def fileWrite():
    for each_file in paths('c:\\peter.txt'):
        text = 'Hello World'
        file = open(each_file, 'w')
        file.write(text)
        file.close()

Or, as Ipthnc points out below, the paths function can be shortened to:

def paths(filename):
    return [os.path.join(s, filename) for s in ('c:\\', 'y:\\', 'z:\\')]