使用非空文件夹填充 TreeView

Question

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 文件的空文件夹。我怎样才能避免它？

Answer 1

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

如果目录不符合规范，一种方法是从 CreateDirectoryNode 方法 return 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;
}

Answer 2

解决方法是递归检查return上是否有文件，有则添加。

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

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

使用非空文件夹填充 TreeView

Populate TreeView with non-Empty Folders

c#

treeview