flyapps/fir_ser/common/notify/ntasks.py

#!/usr/bin/env python
# -*- coding:utf-8 -*-
# project: fir_ser
# filename: ntasks
# author: liuyu
# data: 2022/3/26
import datetime
import logging

from api.utils.modelutils import get_notify_wx_queryset, get_wx_nickname
from common.base.magic import magic_wrapper, magic_notify
from common.cache.storage import NotifyLoopCache
from common.core.sysconfig import Config
from common.libs.mp.wechat import WxTemplateMsg
from common.libs.sendmsg.template_content import get_user_download_times_not_enough_html_content, \
    get_developer_devices_not_enough_html_content, get_developer_cert_expired_html_content, \
    get_user_download_times_over_limit_html_content
from common.notify.utils import notify_by_email
from xsign.models import AppIOSDeveloperInfo
from xsign.utils.modelutils import get_developer_devices

logger = logging.getLogger(__name__)


def download_times_not_enough(user_obj, msg):
    """
    1, '下载次数不足'
    :param msg:
    :param user_obj:
    :return:
    """
    message_type = 1
    for wx_user_obj in get_notify_wx_queryset(user_obj, message_type):
        res = WxTemplateMsg(wx_user_obj.openid, get_wx_nickname(wx_user_obj.openid)).download_times_not_enough_msg(
            user_obj.first_name, user_obj.download_times, msg)
        logger.info(f'user_obj {user_obj} download times not enough result: {res}')
    notify_by_email(user_obj, message_type, get_user_download_times_not_enough_html_content(user_obj))


def apple_developer_devices_not_enough(user_obj, device_count):
    """
    0, '签名余额不足'
    :param user_obj:
    :return:
    """
    message_type = 0
    msg = f"您当前账户超级签名可用设备仅剩 {device_count}，已超过您设置的阈值 {user_obj.notify_available_signs}，为了避免业务使用，望您尽快添加苹果开发者!"
    for wx_user_obj in get_notify_wx_queryset(user_obj, message_type):
        res = WxTemplateMsg(wx_user_obj.openid,
                            get_wx_nickname(wx_user_obj.openid)).apple_developer_devices_not_enough_msg(
            user_obj.first_name, device_count, msg)
        logger.info(f'user_obj {user_obj} sign devices not enough result: {res}')
    notify_by_email(user_obj, message_type, get_developer_devices_not_enough_html_content(user_obj, device_count))


def apple_developer_cert_expired(user_obj, developer_queryset):
    """
    6, '证书到期消息'
    :param developer_queryset:
    :param user_obj:
    :return:
    """
    message_type = 6
    developer_count = developer_queryset.count()
    developer_obj = developer_queryset.first()
    expired_time = developer_obj.cert_expire_time.strftime("%Y年%m月%d")
    if developer_count == 1:
        issuer_id = developer_obj.issuer_id
        cert_id = developer_obj.certid
    else:
        issuer_id = f'{developer_obj.issuer_id} 等 {developer_count} 个开发者ID'
        cert_id = f'{developer_obj.certid} 等 {developer_count} 个证书ID'

    for wx_user_obj in get_notify_wx_queryset(user_obj, message_type):
        res = WxTemplateMsg(wx_user_obj.openid, get_wx_nickname(wx_user_obj.openid)).cert_expired_msg(issuer_id,
                                                                                                      cert_id,
                                                                                                      expired_time)
        logger.info(f'user_obj {user_obj} apple developer cert expired result: {res}')

    notify_by_email(user_obj, message_type, get_developer_cert_expired_html_content(user_obj, developer_queryset))


def check_user_download_times(user_obj, days=None):
    if days is None:
        days = [0, 3, 7]
    if user_obj.notify_available_downloads == 0 or user_obj.notify_available_downloads < user_obj.download_times:
        return
    msg = get_user_download_times_over_limit_html_content(user_obj)
    notify_rules = [
        {
            'func': magic_wrapper(lambda obj: obj.download_times < obj.notify_available_downloads, user_obj),
            'notify': days,
            'cache': NotifyLoopCache(user_obj.uid, 'download_times'),
            'notify_func': [magic_wrapper(download_times_not_enough, user_obj, msg)]
        }
    ]
    magic_notify(notify_rules)


def check_apple_developer_devices(user_obj, days=None):
    if days is None:
        days = [0, 3, 7]
    developer_queryset = AppIOSDeveloperInfo.objects.filter(user_id=user_obj)
    if developer_queryset.count() == 0:
        return
    developer_used_info = get_developer_devices(developer_queryset)
    device_count = developer_used_info.get('can_sign_number', 0)
    if user_obj.notify_available_signs == 0 or device_count > user_obj.notify_available_signs:
        return
    notify_rules = [
        {
            'func': magic_wrapper(lambda obj: device_count < obj.notify_available_signs, user_obj),
            'notify': days,
            'cache': NotifyLoopCache(user_obj.uid, 'sign_device_times'),
            'notify_func': [magic_wrapper(apple_developer_devices_not_enough, user_obj, device_count)]
        }
    ]
    magic_notify(notify_rules)


def check_apple_developer_cert(user_obj, expire_day=7):
    expire_time = datetime.datetime.now() + datetime.timedelta(days=expire_day)
    developer_queryset = AppIOSDeveloperInfo.objects.filter(user_id=user_obj, status__in=Config.DEVELOPER_USE_STATUS,
                                                            cert_expire_time__lte=expire_time).order_by(
        'cert_expire_time')
    if developer_queryset.count() == 0:
        return

    notify_rules = [
        {
            'func': magic_wrapper(lambda: True),
            'notify': [0, 3, 7],
            'cache': NotifyLoopCache(user_obj.uid, 'developer_cert'),
            'notify_func': [magic_wrapper(apple_developer_cert_expired, user_obj, developer_queryset)]
        }
    ]
    magic_notify(notify_rules)