Tix)dZddlZddlZddlZddlZddlmZmZmZm Z m Z m Z ddl m Z ddlmZddlmZddlmZddlmZed Zej/d d Zej/d dZej5d defdZej/ddZej/ddZej5ddefdZGddeZej5ddefdZ ej/ddZ!ejEdde#fdZ$ej5ddefd Z%ej/d!d"Z&ej/d#d$Z'y)%uF Settings Router - 设置路由模块 该模块提供设置相关的 API 路由端点，包括： - 设置文件存在性检查 - 设置的获取和更新 - 代理配置管理 - 代理连接测试 主要端点： - GET /api/settings/exists - 检查设置文件是否存在 - GET /api/settings - 获取所有设置（敏感信息已掩码） - POST /api/settings - 更新设置 - GET /api/settings/proxy/status - 获取代理状态 - GET /api/settings/proxy/test - 测试代理连接 - GET /api/settings/proxy - 获取代理设置 - POST /api/settings/proxy - 更新代理设置 - GET /api/settings/knowledge/enabled - 获取启用的知识库列表 依赖模块： - services.settings_service - 设置服务 - services.db_service - 数据库服务 - services.config_service - 配置服务 - services.knowledge_service - 知识库服务 N) APIRouter HTTPExceptionRequest UploadFileFileForm) db_service)settings_service) tool_service)list_user_enabled_knowledge) BaseModelz /api/settings)prefixz/existscJKdtjd{iS7w)u+ 检查设置文件是否存在 Returns: dict: 包含 exists 字段的字典，指示设置文件是否存在 Description: 用于前端检查是否需要显示初始设置向导。 如果设置文件不存在，通常需要引导用户进行初始配置。 existsN)r exists_settingsJ/home/jay/workspace/tools/ai-image-gen/jaaz-app/server/routers/settings.pysettings_existsr)s# ,<<>> ??>s #!#c2KtjSw)uy 获取所有设置配置 Returns: dict: 完整的设置配置字典，敏感信息已被掩码处理 Description: 返回所有应用设置，包括代理配置、系统提示词等。 敏感信息（如密码）会被替换为 '*' 字符以保护隐私。 设置会与默认配置合并，确保所有必需的键都存在。 )r get_settingsrrrrr8s  ( ( **srequestcK|jd{}tj|d{}|S7#7w)u 更新设置配置 Args: request (Request): HTTP 请求对象，包含要更新的设置数据 Returns: dict: 操作结果，包含 status 和 message 字段 Description: 接收 JSON 格式的设置数据并更新到配置文件。 支持部分更新，新数据会与现有设置合并而不是完全替换。 Example: POST /api/settings { "proxy": "http://proxy.com:8080" // 或 "no_proxy" 或 "system" } N)jsonr update_settings)rdataresults rrrHs7* D#33D9 9F M 9s>:><>>z /proxy/statuscKtj}|jdd}|dk(rddddS|dk(rddddS|jd rddd dSddd dSw) u 获取代理配置状态 Returns: dict: 代理状态信息，包含以下字段： - enable (bool): 代理是否启用 - configured (bool): 代理是否正确配置 - message (str): 状态描述信息 Description: 检查当前代理配置的状态，包括是否启用和是否正确配置。 该端点不会暴露完整的代理 URL 以保护安全性。 Status Logic: - enable=True, configured=True: 代理已启用且配置正确 - enable=True, configured=False: 代理已启用但配置有误 - enable=False, configured=False: 代理未启用 proxysystemno_proxyFTzProxy is disabled)enable configuredmessagezUsing system proxyzhttp://zhttps://z socks4://z socks5://zUsing custom proxyzProxy configuration is invalid)r get_raw_settingsget startswith)settings proxy_settings rget_proxy_statusr,bs* 002HLL(3M "*  ( "+   ! !"S T+  7  sAAz/proxyc:Ktj}d|iSw)uh 获取代理设置 Returns: dict: 代理配置字典，包含 proxy 字段 Description: 仅返回代理相关的设置，不包含其他配置项。 用于前端代理设置页面的数据加载。 Response Format: { "proxy": "no_proxy" | "system" | "http://proxy.example.com:8080" } r )r get_proxy_config) proxy_configs rget_proxy_settingsr0s!"$446L \ ""scNK|jd{}t|trd|vr tdd|d}t|ts tdd|dvr|j ds tdd t jd|id{}|S77w) u 更新代理设置 Args: request (Request): HTTP 请求对象，包含代理配置数据 Returns: dict: 操作结果，包含 status 和 message 字段 Raises: HTTPException: 当代理配置数据格式不正确时抛出 400 错误 Description: 仅更新代理相关的设置，不影响其他配置项。 代理配置应该是一个包含 "proxy" 键的对象。 Example: POST /api/settings/proxy { "proxy": "no_proxy" // 不使用代理 } 或 { "proxy": "system" // 使用系统代理 } 或 { "proxy": "http://proxy.example.com:8080" // 使用指定代理 } Nr z@Invalid proxy configuration. Expected format: {'proxy': 'value'} status_codedetailzProxy value must be a string)r"r!r&zGInvalid proxy value. Must be 'no_proxy', 'system', or a valid proxy URL)r isinstancedictrstrr)r r)r proxy_data proxy_valuers rupdate_proxy_settingsr;s@||~%J j$ '7*+DUW WW%K k3 '13 3 009O9OQB:C\^ ^ $33Wk4JK KF M1&.Ls"B%B!BB%B#B%#B%cDeZdZUeed<eed<eed<eed<dZeed<y)CreateWorkflowRequestnameapi_json descriptioninputsNoutputs)__name__ __module__ __qualname__r8__annotations__r7listrBrrrr=r=s! IN LGSrr=z/comfyui/create_workflowcK|js tdd|js tdd|js tdd|js tdd |jj dd}t j|j}t j|j}t j|j}tj|||j||d{tjd{d d iS7$7 #t$r}tdd t|d}~wwxYww) Nr2zName is requiredr3zAPI JSON is requiredzDescription is requiredzInputs are required _successTzFailed to create workflow: )r>rr?r@rAreplacerdumpsrBr create_comfy_workflowr initialize Exceptionr8)rr>r?rArBes rcreate_workflowrRs. <<4FGG   4JKK   4MNN >>4IJJ L||##C-::g../GNN+**W__-..tXw?R?RTZ\cddd%%'''4   e' L&A#a&$JL LLsOA%E(BD0D,D0#D.$D0+E,D0.D00 E9EEEz/comfyui/list_workflowscFKtjd{S7wN)r list_comfy_workflowsrrrlist_workflowsrVs002 22 2s !!z/comfyui/delete_workflow/{id}idcKtj|d{}tjd{|S7"7wrT)r delete_comfy_workflowr rO)rWrs rdelete_workflowrZ s933B7 7F  ! ! ### M8#sA>AAAAz/comfyui/proxycK |jd{}|jd}|jdd}|r|s tdd||}tj4d{}|j|d{}|jcdddd{S77@7)7 #1d{7swYyxYw#t $r}tddt |d}~wwxYww) Nurlpathrr2z'Missing 'url' or 'path' in request bodyr3izProxy request failed: )rr(rhttpx AsyncClientrPr8)rr target_urlr]full_urlclientresponserQs r comfyui_proxyrdsG\\^#XXe_ xx#(QS S!\$($$& # #&#ZZ11H==? # # #$ #1 # # # # G& <CC C4C C1C,,C11C4z/knowledge/enabledcK t}d|t|dS#t$r}dt|gdcYd}~Sd}~wwxYww)ui 获取启用的知识库列表 Returns: dict: 包含启用知识库列表的响应 T)rKrcountF)rKerrorrN)r lenrPr8)knowledge_listrQs rget_enabled_knowledgerj-sS  46"(   V   s(AA ?:?A?Az/my_assets_dir_pathcKddlm} tj|dd|ddS#t$r}dt |dd cYd }~Sd }~wwxYww) uj 获取用户的My Assets目录路径 Returns: dict: 包含目录路径的响应 r) FILES_DIRT)exist_okz/My Assets directory path retrieved successfully)rKr]r%Fr)rKrgr]N)services.config_servicerlosmakedirsrPr8)rlrQs rget_my_assets_dir_pathrqDs[2  I-H   V   s.A 'A  A AA A A  A )(__doc__rroshutilr^fastapirrrrrrservices.db_servicer services.settings_servicer services.tool_servicer services.knowledge_servicer pydanticr routerr(rrpostrr,r0r;r=rRrVdeleteintrZrdrjrqrrrr~s2 MM*6.B / *I @ @B + +R72O2 2 jH##(X777tI '(L#8L)L, %&3'3./c0  GGG2 ! " , !" # r