Browse Source

Use upload_name as an export file name

pull/1779/head
Hironsan 2 years ago
parent
commit
9490b74f36
2 changed files with 8 additions and 10 deletions
  1. 6
      backend/data_export/pipeline/repositories.py
  2. 12
      backend/data_export/tests/test_repositories.py

6
backend/data_export/pipeline/repositories.py

@ -31,7 +31,7 @@ class FileRepository(BaseRepository):
for user, label in label_per_user.items():
yield Record(
data_id=example.id,
data=str(example.filename).split("/")[-1],
data=example.upload_name,
label=label,
user=user,
metadata=example.meta,
@ -43,9 +43,7 @@ class FileRepository(BaseRepository):
# with the user who approved the doc.
# This means I will allow each user to be able to approve the doc.
if len(label_per_user) == 0:
yield Record(
data_id=example.id, data=str(example.filename).split("/")[-1], label=[], user="unknown", metadata={}
)
yield Record(data_id=example.id, data=example.upload_name, label=[], user="unknown", metadata={})
def label_per_user(self, example) -> Dict:
label_per_user = defaultdict(list)

12
backend/data_export/tests/test_repositories.py

@ -279,12 +279,12 @@ class TestSpeech2TextRepository(TestRepository):
self.prepare_data(project)
expected = [
{
"data": self.example.filename,
"data": self.example.upload_name,
"label": [self.text1.text],
"user": project.admin.username,
},
{
"data": self.example.filename,
"data": self.example.upload_name,
"label": [self.text2.text],
"user": project.annotator.username,
},
@ -297,7 +297,7 @@ class TestSpeech2TextRepository(TestRepository):
self.prepare_data(project)
expected = [
{
"data": self.example.filename,
"data": self.example.upload_name,
"label": [self.text1.text, self.text2.text],
"user": "all",
}
@ -317,12 +317,12 @@ class TestFileRepository(TestRepository):
self.prepare_data(project)
expected = [
{
"data": self.example.filename,
"data": self.example.upload_name,
"label": [self.category1.label.text],
"user": project.admin.username,
},
{
"data": self.example.filename,
"data": self.example.upload_name,
"label": [self.category2.label.text],
"user": project.annotator.username,
},
@ -335,7 +335,7 @@ class TestFileRepository(TestRepository):
self.prepare_data(project)
expected = [
{
"data": self.example.filename,
"data": self.example.upload_name,
"label": [self.category1.label.text, self.category2.label.text],
"user": "all",
}

Loading…
Cancel
Save