1414)
1515from modules .eximport .codes .import_codes import import_codes_to_proj
1616from modules .eximport .cota .import_cota import import_cota_to_proj
17+ from modules .eximport .folder .import_folders import import_folders_to_proj
1718from modules .eximport .memos .import_memos import import_memos_to_proj
1819from modules .eximport .project_metadata .import_project_metadata import (
1920 import_project_metadata_to_proj ,
@@ -38,6 +39,7 @@ class ExportEntity(Enum):
3839 BBOX_ANNOTATION = "bbox_annotation"
3940 CODE = "code"
4041 COTA = "cota"
42+ FOLDER = "folder"
4143 MEMO = "memo"
4244 PROJECT_METADATA = "project_metadata"
4345 SDOC = "sdoc"
@@ -56,6 +58,8 @@ def get_regex(self) -> str:
5658 return r"project_\d+_all_codes.csv"
5759 case ExportEntity .COTA :
5860 return r"project_\d+_all_cota.csv"
61+ case ExportEntity .FOLDER :
62+ return r"project_\d+_all_folders.csv"
5963 case ExportEntity .MEMO :
6064 return r"project_\d+_all_memos.csv"
6165 case ExportEntity .PROJECT_METADATA :
@@ -83,6 +87,8 @@ def import_entity(self, db: Session, data, project_id: int) -> None:
8387 import_codes_to_proj (db = db , df = data , project_id = project_id )
8488 case ExportEntity .COTA :
8589 import_cota_to_proj (db = db , df = data , project_id = project_id )
90+ case ExportEntity .FOLDER :
91+ import_folders_to_proj (db = db , df = data , project_id = project_id )
8692 case ExportEntity .MEMO :
8793 import_memos_to_proj (db = db , df = data , project_id = project_id )
8894 case ExportEntity .PROJECT_METADATA :
@@ -110,6 +116,7 @@ def import_entity(self, db: Session, data, project_id: int) -> None:
110116 ExportEntity .USER ,
111117 ExportEntity .CODE ,
112118 ExportEntity .TAG ,
119+ ExportEntity .FOLDER ,
113120 ExportEntity .SDOC ,
114121 ExportEntity .BBOX_ANNOTATION ,
115122 ExportEntity .SPAN_ANNOTATION ,
0 commit comments