DAS_2024_1/kashin_maxim_lab_2/program_1/main.py

from pathlib import Path
import os

# Путь к папке, где будут создаваться файлы
folder_path = '/result'
# Путь к папке, где созданы файлы
folder_generator = '/data'

# Функция для создания папки, если она не существует
def creat_folder():
    if not os.path.exists(folder_path):
        os.makedirs(folder_path)
        print(f"Папка `{folder_path}` создана!")
    else:
        print(f"Папка `{folder_path}` уже существует!")

# Функция для удаления всех файлов в папке
def delete_files_in_folder():
    for filename in os.listdir(folder_path):
        file_path = os.path.join(folder_path, filename)
        try:
            if os.path.isfile(file_path):
                os.remove(file_path)
                print(f"Файл `{file_path}` удален.")
        except Exception as e:
            print(f'Ошибка при удалении файла {file_path}. {e}')

# Функция для создания файлов
def creat_file():
    file = open(f'{folder_path}/data.txt', 'w+')
    print(f"Создан файл: {folder_path}/data.txt")
    for count in files_count():
        file.write(f'{count} \n')
    file.close()

# Функция для подсчета количества строк в файлах
def files_count():
    files = []
    files_count = []
    files += os.listdir(folder_generator)
    for file in files:
        line_count = sum(1 for line in open(folder_generator + '/' + file))
        files_count.append(line_count)
        print(f"Файл {file} имеет {line_count} количество строк")
    return files_count


# Основная логика: проверка существования папки и выполнение операций
if os.path.exists(folder_path):
    print(f"Папка `{folder_path}` существует, удаляем старые файлы...")
    delete_files_in_folder()
    creat_file()
else:
    print(f"Папка `{folder_path}` не существует, создаем...")
    creat_folder()
    creat_file()

print("Скрипт на задание 1.3 (Формирует файл /var/result/data.txt так, что каждая строка файла - количество строк в файлах из каталога /var/data.) выполнен.")