|
|
import hashlib
|
|
|
import os
|
|
|
from random import random
|
|
|
import time
|
|
|
from django.db import models
|
|
|
from boards import thumbs
|
|
|
from boards.models.base import Viewable
|
|
|
|
|
|
__author__ = 'neko259'
|
|
|
|
|
|
|
|
|
IMAGE_THUMB_SIZE = (200, 150)
|
|
|
IMAGES_DIRECTORY = 'images/'
|
|
|
FILE_EXTENSION_DELIMITER = '.'
|
|
|
HASH_LENGTH = 36
|
|
|
|
|
|
CSS_CLASS_IMAGE = 'image'
|
|
|
CSS_CLASS_THUMB = 'thumb'
|
|
|
|
|
|
|
|
|
class PostImage(models.Model, Viewable):
|
|
|
class Meta:
|
|
|
app_label = 'boards'
|
|
|
ordering = ('id',)
|
|
|
|
|
|
def _update_image_filename(self, filename):
|
|
|
"""
|
|
|
Gets unique image filename
|
|
|
"""
|
|
|
|
|
|
path = IMAGES_DIRECTORY
|
|
|
new_name = str(int(time.mktime(time.gmtime())))
|
|
|
new_name += str(int(random() * 1000))
|
|
|
new_name += FILE_EXTENSION_DELIMITER
|
|
|
new_name += filename.split(FILE_EXTENSION_DELIMITER)[-1:][0]
|
|
|
|
|
|
return os.path.join(path, new_name)
|
|
|
|
|
|
width = models.IntegerField(default=0)
|
|
|
height = models.IntegerField(default=0)
|
|
|
|
|
|
pre_width = models.IntegerField(default=0)
|
|
|
pre_height = models.IntegerField(default=0)
|
|
|
|
|
|
image = thumbs.ImageWithThumbsField(upload_to=_update_image_filename,
|
|
|
blank=True, sizes=(IMAGE_THUMB_SIZE,),
|
|
|
width_field='width',
|
|
|
height_field='height',
|
|
|
preview_width_field='pre_width',
|
|
|
preview_height_field='pre_height')
|
|
|
hash = models.CharField(max_length=HASH_LENGTH)
|
|
|
|
|
|
def save(self, *args, **kwargs):
|
|
|
"""
|
|
|
Saves the model and computes the image hash for deduplication purposes.
|
|
|
"""
|
|
|
|
|
|
if not self.pk and self.image:
|
|
|
md5 = hashlib.md5()
|
|
|
for chunk in self.image.chunks():
|
|
|
md5.update(chunk)
|
|
|
self.hash = md5.hexdigest()
|
|
|
super(PostImage, self).save(*args, **kwargs)
|
|
|
|
|
|
def __str__(self):
|
|
|
return self.image.url
|
|
|
|
|
|
def get_view(self):
|
|
|
return '<div class="{}">' \
|
|
|
'<a class="{}" href="{}">' \
|
|
|
'<img' \
|
|
|
' src="{}"' \
|
|
|
' alt="{}"' \
|
|
|
' width="{}"' \
|
|
|
' height="{}"' \
|
|
|
' data-width="{}"' \
|
|
|
' data-height="{}" />' \
|
|
|
'</a>' \
|
|
|
'</div>'\
|
|
|
.format(CSS_CLASS_IMAGE, CSS_CLASS_THUMB, self.image.url,
|
|
|
self.image.url_200x150,
|
|
|
str(self.hash), str(self.pre_width),
|
|
|
str(self.pre_height), str(self.width), str(self.height))
|
|
|
|