import os


def count_file_name_lengths(source_dir, result_file):
    # Проверяем и создаем директорию для результатов
    result_dir = os.path.dirname(result_file)
    if not os.path.exists(result_dir):
        os.makedirs(result_dir)
        print(f"Created directory {result_dir}")

    # Поиск файлов в исходной директории
    files = [f for f in os.listdir(source_dir) if os.path.isfile(os.path.join(source_dir, f))]

    # Если файлы не найдены, выводим сообщение
    if not files:
        print(f"No files found in {source_dir}")
        return

    # Запись длин имен файлов в result_file
    with open(result_file, 'w') as f_out:
        for file in files:
            f_out.write(f"{len(file)}\n")

    print(f"Wrote file name lengths to {result_file}")

if __name__ == '__main__':
    count_file_name_lengths('/var/data', '/var/result/data.txt')