Script de Python para obtener el directorio de archivos y el nombre del archivo en el directorio especificado

1. Introducción

En el proceso diario de depuración, a menudo se encuentran que se agregan lotes de archivos de origen.Este artículo presenta un método de uso de secuencias de comandos de Python para obtener todos los directorios de archivos y nombres de archivos especificados en un directorio específico como referencia.

Dos, introducción al guión

import os

def get_dir_and_file(path, file_set, dir_set):
	file_list = os.listdir(path)
	for file in file_list:
		#print(file)
		cur_path = os.path.join(path, file)
		if os.path.isdir(cur_path):
			dir_set.add(cur_path)
			get_dir_and_file(cur_path, file_set, dir_set)
		else:
			if file.endswith(".c"):
				file_set.add(cur_path)
			if file.endswith(".txt"):
				file_set.add(cur_path)

if __name__ == "__main__":
	file_set = set([])
	dir_set = set([])
	get_dir_and_file("./files", file_set, dir_set)
	with open('./file_name.txt', 'w', encoding='utf-8') as f:  # txtname 根据具体所需的命名即可
		for val_name in file_set:
			f.write(val_name + '\n')

	with open('./dir_name.txt', 'w', encoding='utf-8') as f:  # txtname 根据具体所需的命名即可
		for val_name in dir_set:
			f.write(val_name + '\n')

	print('-----------------')
	print(file_set)
	print('-----------------')
	print(dir_set)
	print('finish!\r\n')

tres, resumen

Este artículo presenta principalmente cómo encontrar el directorio y el nombre del archivo en la ruta especificada y escribirlo en el archivo TXT correspondiente como referencia. Bienvenido a discutir e intercambiar ~