使用Python实现一个强大的文件系统结构创建器

def on_choose_target(self, event):
    dlg = wx.DirDialog(self, "选择目标文件夹", style=wx.DD_DEFAULT_STYLE)
    if dlg.ShowModal() == wx.ID_OK:
        self.target_folder = dlg.GetPath()
        self.target_text.SetValue(self.target_folder)
        self.tree.SetItemText(self.root, os.path.basename(self.target_folder) or "根节点")
        logging.debug(f"Updated target folder to {self.target_folder}, root node text to {self.tree.GetItemText(self.root)}")
    dlg.Destroy()

def on_import_folder(self, event):
    dlg = wx.DirDialog(self, "选择要导入的文件夹", style=wx.DD_DEFAULT_STYLE)
    if dlg.ShowModal() == wx.ID_OK:
        import_folder = dlg.GetPath()
        self.tree.DeleteChildren(self.root)
        self.tree.SetItemText(self.root, os.path.basename(import_folder) or "根节点")
        self.target_folder = import_folder
        self.target_text.SetValue(self.target_folder)
        self.populate_tree_from_folder(self.root, import_folder)
        self.tree.Expand(self.root)
        logging.debug(f"Imported folder structure from {import_folder}")
    dlg.Destroy()

​​​​​​​def populate_tree_from_folder(self, parent_item, folder_path):
    try:
        for entry in os.scandir(folder_path):
            if entry.is_dir():
                new_item = self.tree.AppendItem(parent_item, entry.name)
                self.tree.SetItemImage(new_item, 0)
                self.populate_tree_from_folder(new_item, entry.path)
            elif entry.is_file():
                new_item = self.tree.AppendItem(parent_item, f"[FILE] {entry.name}")
                self.tree.SetItemImage(new_item, 1)
    except Exception as e:
        wx.MessageBox(f"无法读取文件夹 '{folder_path}': {e}", "错误", wx.OK | wx.ICON_ERROR)

def on_clear_tree(self, event):
    self.tree.DeleteChildren(self.root)
    self.tree.SetItemText(self.root, "根节点")
    self.target_folder = ""
    self.target_text.SetValue("")
    self.tree.SelectItem(self.root)
    logging.debug("Cleared tree and reset root node")
    wx.MessageBox("树已清空！", "提示", wx.OK | wx.ICON_INFORMATION)

def is_valid_name(self, name):
    invalid_chars = r'[<>:"/\\|?*\x00-\x1F]'
    if not name or re.search(invalid_chars, name) or len(name.strip()) == 0:
        return False
    return True

def create_nodes(parent_item, parent_path):
    item, cookie = self.tree.GetFirstChild(parent_item)
    while item and item.IsOk():
        node_text = self.tree.GetItemText(item)
        current_path = os.path.join(parent_path, node_text.replace("[FILE] ", ""))
        if node_text.startswith("[FILE]"):
            try:
                if os.path.exists(current_path):
                    dlg = wx.MessageDialog(self, f"文件 '{current_path}' 已存在，是否覆盖？", 
                                         "警告", wx.YES_NO | wx.ICON_WARNING)
                    if dlg.ShowModal() != wx.ID_YES:
                        item, cookie = self.tree.GetNextChild(parent_item, cookie)
                        continue
                with open(current_path, 'w') as f:
                    pass
            except Exception as e:
                wx.MessageBox(f"创建文件 '{current_path}' 失败: {e}", "错误", wx.OK | wx.ICON_ERROR)
        else:
            try:
                os.makedirs(current_path, exist_ok=True)
                create_nodes(item, current_path)
            except Exception as e:
                wx.MessageBox(f"创建文件夹 '{current_path}' 失败: {e}", "错误", wx.OK | wx.ICON_ERROR)
        item, cookie = self.tree.GetNextChild(parent_item, cookie)