Проверка CSV файла при чтении по заголовкам python

Я хочу проверить CSV файл с заголовками: name, last_name. Идея заключается в том, что если в загруженном csv файле в первой строке есть имя и фамилия, то нужно продолжить, если нет - выдать ошибку.

def write():
    header = ['name', 'last_name']
    buffer = io.StringIO()
    writer = csv.writer(buffer)
    writer.writerow(header)
    # after writing data from table


def create(self, request, *args, **kwargs):
    serializer = CsvSerializer(data=request.data)
    if not serializer.is_valid():
        return Response(serializer.errors, status=status.HTTP_422_UNPROCESSABLE_ENTITY)
    csv_file = serializer.validated_data['file']
    file = csv_file.read().decode('utf-8')
    csv_fields = ['name', 'last_name']
    csv_reader = csv.DictReader(io.StringIO(file), fieldnames=csv_fields)
    next(csv_reader) # skip headers from csv for processing the file and do other stuff

Вернуться на верх