添加对数据源添加时候的提示

airda/agent/exception/already_exists_error.py

@@ -0,0 +1,5 @@
+from airda.framework.exception.agent_error import AgentError
+
+
+class AlreadyExistsError(AgentError):
+    CODE = 10002

airda/agent/storage/repositories/datasource_repository.py

@@ -1,6 +1,7 @@
 from typing import List
 
 from airda.agent import log
+from airda.agent.exception.already_exists_error import AlreadyExistsError
 from airda.agent.storage.entity.datasource import Datasource
 from airda.agent.storage.repositories.respository import Repository
 
@@ -13,7 +14,7 @@ class DatasourceRepository(Repository):
     def add(self, datasource: Datasource) -> Datasource:
         has = self.find_one(datasource.name)
         if has:
-            raise Exception("datasource already exists")
+            raise AlreadyExistsError()
         datasource_dict = datasource.dict(exclude={"id"})
         datasource.id = str(self.storage.insert_one(DB_COLLECTION, datasource_dict))
         return datasource

airda/cli/startup.py

@@ -12,6 +12,7 @@
 from airda.agent import DataAgentKey
 from airda.agent.agent import DataAgent
 from airda.agent.env import DataAgentEnv
+from airda.agent.exception.already_exists_error import AlreadyExistsError
 from airda.agent.planner.data_agent_planner_params import DataAgentPlannerParams
 from airda.agent.storage import StorageKey
 from airda.agent.storage.entity.datasource import Datasource, Kind
@@ -174,17 +175,21 @@ def add(name: str, host: str, port: int, database: str, kind: str, username: str
     datasource_repository = context.get_repository(StorageKey.DATASOURCE).convert(
         DatasourceRepository
     )
-    datasource_repository.add(
-        Datasource(
-            name=name,
-            host=host,
-            port=port,
-            database=database,
-            kind=kind,
-            username=username,
-            password=password,
+    try:
+        datasource_repository.add(
+            Datasource(
+                name=name,
+                host=host,
+                port=port,
+                database=database,
+                kind=kind,
+                username=username,
+                password=password,
+            )
         )
-    )
+        output_colored_text("执行成功", "success")
+    except AlreadyExistsError:
+        output_colored_text(f"执行失败, [{name}]数据源已存在", "error")
 
 
 @datasource.command(help="同步数据源的表字段信息并进行向量化")

airda/framework/exception/agent_error.py

@@ -1,5 +1,5 @@
 class AgentError(Exception):
     CODE = -1
 
-    def __init__(self, msg: str | None):
+    def __init__(self, msg: str = None):
         super().__init__(msg)