我在从0到1打造一个开源平台, 也在编写一套完整的接口测试平台系列教程，希望大家能够多多支持。

欢迎关注我的龚仲耗测试开发坑货，获取最新文章教程!

上一篇我们开了个小差，去搞了下编辑器那块。为什么我如此重视编辑器的体验，那是因为后续我们需要用户在页面上补足Python或其他脚本，如果没有一个还算好用的编辑器，那将非常鸡肋。

但之前发现编辑器会略卡，找到了具体的原因: 我们每次在render过程中都会添加代码补全提示，这样是不合理的，所以我们需要对编辑器进行一下改造。

其实现在的情况比较复杂，我们在写用例的时候，必须要先进入项目之中，然后在项目里面找到用例列表tab页。

有没有觉得这样不太好~

我们如果左侧是目录，右侧是case列表，其实会好很多。而且随着业务的深入，目录分类肯定不能像目前一样只有2级。

不管怎么说，我们先试验一下吧~

抛开主键，更新/创建时间和用户这样的基础信息之后，其实核心只有3个字段:

• 我们的目录需要和项目挂钩，这样的话能对目录进行很好的分类，从而达到对case进行归类的目的。

• 目录名称，这个不需要多说了。

• 这个是很关键的字段，因为我们的目录是支持嵌套的，目录如果没有父级，那么它就是一个根目录。

  我们要展示一个树结构，可离不开这玩意儿

有的人可能很疑惑，为什么明明设置了唯一索引，还要去先查询一次数据库呢？其实吧，唯一索引是咱们的兜底方案，是为了控制数据的唯一性。但我们更应该从业务的角度去规避掉重复数据出现的可能，所以有这样一层判断，这也只是个人理解哈。

可以看到代码比较清晰，通过with session.begin在session.add之后自动提交到数据库，完成对数据的录入。

基本上都是老一套，这也就是为什么大家都说CRUD BOY很没意思，因为看起来技术含量确实一般。

接着去完善修改和删除的接口:

重点来了，接着是编写获取目录树的接口了~

我们需要一次性拿出一个项目下面的所有目录。

这里稍微有点儿复杂，我们一个一个来分析:

1. 通过项目id获取到所有目录树
2. 创建最终返回结果: ans
3. 创建目录id -> 目录的映射关系
4. 创建父目录 -> 子目录的映射，子目录是一个list
5. 遍历获取到的目录，如果parent为空，说明是根目录，我们直接加到ans之中，否则说明他不是根目录，我们拿到他的parent并写入parent_map之中

如果一来，我们通过一次遍历拿到了所有根目录，以及根目录有哪些孩子，目录id->目录的3个重要线索。

1. 遍历ans，把他们的子子孙孙都塞进children列表里去

来看看get_directory方法，先通过parent_map拿到当前的父目录下面是否还有子目录->current，如果没有，则说明这一层目录已经没有子目录了，可以return了。

如果有的话，遍历current，也就是子目录列表，接着从ans_map拿到子目录的信息，新建child列表，并把当前子目录的信息添加到父目录下面的children，接着递归继续找子目录的子目录。

如此反复，即可生成一棵树。编写获取目录树的接口:

今天的内容就分享到这儿了，有兴趣的朋友们可以一起动手写起来呀！~