码迷,mamicode.com
首页 > 其他好文 > 详细

提取文件名(非查找文件)

时间:2015-06-27 19:37:25      阅读:105      评论:0      收藏:0      [点我收藏+]

标签:

曾经有这么一个需要,就想自己实现一个。虽然我知道java内置文件操作的库而C++没有,但还是不想换语言(文人相轻,程序猿更相轻)。于是就上网找了一段C版本的代码,就是这个:

 1 #include <iostream>
 2 #include <vector>
 3 #include <string>
 4 #include <cstring>
 5 #include <io.h>
 6 using namespace std;
 7 /**
 8 *获取文件夹中所有文件名
 9     input
10         path:   文件夹路径
11         exd:    文件后缀,如为空则全选
12     output
13         files:  容器
14 */
15 void getFileName(string path, string exd, vector<string>& files)
16 {
17     /// 文件句柄
18     long hFile = 0;
19     /// 文件信息
20     struct _finddata_t fileinfo;
21     string pathName(path), exdName = "\\*."+exd;
22     if(hFile = _findfirst((pathName+exdName).c_str(), &fileinfo) != -1)
23     {
24         do
25         {///如果有文件夹,迭代之
26             if(fileinfo.attrib&_A_SUBDIR)
27             {
28                 if(strcmp(fileinfo.name,".") && strcmp(fileinfo.name,".."))
29                     getFileName(pathName+"\\"+fileinfo.name, exd, files);
30             }
31             else
32             {
33                 if(strcmp(fileinfo.name,".") && strcmp(fileinfo.name,".."))
34                     files.push_back(fileinfo.name);
35             }
36         }while(_findnext(hFile, &fileinfo)==0);
37         /**这段代码跑出来实际上只能找到一个文件就结束了,找原因还很麻烦
38         */
39         _findclose(hFile);
40     }
41 }
42 int main()
43 {
44     string path("E:");
45     vector<string> files;
46     getFileName(path,"",files);
47     for(int i = 0; i < files.size(); i++)
48         cout<<files[i]<<endl;
49     return 0;
50 }

结果调了一上午还没调过,真是坑爹。

后来发现用dos也能做,而且简单到让我想撞墙,就像这样

@echo off
dir /s/b *.txt > 文件名.txt
exit

就这么一行,dir就不用说了,/s表示递归遍历,/b表示一个文件一行,*.txt表示查找txt文件,然后写入到一个文件中。

把它保存为bat,放入当前文件夹就可了。不过它输出的是绝对路径,还得自己再加工。

问题是解决了,可是也太快了,我还没感觉呢。

后来,我试着用NB的python解决,在实现的过程中,我感觉我就是在搞翻译。比如我知道要“获取当前文件夹”,然后python里是哪个函数有这个功能呢?找。我知道要“递归遍历”,但是python里是哪个函数有这个功能呢?找。大部分时间就是在看文档,学习语法,模块,并没有什么乐趣,也许是因为我对这个语言不熟吧。最后实现了递归与非递归两个版本,如下所示:

1 import os
2 import fnmatch
3 f=open("图片.txt","w+")
4 for file in os.listdir(os.getcwd()):
5     if fnmatch.fnmatch(file, *.jpg):
6         f.write(file+"\n")
7     #    f.write(os.path.realpath(file)+"\n")
8 f.close()
 1 import os
 2 import fnmatch
 3 f=open("图片.txt","w+")
 4 for tup in os.walk(os.getcwd()):
 5     lst = tup[2]
 6     dirpath = tup[0]
 7     for file in lst:
 8         if fnmatch.fnmatch(file, *.jpg):
 9             f.write(os.path.join(dirpath, file)+"\n")
10        #     f.write(file+"\n")
11 f.close()

其中,getcwd()就是获取当前目录,listdir()是获取目录下所有子目录(非递归),walk()是递归获取,返回一个list of tuple,tuple包含三个元素,分别是当前目录,list of当前目录下的文件夹,list of当前目录下的文件。fnmatch()就是一个匹配的函数;realpath()返回绝对路径,join()连接路径。不得不说,脚本语言还是很方便的,源代码可以当exe直接运行。

后来,我遇到了吊炸天的Boost,终于又可以用C++实现了。Boost的filesystem库实现了文件操作,而且它的可读性非常好,几乎不需要解读。

 1 #include <iostream>
 2 #include <fstream>
 3 #include <string>
 4 #include <boost/filesystem/operations.hpp>
 5 using namespace std;
 6 using namespace boost;
 7 namespace fs = boost::filesystem;
 8 using rd_iter = fs::recursive_directory_iterator;
 9 using d_iter = fs::directory_iterator;
10 int main()
11 {
12     string ext;
13     cout<<"请输入文件格式(含.)"<<endl;
14     cin>>ext;
15     ofstream fout("output.txt");
16     rd_iter pos("./"), End;
17     for(; pos != End; ++pos)
18     {
19         if(!is_directory(*pos) &&
20            pos->path().extension() == ext)
21         {   //仅文件
22         //    fout << pos->path().filename().string() << endl;
23             //绝对路径
24         //    fout << system_complete(pos->path()).string() << endl;
25             //相对路径
26             fout << pos->path().relative_path().string() << endl;
27         }
28     }
29 }

 

用"./"表示当前目录,多么简单直观。d_iter是非递归迭代器,rd_iter是递归迭代器,这里用迭代器来遍历路径。Boost把迭代器玩出魔性来了,好多用法第一次看的时候是跪着看的。更多的内容可以去看Chrono的书,这里就不说了。

提取文件名(非查找文件)

标签:

原文地址:http://www.cnblogs.com/lzxskjo/p/4604400.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!