# coding=utf-8 from __future__ import absolute_import, print_function, unicode_literals import random import datetime from datetime import timedelta from celery import shared_task from gm_types.gaia import CONST_STRINGS as const_strings from gm_types.push.enum import AUTOMATED_PUSH from gm_types.mimas import NOTICATION_LIST_CATEGORY from gm_types.push import PUSH_INFO_TYPE from six.moves import xrange from gm_protocol import GmProtocol from qa.utils.statistic import get_async_args_v2 from talos.cache.base import diary_pv_cache from talos.models.diary import Diary from talos.models.diary import DiaryVote from talos.models.topic import TopicVote from talos.services.user import UserService from utils.protocol import gm_protocol from utils.push import vote_push, push_task_to_user_multi from utils.rpc import rpc_client from .tasks import view_increase, view_increase_amount from ..models.topic import Problem from talos.cache.base import fake_vote_cache from talos.tools.vote_tool import VoteTool from talos.cache.base import vote_cache from talos.manager.diary import diary_list_manager from communal.tasks import intelligent_push_task from talos.services.other import get_user_lastest_device_app_version_by_user_id from utils.common import is_version_gray _MAX_SLEEP_PERSON = 5000 def get_fake_view_num(start, end): return random.randint(start, end) def get_sleep_user(repeat_times): user_ids = rpc_client['api/user/get_sleep_user'](number=repeat_times).unwrap() return user_ids def normal_vote(user, item, is_diary=False, incr_range=[5, 15]): """ :param user: 点赞的用户 :param item: 日记本或者日记贴 :param is_diary: 是否日记本 :param incr_range: 浏览量增加的范围 :return: """ if is_diary: diary = item like_num = diary.like_num # 增加点赞数 diary.like_num += 1 diary.save() # 增加浏览量 fake_view_num = get_fake_view_num(*incr_range) diary_pv_cache.incrby(str(diary.id), fake_view_num) # 点赞推送 日记本点赞进入防干扰 alert = u'有1位朋友给你点了赞，快来看看吧>>' push_url = GmProtocol().get_diary_detail(id=diary.id) version = get_user_lastest_device_app_version_by_user_id(diary.user_id) # 推送跳转到消息页的赞列表 if is_version_gray(version, '7.29.0'): push_url = GmProtocol().vote_or_favor_list( segment_index=NOTICATION_LIST_CATEGORY.VOTE, new_vote=True, ) intelligent_push_task.delay( content_id=diary.id, user_ids=[diary.user_id], push_type=AUTOMATED_PUSH.DIARY_RECEIVED_PRAISE, platform=None, extra={ 'type': PUSH_INFO_TYPE.GM_PROTOCOL, 'pushUrl': push_url, 'push_url': push_url, }, alert=alert, others={ "title": "@{} 给你点了赞".format(user.nick_name), "alert": alert, }, labels={}, ) else: topic = item # 增加点赞数 view_increase(Problem.get_redis_vote_key(), topic.id) if topic.diary_id: # 增加对应的日记点赞数 view_increase(Diary.get_redis_diary_topic_vote_key(), topic.diary_id) # 增加浏览量 view_increase_amount(const_strings.TOPIC_VIEW, topic.id, get_fake_view_num(*incr_range)) #topic.diary.operation_time 手术时间 topic.operation_date if not topic.diary or not topic.diary.operation_time or not topic.operation_date: return end, start = topic.operation_date, topic.diary.operation_time days = (end.replace(hour=0, minute=0, second=0, microsecond=0) - start.replace(hour=0, minute=0, second=0, microsecond=0)).days diary_title = u'{diary_title}（第{days}天）'.format(diary_title=topic.diary.title, days=days) alert = u'{user_name}刚刚赞了你的《{diary_title}》，比心~'.format(user_name=str(user.nick_name), diary_title=diary_title) push_url = GmProtocol().get_topic_detail(id=topic.id) user_id = topic.user_id version = get_user_lastest_device_app_version_by_user_id(user_id) # 推送跳转到消息页的赞列表 if is_version_gray(version, '7.29.0'): push_url = GmProtocol().vote_or_favor_list( segment_index=NOTICATION_LIST_CATEGORY.VOTE, new_vote=True, ) vote_push(user_id=user_id, alert=alert, push_type=AUTOMATED_PUSH.DIARY_POST_RECEIVED_PRAISE, push_url=push_url) @shared_task def fake_vote(repeat_times, item_id, need_view_increase=True, incr_range=[1, 1], is_topic=True, force_push=False, author_name=None, topic_title=None): """ Get random {amount} users who's last login time earlier than 2015,and let them vote for topic each person one time :param repeat_times: :param topic: :param need_view_increase :param incr_num 增加多少浏览数 :param is_topic :return: """ # 获取沉睡用户 user_extras = get_sleep_user(repeat_times) if is_topic: return fake_vote_topic(user_extras, item_id, need_view_increase, incr_range, force_push, author_name, topic_title) return fake_vote_diary(user_extras, item_id, need_view_increase, incr_range, force_push) def fake_vote_diary(users, diary_id, need_view_increase=True, incr_range=[1, 1], force_push=False): diary = Diary.objects.get(pk=diary_id) # 帖子下线, 不处理 if not diary.is_online: return diary_fake_vote_key = "diary_fake_vote_{id}".format(id=diary_id) is_set = fake_vote_cache.set(diary_fake_vote_key, 1, nx=True, ex=30) if not is_set: return cnt, u_nick_name = 0, '' for ue in users: _, created = DiaryVote.objects.get_or_create(user_id=ue, diary_id=diary.id, is_fake=True) if created: # 对日记本作者增加点赞数 u = UserService.get_user_by_user_id(diary.user_id) u.incr_vote_count() if not u_nick_name: vote_user = UserService.get_user_by_user_id(ue) u_nick_name = vote_user.nickname cnt += 1 if not need_view_increase: return # 增加可视点赞数 like_num = diary.like_num before_fake = like_num diary.like_num += cnt diary.save() after_fake = diary.like_num view_num = 0 # 每个点赞增加随机浏览量 for i in xrange(cnt): view_num += random.randint(*incr_range) # 增加可视浏览数 diary_pv_cache.incrby(str(diary.id), view_num) if force_push: alert = u'{user_name} 刚刚感谢了你的《{diary_title}》，比心~'.format(user_name=u_nick_name, diary_title=diary.title) push_url = GmProtocol().get_diary_detail(id=diary.id) vote_push(diary.user_id, alert=alert, push_type=AUTOMATED_PUSH.DIARY_RECEIVED_PRAISE, push_url=push_url) def fake_vote_topic(userids, topic_id, need_view_increase=True, incr_range=[1, 1], force_push=False, author_name=None, topic_title=None): # reload data from db now_date = datetime.datetime.now() if not 9 <= now_date.hour < 22: # 超过晚十点灌水任务不执行 return topic_fake_vote_key = "topic_fake_vote_{id}".format(id=topic_id) is_set = fake_vote_cache.set(topic_fake_vote_key, 1, nx=True, ex=30) if not is_set: return topic = Problem.objects.get(pk=topic_id) if not topic.is_online: return cnt, u_nick_name = 0, '' # 增加可视的点赞数 before_fake = topic.vote_amount view_increase_amount(const_strings.TOPIC_VOTE, topic.id, cnt) after_fake = topic.vote_amount for ue in userids: if topic.is_offline: continue topic_vote, created = TopicVote.objects.get_or_create( user_id=ue, topic_id=topic.id, is_fake=True, topic_author_id=topic.user_id, ) if created: # 对日记作者增加点赞数 u = UserService.get_user_by_user_id(topic.user_id) u.incr_vote_count() if not u_nick_name: vote_user = UserService.get_user_by_user_id(ue) u_nick_name = vote_user.nickname cnt += 1 view_increase(Problem.get_redis_vote_key(), topic.id) if need_push(before_fake, after_fake, force_push=force_push): view_increase(Problem.get_redis_vote_key(), topic.id) alert = u'{user_name}刚刚赞了你的《{diary_title}》，比心~'.format( user_name=u_nick_name, diary_title=topic.diary.title) push_url = GmProtocol().get_topic_detail(id=topic_id) vote_push(user_id=topic.user_id, alert=alert, is_force=force_push, push_url=push_url, push_type=AUTOMATED_PUSH.DIARY_POST_RECEIVED_PRAISE) if not need_view_increase: return # 为每个点赞数增加可视浏览数 for i in xrange(cnt): fake_view_topic(topic.id, incr_range) @shared_task def fake_view_topic(topic_id, incr_range=[1, 1]): """ 增加帖子浏览量 :param topic: :return: """ incr_num = get_fake_view_num(*incr_range) view_increase_amount(const_strings.TOPIC_VIEW, topic_id, incr_num) def need_push(before, after, force_push=False): # 放大后点赞数超整10, 或者不满十个赞但需要强制推送 before = int(before) after = int(after) return (after / 10 - before / 10) >= 1 or (after < 10 and force_push) def process_topic_vote_for_create(topic_id): """ 处理创建日记贴之后的点赞和增加浏览量 :param topic: :return: """ # 异步增加点赞数和浏览量 # 规则1, 发帖[30s, 60s], 增加浏览量 countdown, incr_range = get_async_args_v2(const_strings.TOPIC_VOTE_RULE1) fake_view_topic.apply_async(args=(topic_id,), kwargs={'incr_range': incr_range}, countdown=countdown[0]) countdowns, incr_range = get_async_args_v2(const_strings.CREATE_TOPIC_VOTE) for index, cd in enumerate(countdowns): task_info = fake_vote.apply_async(args=(1, topic_id), kwargs={'incr_range': incr_range, 'force_push': True}, countdown=cd ) from talos.logger import qiniu_logger qiniu_logger.info("task_id====={task_id}".format(task_id=task_info.task_id)) @shared_task def diary_push_timing(): # 灌水1.0上线 下线所有之前的灌水逻辑 return