mirror of
https://github.com/YFGaia/dify-plus.git
synced 2026-06-12 18:11:42 +08:00
npc0-hue
17832f2424
本次提交整合了多个功能改进和问题修复： 主要功能： - 批量工作流处理功能完善，支持 Excel 上传和进度跟踪 - 管理中心反向代理和转发配置优化 - 用户同步添加互斥锁，防止并发问题 - 计费系统和额度显示优化 - AI 绘图功能扩展 前端改进： - 文本生成应用显示修复 - 批量任务进度展示优化 - 按钮样式和 CSS 优化，禁止换行 - 多语言支持完善（新增印尼语等） - 构建镜像逻辑优化 - 批量处理进度管理器实现 后端改进： - Docker Compose 配置升级 - 队列任务和 Worker Pool 优化 - Admin API 初始化和验证逻辑改进 - 数据库迁移和初始化完善 - 静态变量处理优化 - URL 签名助手实现 - Celery 扩展优化 - 代码和导入包问题修复（idea 自动调整代码位置） 技术改进： - 兼容性修复 (flask-restx, jschardet) - 钉钉 Web API 版本更新 - 代码格式化和导入包问题修复 - 日志处理优化 - 工作流循环管理优化 Docker 相关： - Nginx 配置更新 - 容器启动脚本优化 - 镜像构建流程改进 - docker-compose.dify-plus.yaml 大幅更新 管理后台： - 工作流批量处理 API 实现 - 工作池初始化 - 批量工作流服务实现 - 转发扩展配置 - 用户服务扩展
61 lines
2.4 KiB
Python
61 lines
2.4 KiB
Python
|
|
from flask_login import current_user
|
|
|
|
from configs import dify_config
|
|
from extensions.ext_database import db
|
|
from models.account import Tenant, TenantAccountJoin, TenantAccountRole
|
|
from services.account_service import TenantService
|
|
from services.account_service_extend import TenantExtendService
|
|
from services.feature_service import FeatureService
|
|
|
|
|
|
class WorkspaceService:
|
|
@classmethod
|
|
def get_tenant_info(cls, tenant: Tenant):
|
|
if not tenant:
|
|
return None
|
|
tenant_info = {
|
|
"id": tenant.id,
|
|
"name": tenant.name,
|
|
"plan": tenant.plan,
|
|
"status": tenant.status,
|
|
"created_at": tenant.created_at,
|
|
"trial_end_reason": None,
|
|
"role": "normal",
|
|
}
|
|
|
|
# Get role of user
|
|
tenant_account_join = (
|
|
db.session.query(TenantAccountJoin)
|
|
.where(TenantAccountJoin.tenant_id == tenant.id, TenantAccountJoin.account_id == current_user.id)
|
|
.first()
|
|
)
|
|
assert tenant_account_join is not None, "TenantAccountJoin not found"
|
|
tenant_info["role"] = tenant_account_join.role
|
|
|
|
# ----------------------- 二开部分Start 添加用户权限 - ----------------------
|
|
tenant_extend_service = TenantExtendService
|
|
super_admin_id = tenant_extend_service.get_super_admin_id().id
|
|
super_admin_tenant_id = tenant_extend_service.get_super_admin_tenant_id().id
|
|
tenant_info["admin_extend"] = (super_admin_id == current_user.id)
|
|
tenant_info["tenant_extend"] = (super_admin_tenant_id == tenant.id)
|
|
# ----------------------- 二开部分Stop 添加用户权限 - ----------------------
|
|
|
|
can_replace_logo = FeatureService.get_features(tenant.id).can_replace_logo
|
|
|
|
if can_replace_logo and TenantService.has_roles(tenant, [TenantAccountRole.OWNER, TenantAccountRole.ADMIN]):
|
|
base_url = dify_config.FILES_URL
|
|
replace_webapp_logo = (
|
|
f"{base_url}/files/workspaces/{tenant.id}/webapp-logo"
|
|
if tenant.custom_config_dict.get("replace_webapp_logo")
|
|
else None
|
|
)
|
|
remove_webapp_brand = tenant.custom_config_dict.get("remove_webapp_brand", False)
|
|
|
|
tenant_info["custom_config"] = {
|
|
"remove_webapp_brand": remove_webapp_brand,
|
|
"replace_webapp_logo": replace_webapp_logo,
|
|
}
|
|
|
|
return tenant_info
|