Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Feature/json export #778

Closed
wants to merge 8 commits into from
Closed
Prev Previous commit
Next Next commit
add format for CSVPainter method paint
  • Loading branch information
cgill95 committed May 19, 2020
commit 4e30538fe1822dd70eebeb32639dd2973e8cab0b
4 changes: 2 additions & 2 deletions app/api/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -456,8 +456,8 @@ def paint_labels(documents, labels, export_format='jsonl'):

class CSVPainter(JSONPainter):

def paint(self, documents):
data = super().paint(documents)
def paint(self, documents, export_format='csv'):
data = super().paint(documents, export_format='csv')
res = []
for d in data:
annotations = d.pop('annotations')
Expand Down
8 changes: 4 additions & 4 deletions app/api/views.py
Original file line number Diff line number Diff line change
Expand Up @@ -333,14 +333,14 @@ def get(self, request, *args, **kwargs):
# json and jsonl format - "annotations": [{"label": 5, "start_offset": 0, "end_offset": 2, "user": 1},..]
if format == 'jsonltext':
labels = project.labels.all()
data = JSONPainter.paint_labels(documents, labels, 'jsonl')
data = JSONPainter.paint_labels(documents, labels, export_format='jsonl')
elif format == 'jsontext':
labels = project.labels.all()
data = JSONPainter.paint_labels(documents, labels, 'json')
data = JSONPainter.paint_labels(documents, labels, export_format='json')
elif format == 'jsonl':
data = painter.paint(documents, 'jsonl')
data = painter.paint(documents, export_format='jsonl')
else:
data = painter.paint(documents, 'json')
data = painter.paint(documents, export_format='json')

return Response(data)

Expand Down