Skip to content

Commit f70a993

Browse files
committed
added folders to project eximport
1 parent c7dffe2 commit f70a993

File tree

2 files changed

+9
-0
lines changed

2 files changed

+9
-0
lines changed

backend/src/modules/eximport/project/export_project.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111
)
1212
from modules.eximport.codes.export_codes import export_all_codes
1313
from modules.eximport.cota.export_cota import export_all_cota
14+
from modules.eximport.folder.export_folders import export_all_folders
1415
from modules.eximport.memos.export_memos import export_all_memos
1516
from modules.eximport.no_data_export_error import NoDataToExportError
1617
from modules.eximport.project_metadata.export_project_metadata import (
@@ -57,6 +58,7 @@ def __export_project(
5758
export_all_bbox_annotations,
5859
export_all_codes,
5960
export_all_cota,
61+
export_all_folders,
6062
export_all_memos,
6163
export_all_project_metadatas,
6264
export_all_sdocs,

backend/src/modules/eximport/project/import_project.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@
1414
)
1515
from modules.eximport.codes.import_codes import import_codes_to_proj
1616
from modules.eximport.cota.import_cota import import_cota_to_proj
17+
from modules.eximport.folder.import_folders import import_folders_to_proj
1718
from modules.eximport.memos.import_memos import import_memos_to_proj
1819
from 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

Comments
 (0)