Import/export files and images about django-import-export HOT 10 CLOSED

class MemberResource(resources.ModelResource):
    class Meta:
        model = Member

    def __init__(self):
        super(MemberResource, self).__init__()
        # Introduce a class variable to pass dry_run into methods that do not get it
        self.in_dry_run = False

    def before_import(self, dataset, using_transactions, dry_run, **kwargs):
        # Set helper class method to dry_run value
        self.in_dry_run = dry_run

    def before_import_row(self, row, row_number=None, **kwargs):
        if not self.in_dry_run:
            # Get URL and split image name from import file
            image_url = row['portrait']
            image_name = image_url.split('/')[-1]

            # Generate temporary file and download image from provided URL
            tmp_file = NamedTemporaryFile(delete=True, dir=f'{settings.MEDIA_ROOT}')
            tmp_file.write(urllib.request.urlopen(image_url).read())
            tmp_file.flush()

            # Add file object to row
            row['portrait'] = File(tmp_file, image_name)

class XLSXWithPicture(XLSX):
    def create_dataset(self, in_stream):
        """
        Create dataset from first sheet.
        """
        from io import BytesIO

        import openpyxl

        xlsx_book = openpyxl.load_workbook(BytesIO(in_stream))
        sheet = xlsx_book.active
        image_loader = SheetImageLoader(sheet)

        dataset = tablib.Dataset()
        # obtain generator
        rows = sheet.rows
        dataset.headers = [cell.value for cell in next(rows)]

        for row in rows:
            row_values = []
            for cell in row:
                if image_loader.image_in(cell.coordinate):
                    image = image_loader.get(cell.coordinate)
                    buffer = io.BytesIO()
                    image.save(buffer, format=image.format)
                    row_values.append(buffer.getvalue())
                else:
                    row_values.append(cell.value)
            dataset.append(row_values)
        return dataset

class SomeModelResource(resources.ModelResource):
    logo = Field(column_name="logo", attribute="logo")

    class Meta:
        model = SomeModel

    def __init__(self):
        super().__init__()
        # Introduce a class variable to pass dry_run into methods that do not get it
        self.in_dry_run = False

    def before_import(self, dataset, using_transactions, dry_run, **kwargs):
        # Set helper class method to dry_run value
        self.in_dry_run = dry_run

    def before_import_row(self, row, row_number=None, **kwargs):
        """Hook to ensure ContentFile is not created during validation run"""
        logo = row.get("logo")
        if logo:
            if self.in_dry_run:
                row["picture"] = f"Image {row_number}"
            else:
                row["picture"] = ContentFile(logo, str(uuid.uuid4()))
                
class SomeModelImport(ImportExportModelAdmin):
    resource_class = SomeModelResource
    formats = [XLSXWithPicture]
       

from urllib.parse import urlparse
import os
import requests
from django.core.files.base import ContentFile

# Import image from url with django import export application
def save_image_from_row(instance, row):
    if row['image']:
        if row['image'] == instance.image:
            image_content = ContentFile(requests.get(row['image']).content)
            url_image = urlparse(row['image'])
            instance.image.save(os.path.basename(url_image.path), image_content)
            instance.save()

--------------------------------------------------------------------

class CatLivreAuthorResource(resources.ModelResource):
    class Meta:
        model = CatLivreAuthor
        fields = ('id', 'first_name', 'last_name', 'copyright', 'image')

    def after_import_row(self, row, row_result, **kwargs):
            instance = CatLivreAuthor.objects.get(id=row['id'])
            save_image_from_row(instance, row)