Advertisement
teslariu

buscar_archivos

Dec 4th, 2021
758
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. import os
  5. import sys
  6.  
  7. """
  8. Script que recibe como argumentos a traves de la consola un ruta y una
  9. extensión y devuelve una lista de los archivos encontrados y la
  10. cantidad total.
  11. Uso:
  12. >>> python buscar_archivos.py  <ruta>  <extension>
  13.  
  14. Ej:
  15. >>> python buscar_archivos.py   C:\\Users\\Usuario   .jpeg
  16. >>> python buscar_archivos.py   C:\\Users\\Usuario\\Desktop   pdf
  17. """
  18. total = 0
  19. # verifico la cantidad de argumentos:
  20. if len(sys.argv) !=3:
  21.     sys.exit("Error: ejecute 'python buscar_archivos.py <ruta> <extension>''")
  22.  
  23. # asigno los parámetros
  24. ruta, extension = sys.argv[1:]
  25.  
  26. # verifico si existe la ruta
  27. if not os.path.exists(ruta):
  28.     sys.exit(f"No existe el directorio '{ruta}'")
  29.    
  30. # busco en forma recursiva dentro de la ruta
  31. # usaremos os.walk()
  32. # os.walk() devuelve 3 valores en cada iteracion:
  33. #  1. el nombre de la carpeta actual
  34. #  2. La lista de carpetas dentro de la carpeta actual
  35. #  3. Una lista de archivos dentro de la carpeta actual
  36.  
  37. for carpeta_actual, lista_carpetas, lista_archivos in os.walk(ruta):
  38.     for archivo in lista_archivos:
  39.         if archivo.endswith(extension):
  40.             print(os.path.join(carpeta_actual,archivo))
  41.             total += 1
  42.            
  43. if not total:
  44.     print(f"No se hallaron archivos con extension '{extension}' en '{ruta}'")
  45. else:
  46.     print(total)
Advertisement
Advertisement
Advertisement
RAW Paste Data Copied
Advertisement