python 操作文件夹练习 找出文件夹下所有txt文件

import os
import sys

'''
找出文件夹中的所有TXT文件
'''
def search_file(root:str):
    for dir,dir_names,file_names in os.walk(root):
        for file in file_names:
            if file.lower().endswith("txt"):
                print(f"{dir}/{file}")



def main():
    root_name = sys.argv
    if len(root_name) <= 1:
        print("请输入文件夹名称")
    else:
        search_file(root_name[1])



if __name__ == "__main__":
    main()