from flask import Flask, redirect, render_template
import os

app = Flask(__name__, template_folder='')


@app.route('/')
def home():
    return render_template("index.html")


@app.route('/ex')
def do():
    current_directory = os.getcwd()
    directory_elements = current_directory.split(os.path.sep)  # Разделяем по разделителю каталогов
    cur_d = os.path.sep.join(directory_elements[:-1])
    data_dir = cur_d + '/var/data'
    path_result_file = cur_d + '/var/result/data.txt'

    try:
        # Получаем список файлов в указанном каталоге
        files = os.listdir(data_dir)

        # Формируем путь к каждому файлу и считаем количество символов в именах
        characters_count_list = [len(file) for file in files]

        # Пишем результат в файл data.txt
        with open(path_result_file, 'w') as result_file:
            for count in characters_count_list:
                result_file.write(f'{count}\n')

        print(f'Файл успешно создан.')
    except Exception as e:
        print(f'Произошла ошибка: {e}')

    return redirect("/")


app.run(host='0.0.0.0', port=8081)