手把手教你给dify工作流加上用户登陆模块

“ 你平时设计完成一个dify工作流，你是怎么让别人使用的？是不是直接把工作流的分享链接发出去就不管了，还是说，就算你发出去，你很担心链接被滥用？今天手把手教你给dify工作流加把锁，通过预设一个用户登录模块，实现有限范围地控制预览链接被滥用。”

01

—

我们可以在开始节点中，预设两个变量，分别是name和password。然后用户在使用的时候，先填写name和password，再判断name和password是否符合我们预设的账户库，如果不符合，就直接报错，如果正确，就对登录状态覆盖赋值。

验证账户的代码和变量设置

我们先说说变量的预设吧：

预设两个变量：login、user。

login是记录本次对话用户的登录状态；

user是工作流开发者自己可预设的一些账号信息，只有符合预设的账户，才能正确登录。后续你可以改成http请求远程数据库，实现网络账户登录也行，本案例只讲预设账户登录。

验证账户的代码：

def main(name: str, password: str, user: list) -> dict:# 初始化结果变量a = 0# 默认值为0b = ""# 默认的错误信息
# 遍历用户数组，验证用户名和密码for u in user:if u["name"] == name and u["password"] == password:a = 1# 如果匹配成功，设置a为1b = "登录成功"# 登录成功的提示break# 找到匹配后可以提前结束循环else:# 如果循环正常结束而没有找到匹配，设置错误信息b = "账号或密码不正确"
# 返回结果return {"result": a, "message": b}

至此，一个简单的登录模块就完成了。

怎么样？是不是很简单？留个作业，你觉得注册模块要怎么做？本文是否能给你一点灵感呢？

最后我想说下，为什么要学会用dify制作一个用户登录模块。有如下几个场景，你极大可能会遇到：

1. 黑盒测试阶段：

   当你开发了一个工作流，你想邀请别人体验一下，用户在看不到你工作流的情况下，他只能正常的对话，给他一个预设的账户，用户才能参与测试，这是对测试风险的把控措施。开发者随时可以修改预设账户库，防止测试链接被滥用。

2. 有限范围的销售：

   如果你的工作流是为客户服务的，你完全可以预设账户库，将账户发给客户，反正陌生人拿到链接后滥用。