from django.db import models
from django.conf import settings
from PIL import Image

from apps.tenant.models import Tenant
from utils.file_operation import UploadFile, DeleteFile

class OperationHelp(models.Model):

    title = models.CharField(verbose_name=u'标题', max_length=100)
    content = models.CharField(verbose_name=u'内容',max_length=500)
    create_user = models.ForeignKey(settings.AUTH_USER_MODEL, verbose_name=u"创建人", on_delete=models.PROTECT,
                                    editable=False)
    create_time = models.DateTimeField(verbose_name='创建时间', auto_now_add=True)
    enable = models.BooleanField(verbose_name=u"在用", default=True)
    tenant = models.ForeignKey(Tenant, verbose_name=u'企业', editable=False, on_delete=models.PROTECT, null=True)
    image = models.CharField(verbose_name=u'图片路径', max_length=250,blank=True)
    width = models.IntegerField(verbose_name=u"图片宽度",blank=True, default=0)
    height = models.IntegerField(verbose_name=u"图片高度",blank=True, default=0)

    class Meta:
        db_table = 'operation_help'
        ordering = ['-id', 'tenant']
        verbose_name = u'运维帮助'

    def _add_img(self, file, user_id):
        width = None
        height = None

        path = "operation_help/"
        filename = UploadFile(file, path, user_id)
        fullname = "%s%s" % (settings.MEDIA_ROOT, filename)

        try:
            img = Image.open(fullname)
            width, height = img.size
            # 缩略图压缩宽或高最大200
            if width > 1440:
                img = img.resize((1440, int((height / width) * 1440)), Image.ANTIALIAS)
                img.save(fullname)
                width, height = img.size
        except:
            pass

        self.image = "%s%s" % (settings.MEDIA_URL, filename)
        self.width = width
        self.height = height
        self.save()

    def del_images(self):
        image = self.image
        # self.delete()
        DeleteFile(image)