Advertisement
teslariu

buscar

Nov 1st, 2021
95
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
  1. #!/usr/bin/env python
  2. # -*- coding: utf-8 -*-
  3.  
  4. """
  5. Script que recibe como argumentos una ruta y una extensión y devuelve
  6. un listado con todos los archivos de ese tipo encontrados
  7. Ej: python buscar_archivos.py <ruta> <extension>
  8.  
  9. Por ejemplo, para buscar todos los pdfs del disco C:
  10.  
  11. >>> python buscar_archivos.py C: .pdf
  12. """
  13.  
  14. import sys
  15. import os
  16.  
  17. hallados = 0
  18.  
  19. # verificar la cantidad de argumentos
  20. if len(sys.argv) != 3:
  21.     sys.exit("Error: debe ejecutar 'python buscar_archivos.py <ruta> <extension>'")
  22.  
  23. # asigno los parámetros
  24. ruta, extension = sys.argv[1:]
  25.  
  26. # verifico la ruta
  27. if not os.path.exists(ruta):
  28.     sys.exit(f"Error, no existe la ruta '{ruta}'")
  29.  
  30. # os.walk() devuelve 3 valores en cada iteración:
  31. # el nombre de la carpeta actual
  32. # una lista de carpetas dentro de la carpeta actual
  33. # una lista de archivos de la carpeta actual
  34. for carpeta, lista_carpetas, lista_archivos in os.walk(ruta):
  35.     for archivo in lista_archivos:
  36.         if archivo.endswith(extension):
  37.             print(os.path.join(carpeta, archivo))
  38.             hallados += 1
  39.            
  40. if hallados:
  41.     print(f"Total de archivos hallados: {hallados}")
  42. else:
  43.     print(f"No se hallaron archivos con extension '{extension}' en '{ruta}'")
  44.  
Advertisement
RAW Paste Data Copied
Advertisement