# coding=utf-8 from __future__ import unicode_literals import datetime import re from django.db import models from management.models import Comment from lms.tools import out_date_format, translit from lms.settings import DOMAIN from access.models import User from courses.templates import comment_fabric class ArticleSection(models.Model): name = models.CharField(verbose_name=u'Раздел', max_length=255) def __str__(self): return '%s' % self.name def __unicode__(self): return u'%s' % self.name class Meta: verbose_name = u'Раздел статьи' verbose_name_plural = u'Разделы статей' class Article(models.Model): public = models.BooleanField(verbose_name=u'Опубликовать', default=True) slug = models.SlugField(max_length=255, editable=False, blank=True, default='', unique=True) head_title = models.CharField(verbose_name=u'Заголовок для соц сетей', max_length=255, blank=True, null=True) head_description = models.CharField(verbose_name=u'Описание в заголовке для соц сетей', max_length=255, blank=True, null=True) head_image = models.ImageField(verbose_name=u'Картинка для соц сетей', upload_to='library', blank=True, null=True) title = models.CharField(verbose_name=u'Заголовок', max_length=255) section = models.ForeignKey(ArticleSection, verbose_name=u'Раздел') preview = models.TextField(verbose_name=u'Превьюшка', blank=True, default='') preview_img = models.ImageField(verbose_name=u'Превьюшка заливки', upload_to='library', blank=True, null=True) page = models.TextField(verbose_name=u'PAGE') css = models.TextField(verbose_name=u'CSS', blank=True, default='', help_text=u'Можно добавлять все, но только исправить http на https') js = models.TextField(verbose_name=u'JS', blank=True, default='', help_text=u'Удалить bootstrap, jquery и заменить http на https') date = models.DateTimeField(verbose_name=u'Дата публикации', default=datetime.datetime.now, editable=False) tags = models.ManyToManyField('Tags', verbose_name=u'Теги', blank=True, null=True) likes = models.ManyToManyField(User, verbose_name=u'Лайки', max_length=255, null=True, blank=True, editable=False, related_name=u'acticle_likes') views = models.ManyToManyField(User, verbose_name=u'Просмотры', max_length=255, default=0, related_name='article_views', editable=False) all_views = models.IntegerField(verbose_name=u'Всего просмотров', default=0, blank=True) comments = models.ManyToManyField(Comment, verbose_name=u'Комментарии', null=True, blank=True, editable=False) favorite = models.ManyToManyField(User, verbose_name=u'В фаворитах', blank=True, null=True, related_name='article_as_favorites', editable=False) def __str__(self): return '%s' % self.title def __unicode__(self): return u'%s' % self.title def get_date(self): return out_date_format(self.date, no_time=True) def set_favorite(self, user): if user not in self.favorite.all(): self.favorite.add(user) def unset_favorite(self, user): if user in self.favorite.all(): self.favorite.remove(user) def get_face(self, user): return { 'id': self.id, 'title': self.title, 'slug': self.slug, 'social': {'title': self.head_title if self.head_title else self.title, 'description': self.head_description if self.head_description else self.preview, 'image': self.head_image.url if self.head_image else DOMAIN + '/static/img/3818cc0ec1.png'}, 'preview': {'title': self.title, 'description': self.preview, 'image': self.preview_img.url if self.preview_img else DOMAIN + '/static/img/3818cc0ec1.png'}, 'page': self.page, 'css': self.css, 'js': self.js, 'date': self.get_date(), 'tags': {'count': self.get_tags_length(), 'content': self.tags.all()}, 'likes': {'count': self.get_likes_length(), 'content': self.likes.all()}, 'views': {'count': self.get_views_length(), 'content': self.views.all()}, 'favorites': {'count': self.get_favorite_length(), 'content': self.favorite.all()}, 'comments': {'count': self.get_comments_length(), 'content': self.comments.all()}, 'section': {'title': self.section.name, 'id': self.section.id}, 'like': self.check_like_exists(user) if user.is_authenticated() else False, 'view': self.check_view_exists(user) if user.is_authenticated() else False, 'favorite': self.check_favorit_exists(user) if user.is_authenticated() else False } def get_comments(self, _user=None): # Получить все комментарии урока return [comment_fabric(comment, __type='L', __user=_user) for comment in self.comments.filter(parent_id=0)] def get_all_comments(self, _user=None): return [comment_fabric(comment, __type='L', __user=_user) for comment in self.comments.all().order_by('-date')] def check_favorit_exists(self, user): return user in self.favorite.all() def check_view_exists(self, user): return user in self.views.all() def check_like_exists(self, user): return user in self.likes.all() def like_plus(self, user): result = False if user not in self.likes.all(): self.likes.add(user) result = True return result def like_minus(self, user): result = False if user in self.likes.all(): self.likes.remove(user) result = True return result def get_likes_length(self): return self.likes.count() def get_views_length(self): return self.all_views def views_plus(self, user): self.all_views += 1 if user.is_authenticated(): if not self.views.filter(id=user.id).exists(): self.views.add(user) self.save() def get_comments_length(self): return self.comments.filter(closed=False).count() def get_favorite_length(self): return self.favorite.count() def get_tags_length(self): return self.tags.count() def get_absolute_url(self): return '/library/article/%s' % self.slug def save(self, *args, **kwargs): self.slug = '{0}'.format(translit(self.title, only_letter=True)) super(Article, self).save(*args, **kwargs) class Meta: verbose_name = u'Статья' verbose_name_plural = u'Статьи' ordering = ['-date'] class Tags(models.Model): name = models.CharField(verbose_name=u'Теги', max_length=255) def __str__(self): return '%s' % self.name def __unicode__(self): return 'u%s' % self.name class Meta: verbose_name = u'Тег' verbose_name_plural = u'Теги'