Lista arquivos da Pasta e Salva em Planilha

#Lista arquivos e seus respectivos tamanhos dentro de uma Pasta e Salva em Planilha.

import os
from openpyxl import Workbook

def listar_arquivos(diretorio):
    # Lista todos os arquivos no diretório com seus tamanhos
    arquivos = []
    total_tamanho = 0
    for nome_arquivo in os.listdir(diretorio):
        caminho_completo = os.path.join(diretorio, nome_arquivo)
        tamanho = os.path.getsize(caminho_completo)  # Tamanho em bytes
        arquivos.append((nome_arquivo, tamanho))
        total_tamanho += tamanho
    return arquivos, len(arquivos), total_tamanho / (1024 * 1024), total_tamanho / (1024 * 1024 * 1024)  # Retornando a soma total em MB e GB

def salvar_em_xlsx(arquivos, diretorio, nome_arquivo, num_arquivos, total_tamanho_mb, total_tamanho_gb):
    # Caminho completo para o arquivo Excel
    caminho_completo = os.path.join(diretorio, nome_arquivo)

    # Cria uma nova planilha
    wb = Workbook()
    ws = wb.active

    # Adiciona cabeçalhos
    ws['A1'] = 'Nome do Arquivo'
    ws['B1'] = 'Tamanho em Megabytes'

    # Escreve os nomes dos arquivos e seus tamanhos na planilha
    for i, (arquivo, tamanho) in enumerate(arquivos, start=2):
        ws[f'A{i}'] = arquivo
        ws[f'B{i}'] = tamanho / (1024 * 1024)  # Convertendo bytes para megabytes

    # Escreve o número de arquivos no final da planilha
    ws[f'A{num_arquivos + 3}'] = 'Número de arquivos:'
    ws[f'B{num_arquivos + 3}'] = num_arquivos

    # Escreve a soma total dos tamanhos dos arquivos em megabytes no final da planilha
    ws[f'A{num_arquivos + 4}'] = 'Tamanho total (MB):'
    ws[f'B{num_arquivos + 4}'] = total_tamanho_mb

    # Escreve a soma total dos tamanhos dos arquivos em gigabytes no final da planilha
    ws[f'A{num_arquivos + 5}'] = 'Tamanho total (GB):'
    ws[f'B{num_arquivos + 5}'] = total_tamanho_gb

    # Salva a planilha
    wb.save(caminho_completo)

def main():
    # Diretório do script
    diretorio_script = os.path.dirname(__file__)

    # Lista os arquivos no diretório do script, calcula o número total de arquivos e a soma total dos tamanhos
    arquivos, num_arquivos, total_tamanho_mb, total_tamanho_gb = listar_arquivos(diretorio_script)

    # Nome do arquivo Excel para salvar
    nome_arquivo = input("Digite o nome do arquivo Excel: ")

    # Adiciona a extensão .xlsx se não estiver presente
    if not nome_arquivo.endswith('.xlsx'):
        nome_arquivo += '.xlsx'

    # Salva os nomes dos arquivos em uma planilha Excel na mesma pasta
    salvar_em_xlsx(arquivos, diretorio_script, nome_arquivo, num_arquivos, total_tamanho_mb, total_tamanho_gb)

    print(f"Nomes dos arquivos foram salvos em {nome_arquivo}.")
    print(f"Número de arquivos: {num_arquivos}")
    print(f"Tamanho total dos arquivos: {total_tamanho_mb} MB")
    print(f"Tamanho total dos arquivos: {total_tamanho_gb} GB")

if __name__ == "__main__":
    main()