python文件路径过滤器_在Python中的过滤器目录

I am trying to get filtered list of all Text and Python file, like below

from walkdir import filtered_walk, dir_paths, all_paths, file_paths

vdir=raw_input ("enter director :")

files = file_paths(filtered_walk(vdir, depth=0,included_files=['*.py', '*.txt']))

I want to:

know the total number of files found in given directory

I have tried options like : Number_of_files= len (files) or for n in files n=n+1 but all are failing as "files" is something called "generator" Object which I searched on python docs but couldn't make use of it

I also want to find a string e.g. "import sys" in the list of files found in above and store the file names having my search string in new file called "found.txt"

解决方案

I believe this does what you want, if I misunderstood your specification, please let me know after you give this a test. I've hardcoded the directory searchdir, so you'll have to prompt for it.

import os

searchdir = r'C:\blabla'

searchstring = 'import sys'

def found_in_file(fname, searchstring):

with open(fname) as infp:

for line in infp:

if searchstring in line:

return True

return False

with open('found.txt', 'w') as outfp:

count = 0

search_count = 0

for root, dirs, files in os.walk(searchdir):

for name in files:

(base, ext) = os.path.splitext(name)

if ext in ('.txt', '.py'):

count += 1

full_name = os.path.join(root, name)

if found_in_file(full_name, searchstring):

outfp.write(full_name + '\n')

search_count += 1

print 'total number of files found %d' % count

print 'number of files with search string %d' % search_count

Using with to open the file will also close the file automatically for you later.