Использование объектов в другом классе в Python

Question

В коде (псевдо), как это

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()

Я могу заставить оба работать. Я хочу, чтобы все элементы из первого были повторены и запущены во втором. Я не совсем уверен, как это сделать. Любая помощь, высоко ценится.

Answer 1

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()

Или, как указывает Ipthnc ниже, функция paths может быть сокращена до:

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

Answer 2

Если я правильно понял вопрос - вы можете добавить в fileWrite дополнительный параметр, например fileWrite (имя файла), и просто выполнить итерацию по «полной» последовательности.