manager.py
190 lines
| 5.9 KiB
| text/x-python
|
PythonLexer
|
r1359 | import logging | ||
from datetime import datetime, timedelta, date | ||||
from datetime import time as dtime | ||||
|
r1602 | from boards.abstracts.exceptions import BannedException, ArchiveException | ||
|
r1359 | from django.db import models, transaction | ||
from django.utils import timezone | ||||
|
r1588 | from django.dispatch import Signal | ||
|
r1359 | |||
import boards | ||||
from boards.models.user import Ban | ||||
from boards.mdx_neboard import Parser | ||||
|
r1590 | from boards.models import Attachment | ||
|
r1359 | from boards import utils | ||
__author__ = 'neko259' | ||||
POSTS_PER_DAY_RANGE = 7 | ||||
NO_IP = '0.0.0.0' | ||||
|
r1588 | post_import_deps = Signal() | ||
|
r1359 | class PostManager(models.Manager): | ||
@transaction.atomic | ||||
def create_post(self, title: str, text: str, file=None, thread=None, | ||||
|
r1704 | ip=NO_IP, tags: list=None, | ||
|
r1660 | tripcode='', monochrome=False, images=[], | ||
file_url=None): | ||||
|
r1359 | """ | ||
Creates new post | ||||
""" | ||||
|
r1449 | if thread is not None and thread.is_archived(): | ||
|
r1602 | raise ArchiveException('Cannot post into an archived thread') | ||
|
r1449 | |||
|
r1362 | if not utils.is_anonymous_mode(): | ||
is_banned = Ban.objects.filter(ip=ip).exists() | ||||
|
r1434 | else: | ||
is_banned = False | ||||
|
r1359 | |||
if is_banned: | ||||
|
r1602 | raise BannedException("This user is banned") | ||
|
r1359 | |||
if not tags: | ||||
tags = [] | ||||
posting_time = timezone.now() | ||||
new_thread = False | ||||
if not thread: | ||||
thread = boards.models.thread.Thread.objects.create( | ||||
|
r1434 | bump_time=posting_time, last_edit_time=posting_time, | ||
|
r1704 | monochrome=monochrome) | ||
|
r1359 | list(map(thread.tags.add, tags)) | ||
new_thread = True | ||||
pre_text = Parser().preparse(text) | ||||
post = self.create(title=title, | ||||
text=pre_text, | ||||
pub_time=posting_time, | ||||
poster_ip=ip, | ||||
thread=thread, | ||||
last_edit_time=posting_time, | ||||
tripcode=tripcode, | ||||
opening=new_thread) | ||||
logger = logging.getLogger('boards.post.create') | ||||
|
r1380 | logger.info('Created post [{}] with text [{}] by {}'.format(post, | ||
|
r1368 | post.get_text(),post.poster_ip)) | ||
|
r1359 | |||
if file: | ||||
|
r1511 | self._add_file_to_post(file, post) | ||
|
r1500 | for image in images: | ||
|
r1604 | post.attachments.add(image) | ||
|
r1660 | if file_url: | ||
post.attachments.add(Attachment.objects.create_from_url(file_url)) | ||||
|
r1359 | |||
|
r1320 | post.set_global_id() | ||
|
r1359 | |||
# Thread needs to be bumped only when the post is already created | ||||
if not new_thread: | ||||
thread.last_edit_time = posting_time | ||||
thread.bump() | ||||
thread.save() | ||||
return post | ||||
def delete_posts_by_ip(self, ip): | ||||
""" | ||||
Deletes all posts of the author with same IP | ||||
""" | ||||
posts = self.filter(poster_ip=ip) | ||||
for post in posts: | ||||
post.delete() | ||||
@utils.cached_result() | ||||
def get_posts_per_day(self) -> float: | ||||
""" | ||||
Gets average count of posts per day for the last 7 days | ||||
""" | ||||
day_end = date.today() | ||||
day_start = day_end - timedelta(POSTS_PER_DAY_RANGE) | ||||
day_time_start = timezone.make_aware(datetime.combine( | ||||
day_start, dtime()), timezone.get_current_timezone()) | ||||
day_time_end = timezone.make_aware(datetime.combine( | ||||
day_end, dtime()), timezone.get_current_timezone()) | ||||
posts_per_period = float(self.filter( | ||||
pub_time__lte=day_time_end, | ||||
pub_time__gte=day_time_start).count()) | ||||
ppd = posts_per_period / POSTS_PER_DAY_RANGE | ||||
return ppd | ||||
|
r1618 | def get_post_per_days(self, days) -> int: | ||
day_end = date.today() + timedelta(1) | ||||
day_start = day_end - timedelta(days) | ||||
day_time_start = timezone.make_aware(datetime.combine( | ||||
day_start, dtime()), timezone.get_current_timezone()) | ||||
day_time_end = timezone.make_aware(datetime.combine( | ||||
day_end, dtime()), timezone.get_current_timezone()) | ||||
return self.filter( | ||||
pub_time__lte=day_time_end, | ||||
pub_time__gte=day_time_start).count() | ||||
|
r1320 | @transaction.atomic | ||
def import_post(self, title: str, text: str, pub_time: str, global_id, | ||||
|
r1557 | opening_post=None, tags=list(), files=list(), | ||
|
r1586 | tripcode=None, version=1): | ||
|
r1360 | is_opening = opening_post is None | ||
if is_opening: | ||||
|
r1320 | thread = boards.models.thread.Thread.objects.create( | ||
bump_time=pub_time, last_edit_time=pub_time) | ||||
list(map(thread.tags.add, tags)) | ||||
else: | ||||
thread = opening_post.get_thread() | ||||
|
r1586 | post = self.create(title=title, | ||
text=text, | ||||
|
r1320 | pub_time=pub_time, | ||
poster_ip=NO_IP, | ||||
last_edit_time=pub_time, | ||||
|
r1360 | global_id=global_id, | ||
|
r1386 | opening=is_opening, | ||
|
r1586 | thread=thread, | ||
tripcode=tripcode, | ||||
version=version) | ||||
|
r1320 | |||
|
r1511 | for file in files: | ||
self._add_file_to_post(file, post) | ||||
|
r1588 | url_to_post = '[post]{}[/post]'.format(str(global_id)) | ||
replies = self.filter(text__contains=url_to_post) | ||||
for reply in replies: | ||||
|
r1613 | post_import_deps.send(reply) | ||
|
r1588 | |||
|
r1586 | @transaction.atomic | ||
def update_post(self, post, title: str, text: str, pub_time: str, | ||||
tags=list(), files=list(), tripcode=None, version=1): | ||||
post.title = title | ||||
post.text = text | ||||
post.pub_time = pub_time | ||||
post.tripcode = tripcode | ||||
post.version = version | ||||
post.save() | ||||
post.clear_cache() | ||||
post.attachments.clear() | ||||
for file in files: | ||||
self._add_file_to_post(file, post) | ||||
thread = post.get_thread() | ||||
thread.tags.clear() | ||||
list(map(thread.tags.add, tags)) | ||||
|
r1511 | def _add_file_to_post(self, file, post): | ||
|
r1590 | post.attachments.add(Attachment.objects.create_with_hash(file)) | ||