DAS_2024_1/kashin_maxim_lab_2/program_1/main.py

59 lines
2.4 KiB
Python

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.) выполнен.")