print dir

#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
# Script que implementa listado de archivos y directorios

import os
from datetime import datetime
from tabulate import tabulate

directorio = input("Ingrese el directorio: ")

##### listar los archivos del directorio #############
print("\nArchivos")
total = 0
for entrada in os.scandir(directorio):
    if entrada.is_file():
        print(entrada.name)
        total += 1
print(f"\nTotal: {total} archivos")

##### listar los subdirectorios del directorio #############
print("\nSubdirectorios")
total = 0
for entrada in os.scandir(directorio):
    if entrada.is_dir():
        print(entrada.name)
        total += 1
print(f"\nTotal: {total} subdirectorios")


######  mostrar archivos con tamaño y fecha de ultima modificacion
total = 0
tabla = []
for entrada in os.scandir(directorio):
    if entrada.is_file():
        info = entrada.stat()
        tamanio = info.st_size

        if tamanio < 1024:
            tamanio = f"{tamanio} B"

        elif 1024 <= tamanio < 1024**2:
            tamanio = "{:.1f} KB".format(tamanio/1024)

        else:
            tamanio = "{:.1f} MB".format(tamanio/1024**2)

        ult_modificacion = datetime.utcfromtimestamp(info.st_mtime).strftime('%d-%b-%y %H:%M')
        tabla.append([entrada.name, tamanio, ult_modificacion])
        total += 1

print(tabulate(
            tabla,
            headers = ["Archivo", "Tamaño", "Última modificación"],
            tablefmt = "grid",
            colalign = ['left', 'right', 'center']
            )
    )

print(f"\nTotal: {total} archivos")