diff --git a/create_requirement_images.py b/create_requirement_images.py index 15037a9..0c9fe39 100755 --- a/create_requirement_images.py +++ b/create_requirement_images.py @@ -163,7 +163,7 @@ def make_line( ) def make_header(position, project_files, files_and_libs): - # pylint: disable=too-many-locals + # pylint: disable=too-many-locals, too-many-branches # Static files make_line( "CIRCUITPY", @@ -198,8 +198,7 @@ def make_header(position, project_files, files_and_libs): icon=file_icon, ) - # TODO: Add settings.toml if it's needed - + # Add settings.toml if it's needed if settings_required(files_and_libs): make_line( "settings.toml", @@ -247,6 +246,10 @@ def make_header(position, project_files, files_and_libs): extra_rows = 0 for i, file in enumerate(sorted(project_folders_to_draw.keys())): + if len(project_folders_to_draw[file]) > 0: + triangle_to_use = down_triangle + else: + triangle_to_use = right_triangle make_line( file, ( @@ -257,7 +260,7 @@ def make_header(position, project_files, files_and_libs): * (begin_y_offset + i + len(project_files_to_draw) + extra_rows) ), ), - triangle_icon=down_triangle, + triangle_icon=triangle_to_use, ) rows_added += 1 for sub_file in sorted(project_folders_to_draw[file]): diff --git a/get_imports.py b/get_imports.py index ea6a03e..46e9146 100644 --- a/get_imports.py +++ b/get_imports.py @@ -16,6 +16,8 @@ BUNDLE_DATA = "latest_bundle_data.json" BUNDLE_TAG = "latest_bundle_tag.json" +SUBDIRECTORY_FILECOUNT_LIMIT = 10 + LEARN_GUIDE_REPO = os.environ.get( "LEARN_GUIDE_REPO", "../Adafruit_Learning_System_Guides/" ) @@ -141,8 +143,9 @@ def get_files_for_project(project_name): if cur_tuple[0].split("/")[-1] == _dir: for _sub_dir in cur_tuple[1]: dir_tuple = (dir_tuple[0], dir_tuple[1] + (_sub_dir,)) - for _sub_file in cur_tuple[2]: - dir_tuple = (dir_tuple[0], dir_tuple[1] + (_sub_file,)) + if len(cur_tuple[2]) < SUBDIRECTORY_FILECOUNT_LIMIT: + for _sub_file in cur_tuple[2]: + dir_tuple = (dir_tuple[0], dir_tuple[1] + (_sub_file,)) # e.g. ("dir_name", ("file_1.txt", "file_2.txt")) found_files.add(dir_tuple)