Python根据文件后缀统计文件巨细、找出文件位置(仿Everything）

需求分析

这个脚本支持扫描并统计某个文件夹内，以某个后缀末端的，好比sql,java(以使用参数为准）。而且可以看到此中的文件巨细和完备文件路径，还支持以CSV、XLSX等格式导出等。这个脚本实用于统计本地代码行数之后，必要查察某个编程语言，好比Java写了多少文件、每个文件巨细是多少、文件路径在那里等操纵。
使用教程

我们使用脚本时，输入想要检测的路径、文件后缀、以及是否导出文件、导出文件的定名（文件数目过多时使用）等就好。
好比python list_zip_contents.py -tp “C:\Users\your_name\Pictures\” -a -o rpng.csv   ,演示时为best.csv。

末了，用编辑器打开即可。

详细代码

1. # -*- coding: utf-8 -*-
2. import zipfile
3. import os
4. import argparse
6. EXTENSIONS = {
7.     'sql': '.sql', 'java': '.java', 'c': '.c', 'cpp': '.cpp', 'h': '.h',
8.     'py': '.py', 'js': '.js', 'html': '.html', 'css': '.css', 'markdown': '.md',
9.     'txt': '.txt', 'xml': '.xml', 'yaml': '.yaml', 'yml': '.yml', 'json': '.json',
10.     'properties': '.properties', 'go': '.go', 'rs': '.rs', 'rb': '.rb',
11.     'php': '.php', 'swift': '.swift', 'kt': '.kt', 'cs': '.cs', 'lua': '.lua',
12.     'sh': '.sh', 'bat': '.bat', 'ps1': '.ps1', 'sql': '.sql'
13. }
15. def split_archive_path(path):
16.     lower_path = path.lower()
17.     all_extensions = ['.tar.gz', '.tar.bz2', '.tar.zst', '.tgz', '.zip', '.7z', '.rar', '.tar', '.gz', '.bz2']
18.     candidates = []
19.     for ext in all_extensions:
20.         ext_lower = ext.lower()
21.         idx = lower_path.rfind(ext_lower)
22.         if idx >= 0:
23.             candidates.append((idx, len(ext), ext, path[:idx + len(ext)], path[idx + len(ext):].lstrip('\\').lstrip('/')))
24.     if candidates:
25.         candidates.sort(key=lambda x: (x[0], -x[1]))
26.         _, _, _, archive_path, sub_path = candidates[0]
27.         if sub_path:
28.             return archive_path, sub_path
29.         else:
30.             return archive_path, ''
31.     return path, ''
33. def is_archive(path):
34.     return path.lower().endswith(('.zip', '.7z', '.rar', '.tar', '.tar.gz', '.tar.bz2', '.tgz', '.gz', '.bz2'))
36. def list_files_in_path(target_path, filter_exts, show_all):
37.     if os.path.isdir(target_path):
38.         return list_files_in_dir(target_path, filter_exts, show_all)
39.     elif os.path.isfile(target_path) and is_archive(target_path):
40.         return list_files_in_archive(target_path, filter_exts, show_all)
41.     else:
42.         archive_path, sub_path = split_archive_path(target_path)
43.         if os.path.isfile(archive_path) and is_archive(archive_path) and sub_path:
44.             return list_files_in_archive_subdir(archive_path, sub_path, filter_exts, show_all)
45.         else:
46.             return {'error': f'Not a valid directory or zip file - {target_path}', 'files': [], 'header': ''}
48. def list_files_in_archive_subdir(zip_path, sub_path, filter_exts, show_all):
49.     import tempfile
50.     temp_dir = tempfile.mkdtemp()
51.     try:
52.         if zip_path.lower().endswith('.zip'):
53.             with zipfile.ZipFile(zip_path, 'r') as z:
54.                 for member in z.namelist():
55.                     normalized = member.replace('/', os.sep).replace('\\', os.sep)
56.                     sub_path_normalized = sub_path.replace('/', os.sep).replace('\\', os.sep)
57.                     if normalized.startswith(sub_path_normalized + os.sep) or normalized == sub_path_normalized:
58.                         z.extract(member, temp_dir)
59.         elif zip_path.lower().endswith('.7z'):
60.             import py7zr
61.             with py7zr.SevenZipFile(zip_path, 'r') as sz:
62.                 sz.extractall(temp_dir)
63.         else:
64.             return {'error': f'Unsupported archive format - {zip_path}', 'files': [], 'header': ''}
65.         extract_subdir = temp_dir
66.         for item in os.listdir(temp_dir):
67.             if sub_path.replace('/', os.sep).replace('\\', os.sep).endswith(item) or item.endswith('.zip') or item.endswith('.7z'):
68.                 potential = os.path.join(temp_dir, item)
69.                 if os.path.isdir(potential):
70.                     extract_subdir = potential
71.                     break
72.         return list_files_recursive(extract_subdir, filter_exts, show_all, zip_path, sub_path)
73.     finally:
74.         import shutil
75.         try:
76.             shutil.rmtree(temp_dir)
77.         except:
78.             pass
80. def list_files_recursive(dir_path, filter_exts, show_all, archive_path=None, sub_path=None):
81.     files_found = []
82.     dirs_to_scan = [dir_path]
84.     while dirs_to_scan:
85.         current_dir = dirs_to_scan.pop()
86.         for root, dirs, files in os.walk(current_dir):
87.             for filename in files:
88.                 filepath = os.path.join(root, filename)
89.                 try:
90.                     size = os.path.getsize(filepath)
91.                 except:
92.                     size = 0
93.                 rel_path = os.path.relpath(filepath, dir_path)
95.                 ext = filename.lower()
96.                 if ext.endswith(('.zip', '.7z', '.rar', '.tar', '.tar.gz', '.tgz', '.gz', '.bz2')):
97.                     nested_result = try_extract_nested(filepath, filter_exts, show_all)
98.                     if nested_result:
99.                         files_found.extend(nested_result['files'])
100.                     continue
102.                 if show_all:
103.                     files_found.append((rel_path, size))
104.                 else:
105.                     file_ext = os.path.splitext(filename)[1].lower()
106.                     if file_ext in filter_exts:
107.                         files_found.append((rel_path, size))
109.     if archive_path and sub_path:
110.         header = f'Archive: {archive_path} / {sub_path}'
111.     elif archive_path:
112.         header = f'Archive: {archive_path}'
113.     else:
114.         header = f'Directory: {dir_path}'
115.     header += f'\nTotal files: {len(files_found)}\n' + '=' * 80
116.     return {'files': sorted(files_found), 'header': header}
118. def try_extract_nested(archive_file, filter_exts, show_all):
119.     import tempfile
120.     temp_dir = tempfile.mkdtemp()
121.     try:
122.         ext = archive_file.lower()
123.         if ext.endswith('.zip'):
124.             with zipfile.ZipFile(archive_file, 'r') as z:
125.                 z.extractall(temp_dir)
126.         elif ext.endswith('.7z'):
127.             import py7zr
128.             with py7zr.SevenZipFile(archive_file, 'r') as sz:
129.                 sz.extractall(temp_dir)
130.         else:
131.             return None
133.         files_found = []
134.         for root, dirs, files in os.walk(temp_dir):
135.             for filename in files:
136.                 filepath = os.path.join(root, filename)
137.                 try:
138.                     size = os.path.getsize(filepath)
139.                 except:
140.                     size = 0
141.                 rel_path = os.path.relpath(filepath, temp_dir)
143.                 if show_all:
144.                     files_found.append((rel_path, size))
145.                 else:
146.                     file_ext = os.path.splitext(filename)[1].lower()
147.                     if file_ext in filter_exts:
148.                         files_found.append((rel_path, size))
149.         return {'files': sorted(files_found), 'header': f'Nested: {archive_file}'}
150.     except:
151.         return None
152.     finally:
153.         import shutil
154.         try:
155.             shutil.rmtree(temp_dir)
156.         except:
157.             pass
159. def list_files_in_archive(zip_path, filter_exts, show_all):
160.     if zip_path.lower().endswith('.zip'):
161.         return list_files_in_zip(zip_path, filter_exts, show_all)
162.     elif zip_path.lower().endswith('.7z'):
163.         return list_files_in_7z(zip_path, filter_exts, show_all)
164.     else:
165.         return {'error': f'Unsupported archive format - {zip_path}', 'files': [], 'header': ''}
167. def list_files_in_7z(archive_path, filter_exts, show_all):
168.     import py7zr
169.     import tempfile
170.     temp_dir = tempfile.mkdtemp()
171.     try:
172.         with py7zr.SevenZipFile(archive_path, 'r') as sz:
173.             sz.extractall(temp_dir)
174.         files_found = []
175.         for root, dirs, files in os.walk(temp_dir):
176.             for filename in files:
177.                 filepath = os.path.join(root, filename)
178.                 try:
179.                     size = os.path.getsize(filepath)
180.                 except:
181.                     size = 0
182.                 rel_path = os.path.relpath(filepath, temp_dir)
183.                 if show_all:
184.                     files_found.append((rel_path, size))
185.                 else:
186.                     ext = os.path.splitext(filename)[1].lower()
187.                     if ext in filter_exts:
188.                         files_found.append((rel_path, size))
189.         return {'files': sorted(files_found), 'header': f'Archive: {archive_path}\nTotal files: {len(files_found)}\n' + '=' * 80}
190.     finally:
191.         import shutil
192.         try:
193.             shutil.rmtree(temp_dir)
194.         except:
195.             pass
197. def list_files_in_zip_subdir(zip_path, sub_path, filter_exts, show_all):
198.     import tempfile
199.     temp_dir = tempfile.mkdtemp()
200.     try:
201.         with zipfile.ZipFile(zip_path, 'r') as z:
202.             for member in z.namelist():
203.                 if member.startswith(sub_path + '/') or member == sub_path:
204.                     z.extract(member, temp_dir)
205.         extract_subdir = os.path.join(temp_dir, sub_path)
206.         if os.path.exists(extract_subdir):
207.             files_found = []
208.             for root, dirs, files in os.walk(extract_subdir):
209.                 for filename in files:
210.                     filepath = os.path.join(root, filename)
211.                     try:
212.                         size = os.path.getsize(filepath)
213.                     except:
214.                         size = 0
215.                     rel_path = os.path.relpath(filepath, extract_subdir)
216.                     if show_all:
217.                         files_found.append((rel_path, size))
218.                     else:
219.                         ext = os.path.splitext(filename)[1].lower()
220.                         if ext in filter_exts:
221.                             files_found.append((rel_path, size))
223.             print(f'Archive: {zip_path} / {sub_path}')
224.             print(f'Total files: {len(files_found)}')
225.             print('=' * 80)
226.             for path, size in sorted(files_found):
227.                 print(f'{size:>12,} bytes  {path}')
228.         else:
229.             print(f'Error: Subdirectory not found in archive - {sub_path}')
230.     finally:
231.         import shutil
232.         try:
233.             shutil.rmtree(temp_dir)
234.         except:
235.             pass
237. def list_files_in_dir(dir_path, filter_exts, show_all):
238.     files_found = []
239.     for root, dirs, files in os.walk(dir_path):
240.         for filename in files:
241.             filepath = os.path.join(root, filename)
242.             try:
243.                 size = os.path.getsize(filepath)
244.             except:
245.                 size = 0
246.             rel_path = os.path.relpath(filepath, dir_path)
247.             if show_all:
248.                 files_found.append((rel_path, size))
249.             else:
250.                 ext = os.path.splitext(filename)[1].lower()
251.                 if ext in filter_exts:
252.                     files_found.append((rel_path, size))
253.     return {'files': sorted(files_found), 'header': f'Directory: {dir_path}\nTotal files: {len(files_found)}\n' + '=' * 80}
255. def list_files_in_zip(zip_path, filter_exts, show_all):
256.     if not os.path.exists(zip_path):
257.         return {'error': f'File not found - {zip_path}', 'files': [], 'header': ''}
259.     with zipfile.ZipFile(zip_path, 'r') as z:
260.         files_found = []
261.         for name in z.namelist():
262.             info = z.getinfo(name)
263.             if info.file_size == 0:
264.                 continue
265.             if show_all:
266.                 files_found.append((name, info.file_size))
267.             else:
268.                 ext = os.path.splitext(name)[1].lower()
269.                 if ext in filter_exts:
270.                     files_found.append((name, info.file_size))
271.         return {'files': sorted(files_found), 'header': f'Archive: {zip_path}\nTotal files: {len(files_found)}\n' + '=' * 80}
273. def main():
274.     import sys
275.    
276.     parser = argparse.ArgumentParser(
277.         description='List files in a zip archive with filtering by type',
278.         formatter_class=argparse.RawDescriptionHelpFormatter,
279.         epilog='''
280. Examples:
281.   python list_zip_contents.py --target_path archive.zip --all
282.   python list_zip_contents.py --target_path archive.zip --type java py js
283.   python list_zip_contents.py --target_path archive.zip --type sql
284.   python list_zip_contents.py archive.zip --type c cpp h
285.         '''
286.     )
288.     parser.add_argument('positional_path', nargs='?', help='Path to the zip archive')
289.     parser.add_argument('-tp', '--target_path', '--target', dest='target_path', help='Path to the zip archive')
290.     parser.add_argument('-t', '--type', '--types', dest='types', nargs='+',
291.                         help=f'File types to filter: {" ".join(EXTENSIONS.keys())}')
292.     parser.add_argument('-a', '--all', dest='show_all', action='store_true',
293.                         help='Show all files instead of code files only')
294.     parser.add_argument('-o', '--output', '--out', dest='output',
295.                         help='Output file path (e.g., result.csv, result.xlsx)')
297.     args = parser.parse_args()
299.     if args.target_path:
300.         zip_path = args.target_path
301.     elif args.positional_path:
302.         zip_path = args.positional_path
303.     else:
304.         parser.print_help()
305.         return
307.     run_dir = os.getcwd()
308.     script_dir = os.path.dirname(os.path.abspath(__file__))
309.     output_dir = run_dir if os.path.exists(run_dir) else script_dir
310.    
311.     types = args.types if args.types else list(EXTENSIONS.keys())
313.     filter_exts = set(EXTENSIONS.values())
314.     if args.types:
315.         filter_exts = {EXTENSIONS.get(t, '.' + t) for t in args.types}
317.     result = list_files_in_path(zip_path, filter_exts, args.show_all)
319.     if result.get('error'):
320.         print(result['error'])
321.         return
323.     if args.output:
324.         output_path = args.output
325.         ext = os.path.splitext(output_path)[1].lower()
326.         
327.         target_name = os.path.basename(zip_path)
328.         target_stem = os.path.splitext(target_name)[0]
329.         
330.         if not os.path.isabs(output_path):
331.             output_path = os.path.join(output_dir, output_path)
333.     filter_exts = set(EXTENSIONS.values())
334.     if args.types:
335.         filter_exts = {EXTENSIONS.get(t, '.' + t) for t in args.types}
337.     result = list_files_in_path(zip_path, filter_exts, args.show_all)
339.     if result.get('error'):
340.         print(result['error'])
341.         return
343.     if args.output:
344.         output_path = args.output
345.         ext = os.path.splitext(output_path)[1].lower()
346.         
347.         target_name = os.path.basename(zip_path)
348.         target_stem = os.path.splitext(target_name)[0]
349.         
350.         if not os.path.isabs(output_path):
351.             output_path = os.path.join(output_dir, output_path)
352.         
353.         if os.path.exists(output_path):
354.             print(f'File "{output_path}" already exists. Overwrite? (Yes/No): ')
355.             choice = input().strip().lower()
356.             if choice != 'yes' and choice != 'y':
357.                 default_name = f'{target_stem}_{ext[1:]}' if ext else target_stem
358.                 print(f'Please modify the file name to: {default_name}')
359.                 new_name = input().strip()
360.                 if new_name:
361.                     if not os.path.isabs(new_name):
362.                         new_name = os.path.join(output_dir, new_name)
363.                     output_path = new_name
364.                 else:
365.                     output_path = os.path.join(output_dir, default_name)
366.         
367.         abs_path = os.path.abspath(output_path)
368.         if ext == '.csv':
369.             import csv
370.             with open(output_path, 'w', newline='', encoding='utf-8') as f:
371.                 writer = csv.writer(f, lineterminator='\n')
372.                 writer.writerow(['Size (bytes)', 'Path'])
373.                 for path, size in result['files']:
374.                     writer.writerow([size, path])
375.             print(f'CSV saved to: {abs_path}')
376.         elif ext in ('.xlsx', '.xls'):
377.             from openpyxl import Workbook
378.             wb = Workbook()
379.             ws = wb.active
380.             ws.title = 'Files'
381.             ws.append(['Size (bytes)', 'Path'])
382.             for path, size in result['files']:
383.                 ws.append([size, path])
384.             for col in ws.columns:
385.                 max_length = 0
386.                 col_letter = col[0].column_letter
387.                 for cell in col:
388.                     if cell.value:
389.                         max_length = max(max_length, len(str(cell.value)))
390.                 ws.column_dimensions[col_letter].width = min(max_length + 2, 80)
391.             wb.save(output_path)
392.             print(f'Excel saved: {abs_path}')
393.         else:
394.             print(f'Error: Unsupported file extension - {ext}')
395.             return
396.     else:
397.         if result['header']:
398.             print(result['header'])
399.         for path, size in result['files']:
400.             print(f'{size:>12,} bytes  {path}')
402. if __name__ == '__main__':
403.     main()

复制代码

优化发起

仅支持csv、xlsx，发起添加md格式，大概结及格式转换等在线网站。
仅支持后缀，发起添加正则匹配，可以向everything那样支持恣意关键字搜刮。
埋伏影响，不支持exe格式等文件格式扫描，大概会被防火墙、EDR等产物或服务制止。

免责声明：如果侵犯了您的权益，请联系站长及时删除侵权内容，谢谢合作！qidao123.com:ToB企服之家，中国第一个企服评测及软件市场,开放入驻,技术点评得现金.