file-code code-view code-diff">
@@ -0,0 +1,68 @@ |
||
| 1 |
+# -*- coding: utf-8 -*- |
|
| 2 |
+ |
|
| 3 |
+from django.conf import settings |
|
| 4 |
+from django.db import models |
|
| 5 |
+from django.utils.translation import ugettext_lazy as _ |
|
| 6 |
+ |
|
| 7 |
+from pai2.basemodels import CreateUpdateMixin |
|
| 8 |
+ |
|
| 9 |
+ |
|
| 10 |
+class UserMessageInfo(CreateUpdateMixin): |
|
| 11 |
+ SYSTEM = 'system' |
|
| 12 |
+ COMMENT = 'comment' |
|
| 13 |
+ THUMBUP = 'thumbup' |
|
| 14 |
+ |
|
| 15 |
+ MESSAGE_TYPE = ( |
|
| 16 |
+ (SYSTEM, u'系统'), |
|
| 17 |
+ (COMMENT, u'评论'), |
|
| 18 |
+ (THUMBUP, u'点赞'), |
|
| 19 |
+ ) |
|
| 20 |
+ |
|
| 21 |
+ MESSAGE_TYPE_INFO = [ |
|
| 22 |
+ {
|
|
| 23 |
+ 'msg_type': SYSTEM, |
|
| 24 |
+ 'msg_avatar': settings.SYSTEM_MESSAGE_AVATAR |
|
| 25 |
+ }, {
|
|
| 26 |
+ 'msg_type': COMMENT, |
|
| 27 |
+ 'msg_avatar': settings.COMMENT_MESSAGE_AVATAR |
|
| 28 |
+ }, {
|
|
| 29 |
+ 'msg_type': THUMBUP, |
|
| 30 |
+ 'msg_avatar': settings.THUMBUP_MESSAGE_AVATAR |
|
| 31 |
+ } |
|
| 32 |
+ ] |
|
| 33 |
+ |
|
| 34 |
+ from_uid = models.CharField(_(u'from_uid'), max_length=255, blank=True, null=True, help_text=u'发送消息用户唯一标识', db_index=True) |
|
| 35 |
+ from_nickname = models.CharField(_(u'from_nickname'), max_length=255, blank=True, null=True, help_text=u'发送消息用户昵称') |
|
| 36 |
+ from_avatar = models.CharField(_(u'from_avatar'), max_length=255, blank=True, null=True, help_text=u'发送消息用户头像') |
|
| 37 |
+ |
|
| 38 |
+ to_uid = models.CharField(_(u'to_uid'), max_length=255, blank=True, null=True, help_text=u'接收消息用户唯一标识', db_index=True) |
|
| 39 |
+ |
|
| 40 |
+ group_id = models.CharField(_(u'group_id'), max_length=255, blank=True, null=True, help_text=u'群组唯一标识') |
|
| 41 |
+ photo_id = models.CharField(_(u'photo_id'), max_length=255, blank=True, null=True, help_text=u'飞图唯一标识') |
|
| 42 |
+ |
|
| 43 |
+ msg_type = models.CharField(_(u'msg_type'), max_length=255, default='system', help_text=u'消息类型', db_index=True) |
|
| 44 |
+ msg_title = models.CharField(_(u'msg_title'), max_length=255, blank=True, null=True, help_text=u'消息标题') |
|
| 45 |
+ msg_content = models.TextField(_(u'msg_content'), blank=True, null=True, help_text=u'消息内容') |
|
| 46 |
+ read = models.BooleanField(_(u'read'), default=False, help_text=u'消息是否已读') |
|
| 47 |
+ |
|
| 48 |
+ class Meta: |
|
| 49 |
+ verbose_name = _('usermessageinfo')
|
|
| 50 |
+ verbose_name_plural = _('usermessageinfo')
|
|
| 51 |
+ |
|
| 52 |
+ def __unicode__(self): |
|
| 53 |
+ return u'{0.title}'.format(self)
|
|
| 54 |
+ |
|
| 55 |
+ @property |
|
| 56 |
+ def msg_info(self): |
|
| 57 |
+ return {
|
|
| 58 |
+ 'pk': self.pk, |
|
| 59 |
+ 'from_uid': self.from_uid, |
|
| 60 |
+ 'from_nickname': self.from_nickname, |
|
| 61 |
+ 'from_avatar': self.from_avatar, |
|
| 62 |
+ 'group_id': self.group_id, |
|
| 63 |
+ 'photo_id': self.photo_id, |
|
| 64 |
+ 'msg_title': self.msg_title, |
|
| 65 |
+ 'msg_content': self.msg_content, |
|
| 66 |
+ 'read': self.read, |
|
| 67 |
+ 'created_at': self.created_at, |
|
| 68 |
+ } |
@@ -0,0 +1,3 @@ |
||
| 1 |
+from django.test import TestCase |
|
| 2 |
+ |
|
| 3 |
+# Create your tests here. |
@@ -0,0 +1,73 @@ |
||
| 1 |
+# -*- coding: utf-8 -*- |
|
| 2 |
+ |
|
| 3 |
+from django.conf import settings |
|
| 4 |
+from django.http import JsonResponse |
|
| 5 |
+ |
|
| 6 |
+from message.models import UserMessageInfo |
|
| 7 |
+ |
|
| 8 |
+from utils.page_utils import pagination |
|
| 9 |
+ |
|
| 10 |
+ |
|
| 11 |
+def message_list_api(request): |
|
| 12 |
+ messages = UserMessageInfo.MESSAGE_TYPE_INFO |
|
| 13 |
+ |
|
| 14 |
+ final_messages = [] |
|
| 15 |
+ for message in messages: |
|
| 16 |
+ type_messages = UserMessageInfo.objects.filter( |
|
| 17 |
+ msg_type=message['msg_type'] |
|
| 18 |
+ ).order_by( |
|
| 19 |
+ '-updated_at' |
|
| 20 |
+ )[:settings.MESSAGE_NUM_PER_PAGE] |
|
| 21 |
+ type_messages = [msg.msg_info for msg in type_messages] |
|
| 22 |
+ message['msg_list'] = type_messages |
|
| 23 |
+ final_messages.append(message) |
|
| 24 |
+ |
|
| 25 |
+ return JsonResponse({
|
|
| 26 |
+ 'status': 200, |
|
| 27 |
+ 'message': u'获取消息列表成功', |
|
| 28 |
+ 'data': {
|
|
| 29 |
+ 'messages': final_messages, |
|
| 30 |
+ }, |
|
| 31 |
+ }) |
|
| 32 |
+ |
|
| 33 |
+ |
|
| 34 |
+def message_type_list_api(request, msg_type): |
|
| 35 |
+ page = int(request.GET.get('page', 1))
|
|
| 36 |
+ num = int(request.GET.get('num', settings.MESSAGE_NUM_PER_PAGE))
|
|
| 37 |
+ |
|
| 38 |
+ type_messages = UserMessageInfo.objects.filter( |
|
| 39 |
+ msg_type=msg_type |
|
| 40 |
+ ).order_by( |
|
| 41 |
+ '-updated_at' |
|
| 42 |
+ ) |
|
| 43 |
+ type_messages, left = pagination(type_messages, page, num) |
|
| 44 |
+ type_messages = [msg.msg_info for msg in type_messages] |
|
| 45 |
+ |
|
| 46 |
+ return JsonResponse({
|
|
| 47 |
+ 'status': 200, |
|
| 48 |
+ 'message': u'获取消息列表成功', |
|
| 49 |
+ 'data': {
|
|
| 50 |
+ 'messages': type_messages, |
|
| 51 |
+ 'left': left, |
|
| 52 |
+ }, |
|
| 53 |
+ }) |
|
| 54 |
+ |
|
| 55 |
+ |
|
| 56 |
+def message_read_api(request): |
|
| 57 |
+ pk = int(request.GET.get('pk', -1))
|
|
| 58 |
+ |
|
| 59 |
+ try: |
|
| 60 |
+ message = UserMessageInfo.objects.get(pk=pk) |
|
| 61 |
+ except UserMessageInfo.DoesNotExist: |
|
| 62 |
+ return JsonResponse({
|
|
| 63 |
+ 'status': 4091, |
|
| 64 |
+ 'message': u'该消息不存在' |
|
| 65 |
+ }) |
|
| 66 |
+ |
|
| 67 |
+ message.read = True |
|
| 68 |
+ message.save() |
|
| 69 |
+ |
|
| 70 |
+ return JsonResponse({
|
|
| 71 |
+ 'status': 200, |
|
| 72 |
+ 'message': u'已读消息成功', |
|
| 73 |
+ }) |
@@ -44,8 +44,9 @@ INSTALLED_APPS = ( |
||
| 44 | 44 |
'api', |
| 45 | 45 |
'account', |
| 46 | 46 |
'group', |
| 47 |
- 'photo', |
|
| 47 |
+ 'message', |
|
| 48 | 48 |
'operation', |
| 49 |
+ 'photo', |
|
| 49 | 50 |
) |
| 50 | 51 |
|
| 51 | 52 |
INSTALLED_APPS += ('multidomain', )
|
@@ -172,6 +173,15 @@ THUMBNAIL_MAX_WIDTH = 360 |
||
| 172 | 173 |
DOMAIN = 'http://pai.ai' |
| 173 | 174 |
IMG_DOMAIN = 'http://img.pai.ai' |
| 174 | 175 |
|
| 176 |
+# 消息图片设置 |
|
| 177 |
+PAI2_LOGO_URL = DOMAIN + '/static/pai2/img/paiai_96_96.png' |
|
| 178 |
+ |
|
| 179 |
+SYSTEM_MESSAGE_AVATAR = PAI2_LOGO_URL |
|
| 180 |
+COMMENT_MESSAGE_AVATAR = PAI2_LOGO_URL |
|
| 181 |
+THUMBUP_MESSAGE_AVATAR = PAI2_LOGO_URL |
|
| 182 |
+ |
|
| 183 |
+MESSAGE_NUM_PER_PAGE = 10 |
|
| 184 |
+ |
|
| 175 | 185 |
try: |
| 176 | 186 |
from local_settings import * |
| 177 | 187 |
except ImportError: |
@@ -0,0 +1,15 @@ |
||
| 1 |
+# -*- coding: utf-8 -*- |
|
| 2 |
+ |
|
| 3 |
+from django.db.models.query import QuerySet |
|
| 4 |
+ |
|
| 5 |
+ |
|
| 6 |
+def pagination(queryset, page, num=10): |
|
| 7 |
+ """ |
|
| 8 |
+ DIY Pagination Funciton |
|
| 9 |
+ :param queryset: |
|
| 10 |
+ :param page: |
|
| 11 |
+ :param num: the number of query for one page |
|
| 12 |
+ :return: the query of the page, the number of query left after the page |
|
| 13 |
+ """ |
|
| 14 |
+ start, end, total = num * (page - 1), num * page, queryset.count() if isinstance(queryset, QuerySet) else len(queryset) |
|
| 15 |
+ return queryset[start: end], max(total - end, 0) |