# coding=utf8 from __future__ import unicode_literals, absolute_import, print_function from django.conf import settings from django.db import models from django.db.models import Q from django.utils import timezone from django.utils.html import escape from gm_types.gaia import DOCTOR_TYPE from gm_types.gaia import PROBLEM_FLAG_CHOICES from gm_types.gaia import TOPIC_TYPE from gm_types.gaia import USER_TYPE from gm_types.mimas import TRACTATE_SORT_TYPE from gm_upload import IMG_TYPE from gm_upload import ImgUrlField from talos.libs.datetime_utils import get_humanize_datetime from talos.libs.datetime_utils import get_timestamp from talos.libs.datetime_utils import get_timestamp_or_none from talos.models.diary.diary import Diary from talos.models.topic.topic import Problem from talos.rpc import CODES from talos.rpc import gen from talos.rpc import logging_exception from talos.services.doctor import DoctorService from talos.services.hospital import HospitalService from talos.services.user import UserService class TopicReply(models.Model): """ 回复的内容 """ class Meta: verbose_name = u'011.话题回复' verbose_name_plural = u'011. 话题回复' db_table = 'api_topicreply' app_label = 'talos' diary = models.ForeignKey(Diary, verbose_name=u"关联的日记本", null=True, blank=True, default=None) problem = models.ForeignKey(Problem, verbose_name=u"话题", null=True, blank=True, default=None) user_id = models.IntegerField(verbose_name=u'用户外键id') doctor_id = models.CharField(max_length=100, null=True, blank=True, verbose_name=u'医生外键id') hospital_id = models.CharField(max_length=100, null=True, blank=True, verbose_name=u'关联的医院外键id') replied_topic = models.ForeignKey("self", default=None, null=True, blank=True, verbose_name=u"被回复的信息") commented_reply = models.ForeignKey('self', related_name=u'comments', verbose_name=u'被评论的回复', null=True, blank=True, default=None) content = models.TextField(verbose_name=u"回复内容") image = ImgUrlField(img_type=IMG_TYPE.TOPICREPLY, max_length=100, default="", verbose_name=u"图片", null=True, blank=True) audio = models.TextField(default="", verbose_name=u"音频", null=True, blank=True) is_new = models.BooleanField(default=True, verbose_name=u"是否新的回复", db_index=True) is_spam = models.BooleanField(verbose_name=u'是否为疑似广告', default=False, db_index=True) like_num = models.IntegerField(default=0, verbose_name=u"赞的数目") dislike_num = models.IntegerField(default=0, verbose_name=u"踩的数目") # FIXME: reply_date重命名成reply_time，避免混淆含义 reply_date = models.DateTimeField(verbose_name=u"回复时间", db_index=True, default=timezone.now) is_topic_append = models.BooleanField(verbose_name=u'是否是更新', default=False, db_index=True) is_online = models.BooleanField(default=True, help_text=u"是否可以上线", verbose_name=u"上线") is_fake = models.BooleanField(default=False, verbose_name=u"是否是灌水数据") def is_voted(self, user, reply): # TODO CR from .topicreplyvote import TopicReplyVote """ NOTE: 对一级评论进行点赞 """ is_voted = TopicReplyVote.objects.filter(user_id=user.id, topic_reply=reply).exists() if user else False return is_voted def get_reply_info(self): data = { 'id': self.id, 'uid': self.user_id, 'nickname': self.user.nickname, 'content': escape(self.content), 'reply_time': get_timestamp_or_none(self.reply_date), 'images': [], 'is_online': self.is_online, 'can_reply': not self.replied_topic, } return data def get_reply_info_for_doctor(self): data = { 'topic_id': self.problem.id, 'reply_date': get_timestamp_or_none(self.reply_date), 'like_num': self.like_num, 'my_content': escape(self.content), "old_content": self.replied_topic.content if self.replied_topic_id else ( self.problem.ask or self.problem.answer) } return data def __unicode__(self): return self.content def reply_data_after_create(self): if self.user_id: portrait = self.user.portrait else: portrait = u'' user_type = USER_TYPE.NORMAL user_id = self.user_id if self.doctor_id: d = DoctorService.get_doctor_from_doctor_id(self.doctor_id) if d.doctor_type == DOCTOR_TYPE.OFFICER: user_type = USER_TYPE.OFFICER user_id = d.hospital_id elif d.doctor_type == DOCTOR_TYPE.DOCTOR: user_type = USER_TYPE.EXPERT user_id = self.doctor_id reply_data = { 'reply_id': self.id, 'user_type': user_type, 'user_id': user_id, 'user_nickname': escape(self.user.nickname) or u'昵称未设置', 'user_portrait': portrait, 'content': escape(self.content), 'reply_date': get_humanize_datetime(self.reply_date), 'reply_timestamp': get_timestamp(self.reply_date), 'comments': [], 'reply_count': self.comments.count(), 'vote_count': self.like_num, } return reply_data def comment_data_after_create(self): if self.replied_topic and self.replied_topic.user_id == self.user_id: at_nickname = u'' elif self.replied_topic: at_nickname = self.replied_topic.user.nickname else: at_nickname = u'' user_type = 0 if self.doctor_id: d = DoctorService.get_doctor_from_doctor_id(self.doctor_id) if d.doctor_type == DOCTOR_TYPE.OFFICER: user_type = 2 elif d.doctor_type == DOCTOR_TYPE.DOCTOR: user_type = 1 data = { 'comment_id': self.id, 'nickname': escape(self.user.nickname) or u'昵称未设置', 'reply_id': self.replied_topic and self.replied_topic.id or '', 'at_nickname': escape(at_nickname), 'at_user_id': self.replied_topic.user_id if self.replied_topic else 0, 'at_user_type': user_type, 'comment_user_id': self.user_id, 'comment_user_type': '', 'content': escape(self.content), 'reply_count': self.comments.count(), 'vote_count': self.like_num, } return data def is_elite(self): """ 判断是否是热门回复 """ return self.like_num >= settings.ELITE_LIKE_NUM and True or False def get_reply_detail(self, user_id=None): from .replyheadline import ReplyHeadline # 避免循环引用 if self.user_id: city_name = self.user.city_name portrait = self.user.portrait else: city_name = u'' portrait = u'' user_type = 0 doctor_info = UserService.get_doctor_from_user_id(self.user_id) if doctor_info is not None and doctor_info.is_online: doctor = doctor_info if doctor.doctor_type == DOCTOR_TYPE.OFFICER: user_type = USER_TYPE.OFFICER hospital_id = doctor_info.hospital_id else: user_type = USER_TYPE.EXPERT hospital_id = None else: doctor = None hospital_id = None hospital = HospitalService.get_hospital_from_doctor_id(self.doctor_id) if hospital is not None: hospital_name_full = hospital.name or u'' else: hospital_name_full = u'' hospital_name = hospital_name_full[:14] if len(hospital_name_full) <= 14 else hospital_name_full[:14] + '...' reply_data = { 'user': { 'user_id': self.user_id, 'user_name': self.user.nickname, 'is_doctor': True if doctor else False, 'is_hospital': True if hospital_id is not None else False, 'doctor_id': doctor.id if doctor else '', 'doctor_name': doctor and doctor.name, 'hospital_id': hospital_id if hospital_id is not None else '', 'portrait': portrait, 'user_type': user_type, 'membership_level': self.user.membership_level, }, 'is_private': bool(user_id) and user_id == self.user_id, 'doctor_portrait': DoctorService.get_doctor_from_doctor_id(self.doctor_id).portrait if self.doctor_id else '', 'hospital_name': hospital_name, 'is_recommend_doctor': False, 'recommend_doctor_title': '', 'favored': False, 'images': [], 'is_star_user': False, 'city': city_name, 'is_topic_append': self.is_topic_append, 'reply': { 'reply_id': self.id, 'content': escape(self.content), 'favor_amount': self.like_num, 'reply_time': get_timestamp_or_none(self.reply_date), 'is_voted': bool(user_id) and self.topicreplyvote_set.filter(user_id=user_id, topic_reply=self).exists(), 'image': self.image, 'is_new': self.is_new, 'replied_topic_id': self.replied_topic and self.replied_topic.id or '', 'reply_date': get_timestamp_or_none(self.reply_date), 'commented_reply_id': ( self.replied_topic and self.commented_reply and self.commented_reply.id or '' ), 'like_num': self.like_num, 'is_best_reply': self.is_best_reply(), 'is_recommand': ReplyHeadline.is_recommend(self.id), # TODO CR 修改这个函数 'is_elite': self.is_elite(), 'comments_num': self.comments.count(), } } try: reply_data['diary'], reply_data['topic'], reply_data['answer'] = None, None, None if self.problem: topic_info = self.get_reply_topic_info() reply_data['topic'] = topic_info if self.diary: diary_info = self.get_reply_diary_info() reply_data['diary'] = diary_info author_id = self.problem and self.problem.user_id or self.diary.user_id if self.user_id == author_id: reply_data['user_titles'] = [{'title_type': 0, 'title_text': u'楼主'}, ] else: reply_data['user_titles'] = [] if self.is_topic_append: # 增加术后第几天的展示 if self.problem.operation_time: interval = self.reply_date.date() - self.problem.operation_time if interval.days >= 0: reply_data['interval'] = u'{}'.format(interval.days + 1) else: reply_data['interval'] = u'以前' else: reply_data['interval'] = u'' except: logging_exception() return reply_data def get_reply_topic_info(self): """ NOTE: 获取回复相关的帖子topic信息 :return: """ topic_info = { 'id': self.problem_id, 'title': ( self.replied_topic and self.replied_topic.id and escape(self.replied_topic.content) or escape(self.problem.answer) ), 'problem_ask': escape(self.problem.ask), 'problem_answer': escape(self.problem.answer), 'is_deleted': ( self.problem.flag == PROBLEM_FLAG_CHOICES.MARK_DELETED or not self.problem.is_online ), 'user': { 'id': self.problem.user_id, 'nickname': self.replied_topic and self.replied_topic.user.nickname or self.user.nickname }, } if self.problem.topic_type == TOPIC_TYPE.WEIXIN_NUM: topic_info['title'] = self.problem.get_title() return topic_info def get_reply_diary_info(self): """ NOTE: 获取回复相关的日记本信息 其中 显示的title做特殊处理 title = 'tag'+'日记' :return: """ title = escape(self.diary.title) tags_new_era = self.diary.tags_new_era if tags_new_era: title = tags_new_era[0]['name'] + u'日记' diary_info = { 'id': self.diary.id, 'title': title, 'is_deleted': not self.diary.is_online, 'user': { 'id': self.diary.user_id, 'nickname': UserService.get_user_by_user_id(self.diary.user_id).nickname, }, } return diary_info def comment_data(self): if self.replied_topic: if self.replied_topic.user_id == self.user_id: at_nickname = u'' at_doctor = None at_hospital_id = None else: if self.replied_topic.user_id is not None: replied_doctor_info = UserService.get_doctor_from_user_id(self.replied_topic.user_id) at_doctor = replied_doctor_info if at_doctor and at_doctor.is_online and at_doctor.doctor_type == DOCTOR_TYPE.OFFICER: at_hospital_id = at_doctor.hospital_id else: at_hospital_id = None else: at_doctor = None at_hospital_id = None if self.replied_topic.doctor_id: u = UserService.get_user_id_by_doctor_id(self.replied_topic.doctor_id) at_nickname = u and UserService.get_user_by_user_id(u).nickname or u'昵称未设置' else: at_nickname = self.replied_topic.user.nickname or u'昵称未设置' else: at_nickname = u'昵称未设置' at_doctor = None at_hospital_id = None if self.user_id: portrait = self.user.portrait else: portrait = u'' doctor_info = UserService.get_doctor_from_user_id(self.user_id) if doctor_info is not None and doctor_info.is_online: doctor = doctor_info if doctor.doctor_type == DOCTOR_TYPE.OFFICER: hospital_id = doctor.hospital_id else: hospital_id = None else: doctor = None hospital_id = None data = { 'type': 0, 'content': escape(self.content), 'at_nickname': at_nickname, 'at_doctor_id': at_doctor.id if at_doctor else '', 'at_hospital_id': at_hospital_id if at_hospital_id is not None else '', 'user_portrait': portrait, 'at_user_id': self.replied_topic.user_id if self.replied_topic else '', 'nickname': '', 'comment_id': self.id, 'comment_time': self.reply_date.strftime('%m.%d'), 'comment_expert_id': doctor.id if doctor else '', 'at_expert_id': at_doctor.id if at_doctor else '', 'user_id': self.user_id, 'user_is_doctor': True if doctor else False, 'user_is_hospital': True if hospital_id is not None else False, 'doctor_id': doctor.id if doctor else '', 'hospital_id': hospital_id if hospital_id is not None else '', 'membership_level': self.user.membership_level, 'is_best_reply': self.is_best_reply(), } if self.doctor_id is not None: d = DoctorService.get_doctor_from_doctor_id(self.doctor_id) data['nickname'] = d.name if d.user != settings.SUOZHANG_UID: data['nickname'] += u' 医生' else: data['nickname'] = self.user.nickname or u'昵称未设置' return data @classmethod def get_by_id(cls, id): try: return TopicReply.objects.get(pk=id) except: return None def is_best_reply(self): """ 是否是优质问答 """ return self.replyheadline_set.filter(is_online=True).exists() def update_reply_num(self, is_online=False): real_replynum = 1 + TopicReply.objects.filter(commented_reply=self).count() if self.problem: if is_online: self.problem.reply_num += real_replynum self.problem.update_reply_num(is_online=True, reply_num=real_replynum) else: self.problem.reply_num -= real_replynum self.problem.update_reply_num(is_online=False, reply_num=real_replynum) if self.problem.reply_num < 0: return False self.problem.save() elif self.diary: if is_online: self.diary.reply_num += real_replynum else: self.diary.reply_num -= real_replynum if self.diary.reply_num < 0: return False self.diary.save() return True @classmethod def get_replies_info(cls, user, replies, start_num=0, count=10, comment_id=None, new_sub_comment_count=False, is_xcx=False): from talos.manager import comment_data_by_reply_ids, get_reply_details_by_reply_ids result = [] need_add_reply_id = None if comment_id: top_comment = TopicReply.objects.filter(id=comment_id, commented_reply_id__isnull=True).first() replies = list(replies) if not top_comment: reply = TopicReply.objects.get(id=comment_id) replies = [item for item in replies if item.id != reply.commented_reply.id] replies = [reply.commented_reply] + replies if is_xcx and start_num == 0: need_add_reply_id = reply.commented_reply_id else: replies = [item for item in replies if item.id != top_comment.id] replies = [top_comment] + replies _replies = replies[start_num: start_num + count] reply_ids = [r.id for r in _replies] comments_ids = list(TopicReply.objects.filter(Q(commented_reply__id__in=reply_ids) & Q(is_online=True)) .values_list('id', flat=True)) comments_data = comment_data_by_reply_ids(comments_ids) topicreplys_data = get_reply_details_by_reply_ids(reply_ids, user.id) for reply in _replies: replies_data = None for _reply in topicreplys_data: if _reply['reply']['reply_id'] == reply.id: replies_data = _reply break if replies_data is None: continue reply_info = replies_data['reply'] if reply.replyheadline_set.count() == 0: replies_data['reply']['is_recommand'] = False else: replies_data['reply']['is_recommand'] = True replies_data['reply']['is_elite'] = (1 if reply_info['favor_amount'] >= settings.ELITE_LIKE_NUM else 0) replies_data['comments'] = [] if new_sub_comment_count: _reply_comments_ids = list(reply.comments.filter( is_online=True ).order_by('-id').values_list('id', flat=True))[:2] # 取最新两条 else: _reply_comments_ids = list(reply.comments.filter( is_online=True ).values_list('id', flat=True)) second_top_comment = None for c in comments_data: if c['comment_id'] in _reply_comments_ids: replies_data['comments'].append(c) if comment_id and not second_top_comment and c['comment_id'] == int(comment_id): second_top_comment = c # 将二级评论 置顶 if need_add_reply_id and need_add_reply_id == reply.id and second_top_comment: replies_data['comments'] = [item for item in replies_data['comments'] if item['comment_id'] != int(comment_id)] replies_data['comments'].insert(0, second_top_comment) result.append(replies_data) return result @classmethod def get_replies_list_info( cls, user, query, start_num=0, count=10, comment_id=None, new_sub_comment_count=False ): """ NOTE: 首先获取热门评论数目 如果不够，则继续获取最新评论 获取 日记本 帖子 评论列表信息 :param user: :param start_num: :param count: :param new_sub_comment_count: 子评论展示个数控制， 7130 以及评论下展示两个子评论 :return: """ recommand_reply = Q(like_num__gte=settings.ELITE_LIKE_NUM) recommand_reply &= query replies = TopicReply.objects.filter(recommand_reply) hot_count = replies.count() if comment_id: myreplies = TopicReply.objects.filter(id=comment_id) replies = myreplies | replies result = [] if hot_count > start_num: replies = replies.order_by('-like_num', '-reply_date') result = cls.get_replies_info( user, replies, start_num=start_num, count=count, comment_id=comment_id, new_sub_comment_count=new_sub_comment_count ) if len(result) < count: if start_num < hot_count: start_num = 0 else: start_num = start_num - len(result) - hot_count count -= len(result) last_query = Q(like_num__lt=settings.ELITE_LIKE_NUM) last_query &= query last_replies = cls.objects.filter(last_query).order_by('-id').exclude(recommand_reply) last_result = cls.get_replies_info( user, last_replies, start_num=start_num, count=count, comment_id=comment_id, new_sub_comment_count=new_sub_comment_count, ) result.extend(last_result) return result @classmethod def reply_detail_diary(cls, user, reply_id, start_num=0, count=10): from talos.manager.topic_reply import comment_data_by_reply_ids try: reply = TopicReply.objects.get(pk=reply_id) except TopicReply.DoesNotExist: return gen(CODES.TOPIC_NOT_FOUND) topic = reply.problem result = { 'comments': [], 'topic_id': topic.id, 'title': topic.diary.title, 'comment_count': topic.topicreply_set.count(), 'reply_id': reply.id, 'is_private': False if not user else user.id == reply.user_id, 'doctor_id': '', 'doctor_portrait': '', 'doctor_name': '', 'hospital_name': '', 'is_recommend_doctor': False, 'recommend_doctor_title': '', 'favored': False, 'favor_amount': topic.vote_amount, 'content': topic.answer, 'images': [i.get_image_data() for i in topic.images.all()], 'user_id': reply.user_id, 'user_nickname': reply.user.nickname, 'user_portrait': reply.user.portrait, 'is_star_user': False, 'reply_date': get_timestamp_or_none(topic.created_time), 'city': reply.user.city_name, 'is_topic_append': True, } doctor_reply_sql = """ (case when api_topicreply.doctor_id !='' then 1 else 0 end) """ comments = topic.topicreply_set.extra( select={'doctor_reply': doctor_reply_sql}) comments = comments.extra(order_by=['-doctor_reply', 'id']) comments = comments[start_num: start_num + count] _comments_ids = [c.id for c in comments] result['comments'] = comment_data_by_reply_ids(_comments_ids) return result @property def user(self): if hasattr(self, '_user_obj'): return self._user_obj self._user_obj = UserService.get_user_by_user_id(self.user_id) return self._user_obj @property def doctor(self): return DoctorService.get_doctor_from_doctor_id(self.doctor_id) @property def hospital(self): return HospitalService.get_hospital_from_hospital_id(self.hospital_id) def get_reply_for_mine(self): return { 'reply': { 'replied_topic_id': self.replied_topic_id or '', 'content': escape(self.content), 'reply_id': self.id, 'commented_reply_id': self.replied_topic_id and self.commented_reply_id or '', 'reply_date': get_timestamp_or_none(self.reply_date), 'image': self.image, 'is_new': self.is_new, 'commented_content': self.replied_topic.content if self.replied_topic else None, 'commented_user_id': self.replied_topic.user_id if self.replied_topic else None, }, 'user': { 'doctor_name': '', 'user_name': '', 'doctor_id': '', 'user_id': self.user_id, 'user_portrait': '', 'doctor_portrait': '', }, 'topic': { 'user': { 'nickname': '', 'id': self.problem.user_id, 'portrait': '', }, 'id': self.problem_id, 'problem_ask': escape(self.problem.ask), 'title': self.replied_topic_id and escape(self.replied_topic.content) or escape(self.problem.answer), 'is_deleted': self.problem.flag == PROBLEM_FLAG_CHOICES.MARK_DELETED or not self.problem.is_online, } if self.problem_id else None, 'diary': { 'user': { 'nickname': '', 'id': self.diary.user_id, 'portrait': '', }, 'id': self.diary_id, 'title': escape(self.diary.title), 'is_deleted': not self.diary.is_online, } if self.diary_id else None, 'answer': None, }