代码小白liu 发表于 2022-4-14 22:09:00

文件相关问题

本帖最后由 代码小白liu 于 2022-4-14 22:11 编辑

这个代码是怎么实现“当前文件夹内包含文件夹,则进入文件夹继续搜索”的功能的,i部分是将walk返回的文件进行判断了呀



import os


def print_pos(key_dict):
    keys = key_dict.keys()
    keys = sorted(keys)# 由于字典是无序的,我们这里对行数进行排序
    for each_key in keys:
      print('关键字出现在第 %s 行,第 %s 个位置。' % (each_key, str(key_dict)))


def pos_in_line(line, key):
    pos = []
    begin = line.find(key)
    while begin != -1:
      pos.append(begin + 1)# 用户的角度是从1开始数
      begin = line.find(key, begin + 1)# 从下一个位置继续查找

    return pos


def search_in_file(file_name, key):
    f = open(file_name)
    count = 0# 记录行数
    key_dict = dict()# 字典,用户存放key所在具体行数对应具体位置

    for each_line in f:
      count += 1
      if key in each_line:
            pos = pos_in_line(each_line, key)# key在每行对应的位置
            key_dict = pos

    f.close()
    return key_dict


def search_files(key, detail):
    all_files = os.walk(os.getcwd())
    txt_files = []

    for i in all_files:
      for each_file in i:
            if os.path.splitext(each_file) == '.txt':# 根据后缀判断是否文本文件
                each_file = os.path.join(i, each_file)
                txt_files.append(each_file)

    for each_txt_file in txt_files:
      key_dict = search_in_file(each_txt_file, key)
      if key_dict:
            print('================================================================')
            print('在文件【%s】中找到关键字【%s】' % (each_txt_file, key))
            if detail in ['YES', 'Yes', 'yes']:
                print_pos(key_dict)


key = input('请将该脚本放于待查找的文件夹内,请输入关键字:')
detail = input('请问是否需要打印关键字【%s】在文件中的具体位置(YES/NO):' % key)
search_files(key, detail)

ba21 发表于 2022-4-14 22:20:37

给你一个初始目录d:\test。
问,该目录下总共有多少目录和文件。
你能看到的只能是d:\test当前有多少文件和目录,其后的子目录有些什么你不知道。os.listdir('d:\test')能获取当前目录下的目录和文件。
这时你便可以一个一个目录进去用同样的方法来获取目录下的目录和文件。这就是递归方法。
递归方法:
import os

def gci(filepath):
# 遍历filepath下所有文件,包括子目录
files = os.listdir(filepath)
for fi in files:
    fi_d = os.path.join(filepath,fi)            
    if os.path.isdir(fi_d):
      gci(fi_d)                  
    else:
      print(fi_d)

#递归遍历/root目录下所有文件
gci('d:\test')
而你的代码中用了现成的方法os.walk()来获取d:\test下的所有目录和文件(包括子目录)
import os
for fpathe,dirs,fs in os.walk('d:\test'):
for f in fs:
    print(os.path.join(fpathe,f))

jackz007 发表于 2022-4-14 22:42:46

本帖最后由 jackz007 于 2022-4-14 22:44 编辑

root , dirs , files = os . walk(os . getcwd())
      该调用返回一个有 3 个元素(root , dirs , files) 元组的生成器,会从当前目录(os . getcwd()) 开始,递归扫描每一个子目录,每扫描完一个子目录,就反馈一次 root,dirs 和 files,其中,root 是子目录的绝对路径,dirs 是该目录内所有子目录的列表,files 是该子目录内所有普通文件的列表。
页: [1]
查看完整版本: 文件相关问题