Home_Task/HT_8/task1.py - main - IvanKhabarov/python

Home_Task/HT_8/task1.py
72 строки3 KB
Ivan-Khabarov
added new tasks
24 сен 2024, 20:04
24 сен 2024, 20:04950cd31
import os
import json
import csv
import pickle
from pathlib import Path

def get_size(path):
    """Возвращает размер файла или директории."""
    if os.path.isfile(path):
        # Если путь - файл, возвращаем его размер
        return os.path.getsize(path)
    elif os.path.isdir(path):
        total_size = 0
        # Если путь - директория, рекурсивно вычисляем размер всех
        # файлов в директории
        for dirpath, _, filenames in os.walk(path):
            for filename in filenames:
                file_path = os.path.join(dirpath, filename)
                total_size += os.path.getsize(file_path)
        return total_size

def traverse_directory(directory):
    """Рекурсивно обходит директорию и возвращает информацию о файлах
    и директориях."""
    result = []
    # Обход директории с помощью os.walk, который возвращает корневую
    # директорию, поддиректории и файлы
    for root, dirs, files in os.walk(directory):
        for name in dirs + files:
            path = os.path.join(root, name)
            is_dir = os.path.isdir(path)
            size = get_size(path)
            parent = os.path.basename(root)
            # Добавление информации о текущем объекте в список результата
            result.append({
                'name': name,
                'path': path,
                'type': 'directory' if is_dir else 'file',
                'size': size,
                'parent': parent
            })
    return result

def save_to_json(data, filename):
    """Сохраняет данные в формате JSON."""
    with open(filename, 'w') as json_file:
        json.dump(data, json_file, indent=4)

def save_to_csv(data, filename):
    """Сохраняет данные в формате CSV."""
    with open(filename, 'w', newline='') as csv_file:
        writer = csv.DictWriter(csv_file, fieldnames=['name', 'path',
                               'type', 'size', 'parent'])
        writer.writeheader()
        writer.writerows(data)

def save_to_pickle(data, filename):
    """Сохраняет данные в формате Pickle."""
    with open(filename, 'wb') as pickle_file:
        pickle.dump(data, pickle_file)

def main(directory):
    """Основная функция, которая выполняет обход директории и
    сохраняет результаты."""
    data = traverse_directory(directory)
    save_to_json(data, 'directory_info.json')
    save_to_csv(data, 'directory_info.csv')
    save_to_pickle(data, 'directory_info.pkl')

if __name__ == "__main__":
    # Замените 'your_directory' на путь к вашей директории
    main('your_directory')