def setUp(self) -> None: models.User.objects.create(phonenum='17763638641', username='******', password='******', sex='0', requestion='叫啥', answer='彭奥', usermodel='http://1') with open('media/user/default.jpg', 'rb') as fp: image = ImageFile(file=fp) image.open() models.Cloth.objects.create(id=1, phonenum='17763638641', classifycode='1', clothurl=image) self.url = '/wardrobe/getcloth'
class Coffee(models.Model): name = models.CharField(max_length=128) description = models.CharField(max_length=1024) one_pound_price = models.DecimalField(max_digits=5,decimal_places=2) two_pound_price = models.DecimalField(max_digits=5,decimal_places=2) five_pound_price = models.DecimalField(max_digits=5,decimal_places=2) image = models.ImageField(upload_to='images') active = models.BooleanField(default=True) class BadSizeException(Exception): pass def __unicode__(self): return self.name def __str__(self): return self.__unicode__() def value(self, pounds, quantity): if pounds == 1: return quantity * self.one_pound_price elif pounds == 2: return quantity * self.two_pound_price elif pounds == 5: return quantity * self.five_pound_price else: raise Coffee.BadSizeException def set_image(self, image_data): iof = BytesIO(image_data) self.image = ImageFile(iof) self.image.name = os.path.join('images', '%s.png' % self.name) def image_data_url(self): self.image.open() b64 = b64encode(self.image.read()).decode('utf-8') self.image.close() return 'data:image/png;base64,%s' % b64
def _get_image(self, response): """gets image and returns it""" image = ImageFile.open(BytesIO(response.content)) return image