Advertisement
teslariu

buscar_arch

Apr 12th, 2022
1,056
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
  1. #!/usr/bin/env python
  2. # -*- coding: utf-8 -*-
  3. # Programa que saluda
  4. """
  5. Script que recibe como argumentos una ruta a un directorio y una
  6. extensión y devuelve un listado con todos los archivos encontrados
  7. Ejemplo de uso:
  8. >>> python buscar_archivos.py C:\\Users .docx
  9. """
  10. import sys
  11. import os
  12.  
  13. hallados = 0
  14.  
  15. if sys.argv[1] == "--help":
  16.     sys.exit("Ayuda: bla bla bla")
  17.  
  18. # verifico la cantidad de argumentos
  19. if len(sys.argv) != 3:
  20.     sys.exit("Error: debe ejecutar python buscar_archivos.py <ruta> <extension>")
  21.  
  22. # asigno los argumentos
  23. ruta, extension = sys.argv[1:]
  24.  
  25. # verifico si existe la ruta
  26. if not os.path.exists(ruta):
  27.     sys.exit(f"No existe la ruta '{ruta}'")
  28.    
  29. # ahora busco los archivos
  30. # os.walk() busca con 3 iteradores
  31. # 1) el nombre de la carpeta actual
  32. # 2) la lista de carpetas dentro de la carpeta actual
  33. # 3) la lista de archivos dentro de la carpeta actual
  34.  
  35. for carpeta_actual, lista_carpetas, lista_archivos in os.walk(ruta):
  36.     for archivo in lista_archivos:
  37.         if archivo.endswith(extension):
  38.             hallados += 1
  39.             print(os.path.join(carpeta_actual, archivo))
  40.            
  41. if hallados:
  42.     print(f"Total de archivos: {hallados}")
  43. else:
  44.     print(f"No se hallaron archivos '{extension}' en '{ruta}'")
Advertisement
RAW Paste Data Copied
Advertisement