Skip to content

Commit fe04a29

Browse files
committed
Add user specifiable quote character
1 parent 87d4f6e commit fe04a29

3 files changed

Lines changed: 4 additions & 1 deletion

File tree

multinet/api/tasks/upload/csv.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ def process_row(row: Dict[str, Any], cols: Dict[str, TableTypeAnnotation.Type])
3838

3939
@shared_task(base=ProcessUploadTask)
4040
def process_csv(
41-
task_id: int, table_name: str, edge: bool, columns: Dict[str, TableTypeAnnotation.Type], delimiter: str
41+
task_id: int, table_name: str, edge: bool, columns: Dict[str, TableTypeAnnotation.Type], delimiter: str, quotechar: str
4242
) -> None:
4343
upload: Upload = Upload.objects.get(id=task_id)
4444

@@ -48,6 +48,7 @@ def process_csv(
4848
csv_rows = list(csv.DictReader(
4949
StringIO(blob_file.read().decode('utf-8')),
5050
delimiter=delimiter,
51+
quotechar=quotechar,
5152
))
5253

5354
# Cast entries in each row to appropriate type, if necessary

multinet/api/views/serializers.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -256,6 +256,7 @@ class CSVUploadCreateSerializer(UploadCreateSerializer):
256256
default=dict,
257257
)
258258
delimiter = serializers.CharField()
259+
quotechar = serializers.CharField()
259260

260261

261262
class D3JSONUploadCreateSerializer(UploadCreateSerializer):

multinet/api/views/upload.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -91,6 +91,7 @@ def upload_csv(self, request, parent_lookup_workspace__name: str):
9191
edge=serializer.validated_data['edge'],
9292
columns=serializer.validated_data['columns'],
9393
delimiter=serializer.validated_data['delimiter'],
94+
quotechar=serializer.validated_data['quotechar'],
9495
)
9596

9697
return Response(UploadReturnSerializer(upload).data, status=status.HTTP_200_OK)

0 commit comments

Comments
 (0)