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
