新增工程数据

backend/app/engine/loaders/__init__.py

@@ -3,8 +3,10 @@ import yaml
 from app.engine.loaders.db import DBLoaderConfig, get_db_documents
 from app.engine.loaders.file import FileLoaderConfig, get_file_documents
 from app.engine.loaders.web import WebLoaderConfig, get_web_documents
+from app.engine.loaders.projectJson import getProjectName
 import os
 
+
 logger = logging.getLogger(__name__)
 
 def load_configs():
@@ -55,6 +57,31 @@ def get_document_Types():
             types.append(path_difference(rootPath,curDir))
     return types
 
+def getProjectInfos():
+    config = load_configs()
+    if config is None or len(config.items()) == 0:
+        return None
+      
+    prjDir = None
+    for loader_type, loader_config in config.items():
+        if loader_config.get('enable', True):  
+            loader_config = loader_config or []
+            config = FileLoaderConfig(**loader_config)
+            prjDir = config.data_dir
+            break
+    if prjDir is None:
+        return None
+    
+    prjInfos = []
+    prjFlags = get_document_Types()
+    for prjFlag in prjFlags:
+        fileDir = os.path.join(config.data_dir,prjFlag.replace('_','\\'))
+        prjInfo = {}
+        prjInfo['flag'] = prjFlag
+        prjInfo['name'] = getProjectName(fileDir)
+        prjInfos.append(prjInfo)
+    return prjInfos   
+
 def get_documents(docType:str):
     documents = []
     config = load_configs()
@@ -80,4 +107,4 @@ def get_documents(docType:str):
                     raise ValueError(f"Invalid loader type: {loader_type}")
             documents.extend(document)
 
-    return documents
+    return documents