カレントディレクトリ配下のテキストファイルの中身を検索したい

Question

#前提・実現したいこと
カレントディレクトリ：C:\Users\user\Desktop\python
このディレクトリには作成したpy fileやtextファイルが複数保存されているが
何で検索しても表示結果がでてこない

path = os.path.join(root,fi)でパスを取得していますが
参照するディレクトリの方法が間違っているなど
何か考えられる原因はありますでしょうか。

#発生している問題
PS C:\Users\user\Desktop\python> python3 .\findtext.py def
PS C:\Users\user\Desktop\python>
”何も表示されない”

#ソースコード

※findtext.py
import sys
import os

if len(sys.argv) <= 1:
    print("[USAGW]) findtext (keyword)")
    sys.exit(0)

keyword = sys.argv[1]

for root, dirs, files in os.walk("."):
    for fi in files:
        result = []
        try:
            path = os.path.join(root,fi)
            with open(path,encoding='utf-8') as f:
                for no, line in enmerate(f):
                    if line.find(keyword) >= 0:
                        line = line.strip()
                        s = "| {0:4}: {1}".format(no+1,line)
                        result.append(s)
        except:
            continue

　　　　#resultに結果があれば表示する
        if len(result) > 0:
            print("+ file:" + fi)
            for li in result:
                print(li) # resultに追加された行を表示

Comments

[LouiS0616]

とりあえず例外発生時にトレースバックを表示されてはいかがですか。

Answer 1

作成されたプログラム、力作ですねー。

1,enumerateのスペルミス。IDE(統合開発環境）をインストールするとこのミスは防げるかと。

for no, line in enmerate(f):

for no, line in enumerate(f):

2,exceptで例外を全部握りつぶしています。よってtryの処理内で例外発生時は質問文のように何も表示されません。

except:
continue

Exception で例外を全てcacthするのもあまり良くないのですが。以下のように記述すれば、例外発生時にエラーメッセージが表示されます。

except Exception as ex:
    print(ex)
    continue

3,プログラムに渡された引数を元に処理を行うには標準ライブラリのargparseを使うと便利です。プログラムに渡す引数が足りない時はusageも表示してくれます。

import argparse

parser = argparse.ArgumentParser()
parser.add_argument('keyword')
args = parser.parse_args()
keyword = args.keyword

4,一つの関数内で処理を行いすぎかと、最初はテストを行いやすいようにある程度処理を分割するほうが良いです。

プログラム内でループを行っている部分があれば、関数化してlist/dictを返すようにしたり、yieldを返す形にしたほうがいいです。

            path = os.path.join(root,fi)
            with open(path,encoding='utf-8') as f:
                for no, line in enumerate(f):
                    if line.find(keyword) >= 0:
                        line = line.strip()
                        s = "| {0:4}: {1}".format(no+1,line)
                        result.append(s)

関数化してlistを返すサンプルコード

def find_text_file(file_path, keyword):
    lines = []
    with open(file_path, encoding='utf-8') as f:
        for no, line in enumerate(f):
            if line.find(keyword) >= 0:
                line = line.strip()
                s = "| {0:4}: {1}".format(no + 1, line)
                lines.append(s)
    return lines

関数の呼び出し側

result = find_text_file(path, keyword)

---

以下のソースコードは上記4点を変更したソースコードになります。

# -*- coding: UTF-8 -*
import sys
import os
import argparse

parser = argparse.ArgumentParser()
parser.add_argument('keyword')
args = parser.parse_args()
keyword = args.keyword


def find_text_file(file_path, keyword):
    lines = []
    with open(file_path, encoding='utf-8') as f:
        for no, line in enumerate(f):
            if line.find(keyword) >= 0:
                line = line.strip()
                s = "| {0:4}: {1}".format(no + 1, line)
                lines.append(s)
    return lines


for root, dirs, files in os.walk("."):
    for fi in files:
        result = []
        try:
            path = os.path.join(root, fi)
            result = find_text_file(path, keyword)
        except Exception as ex:
            print(ex)
            continue

        #resultに結果があれば表示する
        if len(result) > 0:
            print("+ file:" + fi)
            for li in result:
                print(li) # resultに追加された行を表示

Comments

[ItoM]

ありがとうございます。
Visual Studio Codeを使っているのですが、見落としていましたお恥ずかしい限りです。
修正頂いたコードもじっくり読ませていただきます。

Answer 2

× enmerate
○ enumerate

参照するディレクトリの方法が間違っているなど

こんなことは質問するまでもなく、単純なprintデバッグで確認できることですね。

Comments

[ItoM]

ありがとうございます。
printデバックでの解決法ありがとうございます。
次に躓いた時には試してみます。