diff --git a/core/management/commands/addtestdata.py b/core/management/commands/addtestdata.py index ca88287..2d98449 100644 --- a/core/management/commands/addtestdata.py +++ b/core/management/commands/addtestdata.py @@ -60,8 +60,8 @@ class Command(BaseCommand): # create and assign products to companies for company in new_companies: - print("Creating fake products for {company.company_name}") - logging.info(f"Creating Products for {company.company_name}") + print(f"Creating fake products for {company.company_name}") + logging.info(f"Creating fake Products for {company.company_name}") # for i in range(100): for i in range(10): # make up data @@ -77,9 +77,11 @@ class Command(BaseCommand): response.raw.decode_content = True image = Image.open(response.raw) - # read image from memory + img_io = BytesIO() image.save(img_io, format='JPEG') + # option 1: read image from memory + ''' product.image = InMemoryUploadedFile( BytesIO(), field_name=None, @@ -88,10 +90,15 @@ class Command(BaseCommand): size=img_io.tell, charset=None ) + ''' + # option 2: File object + + product.image.save(f"{company.company_name}-{name}.jpg", File(img_io), save=False) product.save() logging.debug(f"New Product {product.name} created") print("*", end = '.') + # import ipdb; ipdb.set_trace() print('') print("Dataset creation finished") \ No newline at end of file