用非空文件夹填充TreeView

本文关键字：填充 TreeView 文件夹 | 更新日期: 2023-09-27 18:15:27

private void ListDirectory(TreeView treeView, string path)
{
    treeView.Nodes.Clear();
    var rootDirectoryInfo = new DirectoryInfo(path);
    treeView.Nodes.Add(CreateDirectoryNode(rootDirectoryInfo));
}
private static TreeNode CreateDirectoryNode(DirectoryInfo directoryInfo)
{
    var directoryNode = new TreeNode(directoryInfo.Name);
    foreach (var directory in directoryInfo.GetDirectories())
        directoryNode.Nodes.Add(CreateDirectoryNode(directory));
    foreach (var file in directoryInfo.GetFiles().Where(s => s.Extension == ".jpg"))
        directoryNode.Nodes.Add(new TreeNode(file.Name));
    return directoryNode;
}
使用这段代码，我可以列出给定目录中的所有jpg文件和文件夹。但它也会列出jpg文件的空文件夹。我怎样才能避免呢?

用非空文件夹填充TreeView

解决方案是检查递归返回的文件中是否有文件，如果有则添加。

foreach (var directory in directoryInfo.GetDirectories())
{
    TreeNode subNode = CreateDirectoryNode(directory);
    if (subNode.Nodes.Count > 0)
        directoryNode.Nodes.Add(CreateDirectoryNode(directory));
}

由于您正在进行深度优先搜索，如果没有任何文件，节点将有效地从叶子到根进行剔除。

检查子目录的个数是否为零，目录下的文件个数是否为零。如果是这种情况，则不应该创建该节点。

一种方法是，如果目录不符合规范，则从CreateDirectoryNode方法返回null。然后只添加非null的子节点

考虑以下代码:

private void ListDirectory(TreeView treeView, string path)
{
    treeView.Nodes.Clear();
    var rootDirectoryInfo = new DirectoryInfo(path);
    var tree_node = CreateDirectoryNode(rootDirectoryInfo);
    if (tree_node != null)
        treeView.Nodes.Add(tree_node);
}
//This method will return null of the specified directory does not have sub folders or JPG files
private static TreeNode CreateDirectoryNode(DirectoryInfo directoryInfo)
{
    //Obtain all sub directories
    var sub_directories = directoryInfo.GetDirectories();
    //Obtain all JPG files
    var jpeg_files =
        directoryInfo.GetFiles()
            .Where(s => s.Extension.Equals(".jpg", StringComparison.OrdinalIgnoreCase))
            .ToArray();
    //If the above arrays are empty, return null
    if (sub_directories.Length == 0 && jpeg_files.Length == 0)
        return null;
    var directoryNode = new TreeNode(directoryInfo.Name);
    foreach (var directory in sub_directories)
    {
        var sub_node = CreateDirectoryNode(directory);
        if(sub_node != null) //Only add sub nodes if they are not null
            directoryNode.Nodes.Add(sub_node);
    }
    foreach (var file in jpeg_files)
        directoryNode.Nodes.Add(new TreeNode(file.Name));

    return directoryNode;
}
相关文章：