Python — Рекурсивный поиск по строкам и файлам?

Question

[Hz An]

Добрый день!
У меня есть две функции:

Эта функция берет файл и переводит его в построчный вид.

def getstrings(content):
        file = open(content)
        return file.readlines()

    oneopenfile = getstrings(content)

Эта функция берет переменную с данным и циклично производит построчный поиск по данным и записывает результат в другой файл.

def search_data(oneopenfile):
        result = open((fileresult + "." + filetype), 'w')
        for i in oneopenfile:
            if i.count(searchinfo):
                result.write(i)

    search_data(oneopenfile)

Каким образом реализовать что бы тоже самое делалось не по конкретному файлу а по всем фалам в директории?

Решено: https://github.com/onixdclxvi/PyDCLXVI/blob/master...

Answer 1

[Дмитрий]

for d, dirs, files in os.walk('/home/user/dir'):
	for f in files:
             getstrings(f)

или

for f in os.listdir('/home/path/dir'):
      getstrings(f)

Answer 2

[Hz An]

Дмитрий Евгений

в общем вот что придумал, но не работает)

Собираю имена файлов с расширением txt в файл и записываю построчно каждое имя в файл.

def filebasedir():
    filetodir = os.walk('.')
    tempfile = open('tempfilebase.txt', 'w')
    for line, dirs, files in filetodir:
        for f in files:
            if f.count(".txt"):
                tempfile.write(f+"\n")

filebasedir()

Далее пытаюсь произвести поиск обойти по очереди все файлы и все строки в них.

def search_all_data():
    result = open(("TESTRESULT.txt"), 'w')
    allopenfile = open('tempfilebase.txt', 'r')
    for linenamefile in allopenfile:
            linenamefile = open(linenamefile)
            print (linenamefile)
            dataonefile = linenamefile #.readlines()
            for data in dataonefile:
               if data.count("root"):
                   result.write(data)

search_all_data()

В итоге получаю ошибку

Traceback (most recent call last):
  File "test.py", line 34, in <module>
    search_all_data()
  File "test.py", line 27, in search_all_data
    linenamefile = open(linenamefile)
FileNotFoundError: [Errno 2] No such file or directory: '111.txt\n'

я так понимаю он не может найти файл 111.txt\n из за того что справа отступ \n?
в общем как это дело исправить? подскажите? второй день только изучаю питон

Comments

[Дмитрий]

for d, dirs, files in os.walk('.'):
  for f in files:
        if f.endswith('.txt'):
             getstrings(f)

И не надо городить никаких велосипедов

[Hz An]

Дмитрий:
Дмитрий будь добр, можешь раскинуть для дилетанта, как осуществить выборку по всем строкам и всем TXT файлам внутри директории, кручу верчу ничего не понимаю

[Дмитрий]

Файл у тебя читается в getstrings, а получаешь файлы при помощи того кода, который я написал

[Hz An]

Дмитрий:

Вот что получилось,

import os

def searchdata(f):
    zapis = ('TESTRESULT.txt')
    result = open(zapis, 'w')
    files = open(f, 'r')
    for i in files:
        if i.count('ftp'):
            print (i)
            result.write(i)

for d, dirs, files in os.walk('.'):
  for f in files:
        if f.endswith('.txt'):
            searchdata(f)

но почему то в файл не записывает, и пробегает судя по принту очень мало, как будто не по всем файлам бегает или не все парсит

[Hz An]

Дмитрий: а нет! все работает! спасибо!

Answer 3

[Евгений]

Читать документацию по функции стандартной библиотеки os.walk()