【每日一坑 4】 查找文件

crossin先生

上一个坑，取出字符串中的数字，在论坛上出现了好几种方法。除了基本的遍历判断来做外，还有一些简便的python解法：

1.正则
''.join(re.findall(r'\d+',text))

2.isdigit
''.join([i for i in text if i.isdigit()])

[i for i in test]这是一种生成list的方法，通过后面的if可以增加生成时的过滤条件。这种写法在python中很常用。

3.filter
filter(lambda x: x.isdigit(), text)

filter是一个过滤器，其中的lambda表达式是过滤的条件。这个稍微高深了一点，有兴趣的同学可以去搜索一下“lambda表达式”。

今天的坑，准备做一些跟文件相关的操作。先来点最基本的：

找出指定文件夹中的所有以txt结尾的文件，包括所有嵌套的子文件夹。

我回来啦

1. # coding:utf-8
   
2. import os
   
3. import fnmatch
   
4. 
   
5. for root, dirs, files in os.walk('.'):
   
6.     for file in files:
   
7.         if fnmatch.fnmatch(file, "*.txt"):
   
8.             print file
   

复制代码

fl0w

findThem.py

1. #! /usr/bin/env python
   
2. # coding:utf-8
   
3. 
   
4. import os
   
5. import sys
   
6. 
   
7. if __name__ == "__main__":
   
8.     if len(sys.argv) != 1 and os.path.exists(sys.argv[1]):
   
9.         themAll = []
   
10.         for dirpath, dirnames, filenames in os.walk(sys.argv[1]):
    
11.             for filename in filenames:              
    
12.                 if os.path.splitext(filename)[1] == '.txt':
    
13.                     themAll.append(os.path.join(dirpath, filename))
    
14.         print themAll
    
15.     else:
    
16.         print 'Please input like this:./findThem.py /home/fl0w'

复制代码

HAO

1. # -*- coding: utf-8 -*-
   
2. 
   
3. import os
   
4. from sys import argv
   
5. 
   
6. filename, path = argv
   
7. 
   
8. txt_files = []
   
9. 
   
10. if os.path.isdir(path):
    
11.     for dir_path, dir_names, file_names in os.walk(path):
    
12.         for file_name in file_names:
    
13.             if file_name.endswith('.txt'):
    
14.                 txt_files.append(file_name)
    
15. 
    
16. 
    
17. print txt_files

复制代码

fl0w

HAO 发表于 2013-12-19 21:33

这样不能找出嵌套的子文件夹中的以.txt结尾的文件

HAO

fl0w 发表于 2013-12-19 21:39
这样不能找出嵌套的子文件夹中的以.txt结尾的文件

哈哈。谢谢提醒。

liucy

1. # coding:utf-8
   
2. import os
   
3. import fnmatch
   
4. 
   
5. def findfile(inputdir):
   
6.     txtlist = []
   
7.     for parent, dirnames, filename in os.walk(inputdir):
   
8.         for filenames in filename:
   
9.             txtlist.append(filenames)
   
10.         return fnmatch.filter(txtlist, '*.txt')
    
11. 
    
12. if __name__ == "__main__":
    
13.     inputdir1 = input("Please input your dir(like this: '/Users/lemon') : ")
    
14.     print findfile(inputdir1)

复制代码

shallecker

带路径+txt

1. # coding:utf-8
   
2. import os, re
   
3. 
   
4. for i in os.walk('d:'+os.sep):
   
5.     for txt in i[2]:
   
6.         try:
   
7.             txt = re.match(r'(.*)(\.txt)', txt).group(0)
   
8.             print os.path.join(i[0], txt)
   
9.         except:
   
10.             pass

复制代码

xuefu

长长的c又来了，有点乱没来得急整理。。。

1. #include <stdio.h>
   
2. #include <unistd.h>
   
3. #include <stdlib.h>
   
4. #include <dirent.h>
   
5. #include <errno.h>
   
6. #include <string.h>
   
7. #include <sys/stat.h>
   
8. 
   
9. #define PATH_SIZE 4096
   
10. 
    
11. char *path;
    
12. 
    
13. int show_txt(char *);
    
14. 
    
15. int main(int argc, const char *argv[])
    
16. {
    
17.   if (argc != 2) {
    
18.     printf("Usage: ./a.out <path of dir>\n");
    
19.     exit(1);
    
20.   }
    
21. 
    
22.   path = malloc(PATH_SIZE);
    
23.   strcpy(path, argv[1]);
    
24. 
    
25.   show_txt(path);
    
26. 
    
27.   free(path);
    
28. 
    
29.   return 0;
    
30. }
    
31. 
    
32. int show_txt(char *path)
    
33. {
    
34.   struct stat statbuf;
    
35.   struct dirent *dirp;
    
36.   DIR *dp;
    
37.   char *ptr;
    
38. 
    
39.   int len = strlen(path);
    
40. 
    
41.   if (lstat(path, &statbuf) < 0) {  /* lstat error */
    
42.     perror("lstat error");
    
43.     exit(1);
    
44.   }
    
45. 
    
46.   if (S_ISDIR(statbuf.st_mode) == 0) {  /* not a directory */
    
47.       if (strcmp(&path[len-4], ".txt") == 0) {
    
48.       printf("%s\n", path);
    
49.     }
    
50.     return 0;
    
51.   }
    
52. 
    
53.   ptr = path + len;
    
54.   *ptr++ = '/';
    
55.   *ptr = 0;
    
56. 
    
57.   if ((dp = opendir(path)) == NULL) { /* open directory error */
    
58.     perror("opendir error");
    
59.     exit(1);
    
60.   }
    
61. 
    
62.   /* iterate the directory(dp)'s each element*/
    
63.   while ((dirp = readdir(dp)) != NULL) {  
    
64.     /* ignore the present directory and the last directory */
    
65.     if (strcmp(dirp->d_name, ".") == 0 || strcmp(dirp->d_name, "..") == 0)
    
66.       continue;
    
67. 
    
68.     strcpy(ptr, dirp->d_name);  /* append the element's name to path */
    
69.     show_txt(path); /* call the show_txt(char*) recursively */
    
70.   }
    
71. 
    
72.   ptr[-1] = 0;
    
73.   closedir(dp);
    
74.   return 0;
    
75. }
    

复制代码

scz_000

1. import os,fnmatch
   
2. def find_txt(root,patterns='*.txt'):#定义主函数，接受目录与所要匹配的拓展名为参数
   
3.     list_of_txt=[]#列表储存txt文件名
   
4.     for dirname,subdir,filenames in os.walk(root):#使用os。walk（遍历包括子目录下所有文件），生成三组列表，第一组为当前目录地址，第二组为当下前目录下的文件夹名（没有则返回一个空列表），第三组为当前目录的所有文件
   
5.         for name in filenames:#遍历每组文件名
   
6.             if fnmatch.fnmatch(name,patterns):#用fnmatch。fnmatch寻找匹配的文件
   
7.                 list_of_txt.append(name)#加入列表
   
8.     return list_of_txt
   
9. if __name__=='__main__':
   
10.     dir=raw_input('directory:')
    
11.     txtfiles=find_txt(dir)
    
12.     print txtfiles

复制代码