signals.py
136 lines
| 5.0 KiB
| text/x-python
|
PythonLexer
/ boards / signals.py
|
r2008 | import os | ||
|
r1512 | import re | ||
|
r1784 | from django.db.models.signals import post_save, pre_save, pre_delete, \ | ||
post_delete | ||||
from django.dispatch import receiver | ||||
from django.utils import timezone | ||||
|
r1590 | from boards import thumbs | ||
|
r2008 | from boards.abstracts.constants import REGEX_REPLY, THUMB_SIZES | ||
|
r1512 | from boards.mdx_neboard import get_parser | ||
|
r2008 | from boards.models import Post, GlobalId, Attachment | ||
|
r1590 | from boards.models.attachment.viewers import FILE_TYPES_IMAGE | ||
|
r1997 | from boards.models.post import REGEX_NOTIFICATION, REGEX_GLOBAL_REPLY | ||
|
r1588 | from boards.models.post.manager import post_import_deps | ||
|
r1512 | from boards.models.user import Notification | ||
|
r1784 | from neboard.settings import MEDIA_ROOT | ||
|
r1512 | |||
@receiver(post_save, sender=Post) | ||||
def connect_replies(instance, **kwargs): | ||||
|
r1837 | if not kwargs['update_fields']: | ||
for reply_number in re.finditer(REGEX_REPLY, instance.get_raw_text()): | ||||
post_id = reply_number.group(1) | ||||
|
r1512 | |||
|
r1837 | try: | ||
referenced_post = Post.objects.get(id=post_id) | ||||
|
r1512 | |||
|
r1837 | if not referenced_post.referenced_posts.filter( | ||
id=instance.id).exists(): | ||||
referenced_post.referenced_posts.add(instance) | ||||
referenced_post.last_edit_time = instance.pub_time | ||||
referenced_post.build_refmap() | ||||
referenced_post.save(update_fields=['refmap', 'last_edit_time']) | ||||
except Post.DoesNotExist: | ||||
pass | ||||
|
r1512 | |||
@receiver(post_save, sender=Post) | ||||
|
r1588 | @receiver(post_import_deps, sender=Post) | ||
|
r1515 | def connect_global_replies(instance, **kwargs): | ||
|
r1837 | if not kwargs['update_fields']: | ||
for reply_number in re.finditer(REGEX_GLOBAL_REPLY, instance.get_raw_text()): | ||||
key_type = reply_number.group(1) | ||||
key = reply_number.group(2) | ||||
local_id = reply_number.group(3) | ||||
|
r1515 | |||
|
r1837 | try: | ||
global_id = GlobalId.objects.get(key_type=key_type, key=key, | ||||
local_id=local_id) | ||||
referenced_post = Post.objects.get(global_id=global_id) | ||||
referenced_post.referenced_posts.add(instance) | ||||
referenced_post.last_edit_time = instance.pub_time | ||||
referenced_post.build_refmap() | ||||
referenced_post.save(update_fields=['refmap', 'last_edit_time']) | ||||
except (GlobalId.DoesNotExist, Post.DoesNotExist): | ||||
pass | ||||
|
r1515 | |||
@receiver(post_save, sender=Post) | ||||
|
r1512 | def connect_notifications(instance, **kwargs): | ||
|
r1837 | if not kwargs['update_fields']: | ||
for reply_number in re.finditer(REGEX_NOTIFICATION, instance.get_raw_text()): | ||||
user_name = reply_number.group(1).lower() | ||||
Notification.objects.get_or_create(name=user_name, post=instance) | ||||
|
r1512 | |||
@receiver(pre_save, sender=Post) | ||||
|
r1588 | @receiver(post_import_deps, sender=Post) | ||
def parse_text(instance, **kwargs): | ||||
|
r1512 | instance._text_rendered = get_parser().parse(instance.get_raw_text()) | ||
@receiver(pre_delete, sender=Post) | ||||
def delete_attachments(instance, **kwargs): | ||||
for attachment in instance.attachments.all(): | ||||
attachment_refs_count = attachment.attachment_posts.count() | ||||
if attachment_refs_count == 1: | ||||
attachment.delete() | ||||
@receiver(post_delete, sender=Post) | ||||
def update_thread_on_delete(instance, **kwargs): | ||||
thread = instance.get_thread() | ||||
thread.last_edit_time = timezone.now() | ||||
thread.save() | ||||
|
r1520 | |||
@receiver(post_delete, sender=Post) | ||||
def delete_global_id(instance, **kwargs): | ||||
if instance.global_id and instance.global_id.id: | ||||
instance.global_id.delete() | ||||
|
r1590 | |||
@receiver(post_save, sender=Attachment) | ||||
def generate_thumb(instance, **kwargs): | ||||
if instance.mimetype in FILE_TYPES_IMAGE: | ||||
for size in THUMB_SIZES: | ||||
(w, h) = size | ||||
split = instance.file.name.rsplit('.', 1) | ||||
thumb_name = '%s.%sx%s.%s' % (split[0], w, h, split[1]) | ||||
if not instance.file.storage.exists(thumb_name): | ||||
# you can use another thumbnailing function if you like | ||||
thumb_content = thumbs.generate_thumb(instance.file, size, split[1]) | ||||
thumb_name_ = instance.file.storage.save(thumb_name, thumb_content) | ||||
if not thumb_name == thumb_name_: | ||||
raise ValueError( | ||||
'There is already a file named %s' % thumb_name_) | ||||
|
r1619 | |||
@receiver(pre_delete, sender=Post) | ||||
def rebuild_refmap(instance, **kwargs): | ||||
for referenced_post in instance.refposts.all(): | ||||
referenced_post.build_refmap(excluded_ids=[instance.id]) | ||||
referenced_post.save(update_fields=['refmap']) | ||||
|
r1784 | |||
@receiver(post_delete, sender=Attachment) | ||||
def delete_file(instance, **kwargs): | ||||
if instance.is_internal(): | ||||
file = MEDIA_ROOT + instance.file.name | ||||
|
r1879 | try: | ||
os.remove(file) | ||||
except FileNotFoundError: | ||||
pass | ||||
|
r1784 | if instance.mimetype in FILE_TYPES_IMAGE: | ||
for size in THUMB_SIZES: | ||||
file_name_parts = instance.file.name.split('.') | ||||
r1829 | thumb_file = MEDIA_ROOT + '{}.{}x{}.{}'.format(file_name_parts[0], size[0], size[1], file_name_parts[1]) | |||
|
r1879 | try: | ||
os.remove(thumb_file) | ||||
except FileNotFoundError: | ||||
pass | ||||
|
r1784 | |||