Uploading a CSV file to Django REST (especially in an atomic setting) is a simple task, but kept me puzzled until I found out some tricks I would be sharing with you.
In this article, I will be using postman (in place of a frontend) and will also share what you need to set on postman for request sending via pictures.
What we desire
- Upload CSV via Django Rest to the DB
- Make the operation atomic i.e any error in any row from the csv should cause complete rollback of the entire operation, so we can avoid the stress of cutting the csv file i.e the headache of identifying the portion of the rows that made it to the DB and those that didnβt due to any error midway!! (partial entry). So we want an all-or-none thing !!
Method
- Assuming, you already have Django and Django REST installed, the first step would be to install pandas, a python library for data manipulation.
pip install pandas
- Next in postman: In the body tab, select form-data and add a key (any arbitrary name). In that same cell, hover on the rightmost of the cell and use the dropdown to change option from text to file. Postman will automatically set Content-Type to multipart/form-data in Headers the moment you do this.
For the value cell, click the 'Select Files' button and upload the CSV. Check the screenshot below
Under headers, set Content-Disposition
and the value to form-data; name="file"; filename="your_file_name.csv"
. Replace your_file_name.csv with your actual file name. Check the screenshot below.
- In the Django views, the code is as follows:
from rest_framework import status
from rest_framework.views import APIView
from rest_framework.parsers import FileUploadParser
from rest_framework.response import Response
from .models import BiodataModel
from django.db import transaction
import pandas as pd
class UploadCSVFile(APIView):
parser_classes = [FileUploadParser]
def post(self,request):
csv_file = request.FILES.get('file')
if not csv_file:
return Response({"error": "No file provided"}, status=status.HTTP_400_BAD_REQUEST)
# Validate file type
if not csv_file.name.endswith('.csv'):
return Response({"error": "File is not CSV type"}, status=status.HTTP_400_BAD_REQUEST)
df = pd.read_csv(csv_file, delimiter=',',skiprows=3,dtype=str).iloc[:-1]
df = df.where(pd.notnull(df), None)
bulk_data=[]
for index, row in df.iterrows():
try:
row_instance= BiodataModel(
name=row.get('name'),
age=row.get('age'),
address =row.get('address'))
row_instance.full_clean()
bulk_data.append(row_instance)
except Exception as e:
return Response({"error": f'Error at row {index + 2} -> {e}'}, status=status.HTTP_400_BAD_REQUEST)
try:
with transaction.atomic():
BiodataModel.objects.bulk_create(bulk_data)
except Exception as e:
return Response({"error": f'Bulk create error--{e}'}, status=status.HTTP_400_BAD_REQUEST)
return Response({"msg":"CSV file processed successfully"}, status=status.HTTP_201_CREATED)
Explaining the code above:
The code begins with importing necessary packages, defining a class based view and setting a parser class (FileUploadParser
). The first part of the post method in the class attempts to get the file from request.FILES
and check its availability.
Then a minor validation checks that it is a CSV by checking the extension.
The next part loads it into a pandas dataframe (very much like a spreadsheet):
df = pd.read_csv(csv_file, delimiter=',',skiprows=3,dtype=str).iloc[:-1]
I will explain some of the arguments passed to the loading function:
skiprows
In reading the loaded csv file, it should be noted that the csv in this case is passed over a network, so some metadata like stuff is added to the beginning and end of the file. These things can be annoying and are not in comma separated value (csv) form so can actually raise errors in parsing. This explains why I used skiprows=3
, to skip the first 3 rows containing metadata and header and land directly on the body of the csv. If you remove skiprows
or use a lesser number, perhaps you might get an error like: Error tokenizing data. C error
or you might notice the data starting from the header.
dtype=str
Pandas likes to prove smart in trying to guess the datatype of certain columns. I wanted all values as string, so I used dtype=str
delimiter
Specifies how the cells are separated. Default is usually comma.
iloc[:-1]
I had to use iloc to slice the dataframe, removing the metadata at the end of the df
.
Then, the next line df = df.where(pd.notnull(df), None)
converts all NaN
values to None
. NaN
is a stand-in value that pandas uses to rep None
.
The next block is a bit tricky. We loop over every row in the dataframe, instantiate the row data with the BiodataModel
, perform model-level validation (not serializer-level) with full_clean()
method because bulk create bypasses Django validation. We then add our create operations to a list called bulk_data
. Yeah , add not run yet ! Remember, we are trying to do an atomic operation (at batch level ) so we want all or None. Saving rows individually wonβt give us all or none behaviour.
Then for the last significant part. Within a transaction.atomic()
block (which provides all or none behaviour), we run BiodataModel.objects.bulk_create(bulk_data)
to save all rows at once.
One more thing. Notice the index variable and the except block in the for loop. In the except block error message, I added 2 to the index
variable derived from df.iterrows()
because the value did not match exactly the row it was on, when looked at in an excel file. The except block catches any error and constructs an error message having the exact row number when opened in excel, so that the uploader can easily locate the line in the excel file!
Thanks for reading!!!
VERSIONS OF TOOLS USED
python==3.11.9
Django==4.2.10
pandas==2.2.3
djangorestframework==3.14.0
Also postman app for windows --> version 9.18.3
Top comments (0)