[Python]12行で作るフォルダ内検索プログラム

1.フォルダ内検索プログラム

ディレクトリ内のファイルをキーワードで検索し、該当するファイル名と該当件数を出力するプログラムを作ります。検索したいディレクトリにこのスクリプトを配下し、コマンドラインの第一引数に検索したい文字列を入力します。

2.コード

from os import listdir
from sys import argv

search_word = argv[1:][0]
files = listdir()
search_results = [f for f in files if search_word in f]
if len(search_results) > 0:
    print("該当件数:", len(search_results), "件")
    for i, result in enumerate(search_results):
        print(i+1, result)
else:
    print("該当するファイル名はありませんでした。")

3.実行結果

NobitanoMacBook-puro:practice nobi$ python3 search_file.py py
該当件数: 113 件
1 password_generator.py
2 auto_keyboard.py
3 file_size_checker.py
4 generate_trans_word.py
5 digest.py
6 select_sort.py
7 tur_shape.py
8 joke.py
9 mols_code_converter.py
10 recrusive.py
11 sos_gui.py
12 caster.py
13 fizz_buzz.py
14 nest_loop.py
15 voice_to_speech.py
16 binary_search.py
17 translate.py
18 practice_pickle.py
19 palindrome.py
20 data_anl.py
21 zen-of-python.txt
22 ex9.py
23 tur_triangle.py
24 find_min_index.py
25 sum_func.py
26 pronounce_test.py
27 auto_mouse.py
28 draw_flower.py
29 search_data.py
30 weather_cast.py
31 int_except.py
32 count_vote.py
33 generator_prime.py
34 add_muldata.py
35 stock_scrap.py
36 handle_shelf.py
37 search_file.py
38 tur.py
39 birth_days.py
40 automate_twitter.py
41 weather.py
42 book_shelf.py
43 num_game.py
44 plot_graph.py
45 ex15.py
46 uni_ord.py
47 pra_json.py
48 tur_random.py
49 __pycache__
50 ex7.py
51 test.py
52 map_pra.py
53 automate_browser.py
54 ex11.py
55 ex32.py
56 rpg.py
57 effort.py
58 memeber_list.py
59 basic_auth.py
60 mikuji.py
61 ex16.py
62 sort_rpg.py
63 ex4.py
64 tur_star.py
65 pra_iter.py
66 dout.py
67 random_word.py
68 stone_game.py
69 search_address.py
70 calc_price.py
71 try_except.py
72 loop_pra.py
73 pong.py
74 collect_sells.py
75 crypt_str.py
76 ex13.py
77 ex33.py
78 calc.py
79 pomodoro.py
80 attr_practice.py
81 key_logger.py
82 ex17.py
83 typing_game.py
84 for_else.py
85 tur_cir2.py
86 html_format.py
87 japan_birth.py
88 amazon_rank.py
89 plot_img.py
90 db_handle.py
91 re_match_object.py
92 guess_10seconds_game.py
93 recursion.py
94 task_mail.py
95 enc_text.py
96 split_func.py
97 send_email.py
98 pra_set.py
99 pycon_scrape.py
100 default_dict.py
101 timesheet.py
102 bmi_calc_test.py
103 prime_num.py
104 bin_game.py
105 plot_practice.py
106 min_distance.py
107 search_by_portalnum.py
108 tur_cir.py
109 mul_list.py
110 spam_message.py
111 reiwa_convert.py
112 login_func.py
113 hoge.py

NobitanoMacBook-puro:practice nobi$ python3 search_file.py XXXXX.py
該当するファイル名はありませんでした。